Exibir código-fonte para DAO - Data Access Object

Links relacionados: [[Banco de Dados]]

= Base de Dados =

Para o exemplo, criei a base de dados chamada dao.

<syntaxhighlight lang=sql line>
create table funcionario (
	idfuncionario serial,
	nome varchar(50),
	salario numeric(7,2),
	primary key (idfuncionario)
);

create table dependente (
	idfuncionario integer references funcionario,
	iddependente serial,
	nome varchar(50),
	idade integer,
	primary key (iddependente)
);
</syntaxhighlight>

= Classe de Conexão =

<syntaxhighlight lang=java line>
import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.SQLException;

public class DatabaseConnection {
	static private Connection connection = null;
	static private String url = "";
	static private String username = "";
	static private String password = "";
	static private String dbdriver = "";

	static public void setConfig(String iurl, String uname, String pass,
			String dbdrv) {
		url = iurl;
		username = uname;
		password = pass;
		dbdriver = dbdrv;
	}

	static public Connection getConnection() {
		if (connection == null) {
			try {
				Class.forName(dbdriver);
				connection = DriverManager.getConnection(url, username,
						password);
			} catch (SQLException e1) {
				e1.printStackTrace();
			} catch (ClassNotFoundException e) {
				e.printStackTrace();
			}
		}
		return connection;
	}
}
</syntaxhighlight>

= Objetos de Transferência ==

== Funcionario ==

<syntaxhighlight lang=java line>
public class Funcionario {
	private int idFuncionario;
	private String nome;
	private float salario;

	public int getIdFuncionario() {
		return idFuncionario;
	}

	public String getNome() {
		return nome;
	}

	public float getSalario() {
		return salario;
	}

	public void setIdFuncionario(int id) {
		idFuncionario = id;
	}

	public void setNome(String n) {
		nome = n;
	}

	public void setSalario(float s) {
		salario = s;
	}
}
</syntaxhighlight>

== Dependente ==

<syntaxhighlight lang=java line>
public class Dependente {
	private int idFuncionario;
	private int idDependente;
	private String nome;
	private int idade;

	public int getIdFuncionario() {
		return idFuncionario;
	}

	public int getIdDependente() {
		return idDependente;
	}

	public String getNome() {
		return nome;
	}

	public int getIdade() {
		return idade;
	}

	public void setIdFuncionario(int id) {
		idFuncionario = id;
	}

	public void setIdDependente(int id) {
		idDependente = id;
	}

	public void setNome(String n) {
		nome = n;
	}

	public void setIdade(int i) {
		idade = i;
	}
}
</syntaxhighlight>

= Objetos de Acesso =

== Funcionario DAO ==

<syntaxhighlight lang=java line>
import java.sql.Connection;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.Statement;
import java.util.Vector;

public class FuncionarioDAO {
	private int Position;
	private Vector<Funcionario> funcionarioList;

	public FuncionarioDAO() {
		Position = 0;
		funcionarioList = new Vector<Funcionario>();
	}

	public void load(String filter) {
		try {
			Connection conn = DatabaseConnection.getConnection();
			Statement stmt = conn.createStatement();

			String query = "select * from funcionario";
			if (filter.length() > 0) {
				query += " where " + filter;
			}
			System.out.println("QUERY: " + query);
			ResultSet rs = stmt.executeQuery(query);
			while (rs.next()) {
				Funcionario func = new Funcionario();
				func.setIdFuncionario(rs.getInt("idfuncionario"));
				func.setNome(rs.getString("nome"));
				func.setSalario(rs.getFloat("salario"));
				funcionarioList.add(func);
			}
			stmt.close();
		} catch (SQLException e) {
			e.printStackTrace();
		}
	}

	public void insert(Funcionario func) {
		try {
			Connection conn = DatabaseConnection.getConnection();
			Statement stmt = conn.createStatement();

			String query = "insert into funcionario (idfuncionario, nome, salario) "
					+ "values ("
					+ func.getIdFuncionario()
					+ ",'"
					+ func.getNome() + "', " + func.getSalario() + ")";
			System.out.println("QUERY: " + query);
			stmt.executeUpdate(query);
			stmt.close();
		} catch (SQLException e) {
			e.printStackTrace();
		}
		this.clear();
	}

	public void update(Funcionario func) {
		try {
			Connection conn = DatabaseConnection.getConnection();
			Statement stmt = conn.createStatement();

			String query = "update funcionario set " + "nome = '"
					+ func.getNome() + "', " + "salario = " + func.getSalario()
					+ " " + "where idfuncionario = " + func.getIdFuncionario();
			System.out.println("QUERY: " + query);
			stmt.executeUpdate(query);
			stmt.close();
		} catch (SQLException e) {
			e.printStackTrace();
		}
		this.clear();
	}

	public void delete(int func) {
		try {
			Connection conn = DatabaseConnection.getConnection();
			Statement stmt = conn.createStatement();
			String query = "delete from dependente where idFuncionario = "
					+ func;
			System.out.println("QUERY: " + query);
			stmt.executeUpdate(query);
			query = "delete from funcionario where idFuncionario = " + func;
			System.out.println("QUERY: " + query);
			stmt.executeUpdate(query);
			stmt.close();
		} catch (SQLException e) {
			e.printStackTrace();
		}
		this.clear();
	}

	public void clear() {
		Position = 0;
		funcionarioList.clear();
	}

	public void reset() {
		Position = 0;
	}

	public int size() {
		return funcionarioList.size();
	}

	public Funcionario next() {
		Funcionario func = null;
		if (Position < funcionarioList.size()) {
			func = funcionarioList.get(Position);
			Position++;
		}
		return func;
	}
}
</syntaxhighlight>

== Dependente DAO ==

<syntaxhighlight lang=java line>
import java.sql.Connection;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.Statement;
import java.util.Vector;

public class DependenteDAO {
	private int Position;
	private Vector<Dependente> dependenteList;

	public DependenteDAO() {
		Position = 0;
		dependenteList = new Vector<Dependente>();
	}

	public void load(int func) {
		try {
			Connection conn = DatabaseConnection.getConnection();
			Statement stmt = conn.createStatement();

			String query = "select * from dependente where idFuncionario = "
					+ func;
			System.out.println("QUERY: " + query);
			ResultSet rs = stmt.executeQuery(query);
			while (rs.next()) {
				Dependente dep = new Dependente();
				dep.setIdFuncionario(func);
				dep.setIdDependente(rs.getInt("iddependente"));
				dep.setNome(rs.getString("nome"));
				dep.setIdade(rs.getInt("idade"));
				dependenteList.add(dep);
			}
			stmt.close();
		} catch (SQLException e) {
			e.printStackTrace();
		}
	}

	public void insert(Dependente dep) {
		try {
			Connection conn = DatabaseConnection.getConnection();
			Statement stmt = conn.createStatement();

			String query = "insert into dependente (idfuncionario, nome, idade) "
					+ "values ("
					+ dep.getIdFuncionario()
					+ ", '"
					+ dep.getNome() + "', " + dep.getIdade() + ")";
			System.out.println("QUERY: " + query);
			stmt.executeUpdate(query);
			stmt.close();
		} catch (SQLException e) {
			e.printStackTrace();
		}
		this.clear();
	}

	public void update(Dependente dep) {
		try {
			Connection conn = DatabaseConnection.getConnection();
			Statement stmt = conn.createStatement();

			String query = "update dependente set " + "nome = '"
					+ dep.getNome() + "', " + "idade = " + dep.getIdade() + " "
					+ "where idfuncionario = " + dep.getIdFuncionario() + " "
					+ "and iddependente = " + dep.getIdDependente();
			System.out.println("QUERY: " + query);
			stmt.executeUpdate(query);
			stmt.close();
		} catch (SQLException e) {
			e.printStackTrace();
		}
		this.clear();
	}

	public void delete(int dep) {
		try {
			Connection conn = DatabaseConnection.getConnection();
			Statement stmt = conn.createStatement();
			String query = "delete from dependente where idDependente = " + dep;
			System.out.println("QUERY: " + query);
			stmt.executeUpdate(query);
			stmt.close();
		} catch (SQLException e) {
			e.printStackTrace();
		}
		this.clear();
	}

	public void clear() {
		Position = 0;
		dependenteList.clear();
	}

	public void reset() {
		Position = 0;
	}

	public int size() {
		return dependenteList.size();
	}

	public Dependente next() {
		Dependente dep = null;
		if (Position < dependenteList.size()) {
			dep = dependenteList.get(Position);
			Position++;
		}
		return dep;
	}
}
</syntaxhighlight>

= Aplicação =

<syntaxhighlight lang=java line>
import java.util.Scanner;

public class Main {
	private FuncionarioDAO funcionarios;
	private DependenteDAO dependentes;
	Scanner entrada;

	public Main() {
		DatabaseConnection.setConfig("jdbc:postgresql://localhost/dao",
				"usuario", "****", "org.postgresql.Driver");
		funcionarios = new FuncionarioDAO();
		dependentes = new DependenteDAO();
		entrada = new Scanner(System.in);
	}

	public void showMenu() {
		System.out.println("Comandos Funcionarios:");
		System.out.println("\t lf : listar funcionario");
		System.out.println("\t if : inserir novo funcionario");
		System.out.println("\t df : deletar funcionario");
		System.out.println("\t af : alterar dados de um funcionario");
		System.out.println("Comandos Dependentes:");
		System.out.println("\t ld : listar dependentes de um funcionario");
		System.out.println("\t id : inserir dependente para um funcionario");
		System.out.println("\t dd : deletar dependente");
		System.out.println("\t ad : alterar dados de um dependente");
		System.out.println("sair : Sai do programa");
		System.out.println("");
		System.out.print("Comando: ");
	}

	public void menu() {
		String choice = "";
		while (!choice.contains("sair")) {
			showMenu();
			choice = entrada.nextLine();
			if (choice.contains("lf"))
				listarFuncionarios();
			else if (choice.contains("if"))
				inserirFuncionario();
			else if (choice.contains("df"))
				deletarFuncionario();
			else if (choice.contains("af"))
				alterarFuncionario();
			else if (choice.contains("ld"))
				listarDependentes();
			else if (choice.contains("id"))
				inserirDependente();
			else if (choice.contains("dd"))
				deletarDependente();
			else if (choice.contains("ad"))
				alterarDependente();
		}
	}

	public void listarFuncionarios() {
		System.out.print("Filtro: ");
		funcionarios.load(entrada.nextLine());
		Funcionario f;
		while ((f = funcionarios.next()) != null) {
			System.out.println("" + f.getIdFuncionario() + " - " + f.getNome()
					+ " - " + f.getSalario());
		}
	}

	public void listarDependentes() {
		System.out.print("Codigo do Funcionario : ");
		dependentes.load(Integer.valueOf(entrada.nextLine()));
		Dependente d;
		while ((d = dependentes.next()) != null) {
			System.out.println("" + d.getIdDependente() + " - " + d.getNome()
					+ " - " + d.getIdade());
		}
	}

	public void deletarFuncionario() {
		System.out.print("Codigo do Funcionario a deletar : ");
		funcionarios.delete(Integer.valueOf(entrada.nextLine()));
	}

	public void deletarDependente() {
		System.out.print("Codigo do Dependente a deletar : ");
		dependentes.delete(Integer.valueOf(entrada.nextLine()));
	}

	public Funcionario lerFuncionario() {
		Funcionario f = new Funcionario();
		System.out.print("idFuncionario: ");
		f.setIdFuncionario(Integer.valueOf(entrada.nextLine()));
		System.out.print("Nome: ");
		f.setNome(entrada.nextLine());
		System.out.print("Salario: ");
		f.setSalario(Float.valueOf(entrada.nextLine()));
		return f;
	}

	public void inserirFuncionario() {
		System.out.println("Inserindo novo Funcionario");
		funcionarios.insert(this.lerFuncionario());
	}

	public void alterarFuncionario() {
		System.out.println("Alterando Funcionario");
		funcionarios.update(this.lerFuncionario());
	}

	public Dependente lerDependente() {
		Dependente d = new Dependente();
		System.out.print("idFuncionario: ");
		d.setIdFuncionario(Integer.valueOf(entrada.nextLine()));
		System.out.print("Nome: ");
		d.setNome(entrada.nextLine());
		System.out.print("Idade: ");
		d.setIdade(Integer.valueOf(entrada.nextLine()));
		return d;
	}

	public void inserirDependente() {
		System.out.println("Inserindo novo Dependente");
		dependentes.insert(this.lerDependente());
	}

	public void alterarDependente() {
		System.out.println("Alterando Dependente");
		System.out.print("idDependente: ");
		int id = Integer.valueOf(entrada.nextLine());
		Dependente d = this.lerDependente();
		d.setIdDependente(id);
		dependentes.update(d);
	}

	public static void main(String[] arguments) {
		Main m = new Main();
		m.menu();
	}
}
</syntaxhighlight>