JAVA 자바 JDBC 문제 연습

🎯 자바 JDBC 연습문제를 풀어본다.

 

 

 

 

0. 기본세팅

String driver = "oracle.jdbc.driver.OracleDriver";
String url = "jdbc:oracle:thin:@localhost:1521:orcl";
String userid = "scott";
String passwd = "tiger";

Connection con= null;
ResultSet rs= null;
PreparedStatement pstmt= null;

public JDBC_Method_divide_Test() throws ClassNotFoundException, SQLException {
        Class.forName(driver);
}

연동을 위한 기본 정보를 입력하고 드라이버를 로딩한다.

 

 

1. search 메소드 만들기

public String search(String loc, String name) {
    //지역만 넘어온 경우 지역만 검색 결과 리턴
    //이름만 넘오온 경우 이름만 검색 결과 리턴
    //이름, 지역이 넘어온 경우 두 가지를 이용 검색 결과 리턴
}

위 세가지 상황에 대한 조건문을 채워야 한다. search함수는 인자로 지역과 이름을 받는다.

 

#1 첫번째 if

public String search(String loc, String dname) {
        String data = null;
        try {
            con = DriverManager.getConnection(url, userid, passwd);

            if(dname==null && loc != null) {
                String sql = "select loc from dept where loc = ?";
                pstmt = con.prepareStatement(sql);
                pstmt.setString(1, loc);
                rs = pstmt.executeQuery();
                while(rs.next()) {
                    String loc1 = rs.getString(1);
                    data = loc1;
                }

지역값만 받은 경우 받은 지역에 대한 검색결과를 리턴하는 sql문을 작성한다.

#2 두번째 else if

}else if(loc == null && dname !=null) {
                String sql = "select dname from dept where dname = ?";
                pstmt = con.prepareStatement(sql);
                pstmt.setString(1, dname);
                rs = pstmt.executeQuery();
                while(rs.next()) {
                    String dname1 = rs.getString("dname");
                    data = dname1;
                }

이름만 받은 경우 받은 해당 이름에 대한 검색결과를 리턴하는 코드를 작성한다.

#3 else 

}else {
                String sql = "select loc, dname from dept where loc=? and dname=?";
                pstmt = con.prepareStatement(sql);
                pstmt.setString(1, loc);
                pstmt.setString(2, dname);
                rs = pstmt.executeQuery();
                while(rs.next()) {
                    String loc1 = rs.getString("loc");
                    String dname1 = rs.getString("dname");
                //    System.out.println(loc1+"\t"+dname1);
                    data = loc1 +"\t"+ dname1;
                }
            }

이름과 지역 둘 다 받은 경우 두 정보 모두 뽑아주는 코드를 작성한다.

#예외잡기

}catch (SQLException e) {
            e.printStackTrace();
        } finally {
            try {
                if(rs!=null)rs.close();
                if(pstmt!=null)pstmt.close();
                if(con!=null)con.close();
            }catch(SQLException e) {
                e.getStackTrace();
            }
        }return data;
    }

try문이 끝났으니 예외를 처리한다. 실행된 역순으로 자원을 반납하고 최종결과를 리턴한다.

# 메인함수에 프린트

public static void main(String[] args) throws SQLException, ClassNotFoundException {
        JDBC_Method_divide_Test test= new JDBC_Method_divide_Test();
        String result1 = test.search("제주", null);
        String result2 = test.search(null, "영업");
        String result3 =test.search("제주", "영업");
        System.out.println(result1);
        System.out.println(result2);
        System.out.println(result3);

인자리스트를 맞춰서 세가지 조건에 해당하는 결과를 출력한다.

 

 

2. getAllDept 메소드 만들기

public String  getAllDept(){
    //부서전체 select 
    //결과를 메인으로 리턴 메인에서 전체 데이터 출력 
}

dept테이블의 전체 데이터를 출력하는 함수를 만든다.

 

public String  getAllDept(){
        String data = "";
        try {
            con = DriverManager.getConnection(url, userid, passwd);
            String sql = "select * from dept";
            pstmt = con.prepareStatement(sql);
            rs = pstmt.executeQuery();

            while (rs.next()) {
                int deptno = rs.getInt(1);
                String dname = rs.getString(2);
                String loc = rs.getString(3);
                data += deptno + "\t" + dname + "\t" + loc +"\n";
            }
        } catch (SQLException e) {
            e.printStackTrace();
        } finally  {
            try {
                if(rs!=null)rs.close();
                if(pstmt!=null)pstmt.close();
                if(con!=null)con.close();
            }catch(SQLException e) {
                e.getStackTrace();
            }
        }return data;

#메인

System.out.println(test.getAllDept());

모든 데이터를 리턴하는 메소드를 만들고 메인에서 호출한다.

 

# ArrayList 사용

public ArrayList<String>  getAllDept(){
        ArrayList<String> list = new ArrayList<String>();
        try {
            con = DriverManager.getConnection(url, userid, passwd);
            String sql = "select * from dept";
            pstmt = con.prepareStatement(sql);
            rs = pstmt.executeQuery();
            while (rs.next()) {
                String data = "";
                data += rs.getInt(1);
                data += rs.getString(2);
                data += rs.getString(3);
                data += "\n";
                list.add(data);
            }
        } catch (SQLException e) {
            e.printStackTrace();
        } finally  {
            try {
                if(rs!=null)rs.close();
                if(pstmt!=null)pstmt.close();
                if(con!=null)con.close();
            }catch(SQLException e) {
                e.getStackTrace();
            }
        }return list;
    }

ArrayList<String> list = test.getAllDept();
        for(String x : list) {
            System.out.println(x);
        }

ArrayList를 사용해서 함수를 만들어본다. select한 데이터를 리스트에 추가하고 리스트를 리턴한다. 메인함수 안에서 반복문을 사용해 리스트의 전체 데이터를 출력한다.

 

 

3. searchDeptByLoc 메소드 만들기

public  String searchDeptByLoc(String loc){
    //주소로 select 
    //검색한 내용을 ? 로 리턴 메인에서 전체 데이터 출력 
}

지역값을 받아서 검색결과를 리턴하는 함수를 만든다.

 

public  String searchDeptByLoc(String loc){
        String data = "";
        try {
            con = DriverManager.getConnection(url, userid, passwd);
            String sql = "select * from dept where loc=?";
            pstmt = con.prepareStatement(sql);
            pstmt.setString(1, loc);
            rs = pstmt.executeQuery();
            while (rs.next()) {
                int deptno = rs.getInt(1);
                String dname = rs.getString(2);
                String loc1 = rs.getString(3);
                data += deptno + "\t" + dname + "\t" + loc1 +"\n";
            }    
        } catch (SQLException e) {
            e.printStackTrace();
        }finally  {
            try {
                if(rs!=null)rs.close();
                if(pstmt!=null)pstmt.close();
                if(con!=null)con.close();
            }catch(SQLException e) {
                e.getStackTrace();
            }
        }
        return data;
    }

System.out.println(test.searchDeptByLoc("제주"));

 

 

4. searchDeptByName 메소드 만들기

public  String searchDeptByName(String dname){
    //부서이름으로로 select 
    //검색한 내용을 ? 로 리턴 메인에서 전체 데이터 출력 
}

부서 이름으로 받아서 검색 결과를 리턴하는 함수를 만든다.

 

public  String searchDeptByName(String dname){
        String data = "";
        try {
            con = DriverManager.getConnection(url, userid, passwd);
            String sql = "select * from dept where dname=?";
            pstmt = con.prepareStatement(sql);
            pstmt.setString(1, dname);
            rs = pstmt.executeQuery();
            while (rs.next()) {
                int deptno = rs.getInt(1);
                String dname1 = rs.getString(2);
                String loc1 = rs.getString(3);
                data += deptno + "\t" + dname1 + "\t" + loc1 +"\n";
            }    
        } catch (SQLException e) {
            e.printStackTrace();
        }finally  {
            try {
                if(rs!=null)rs.close();
                if(pstmt!=null)pstmt.close();
                if(con!=null)con.close();
            }catch(SQLException e) {
                e.getStackTrace();
            }
        }
        return data;
    }

System.out.println(test.searchDeptByName("개발"));

 

 

5. updateDept 메소드 만들기

public int updateDept(String dname, String loc, int deptno){
    //부서번호로 dname, loc업데이트 후 
    //업데이트 결과 갯수 리턴
}

특정 부서번호에 해당하는 부서명과 지역을 update하는 함수를 만든다.

 

public int updateDept(String dname, String loc, int deptno){
        int num = 0;
        try {
            con = DriverManager.getConnection(url, userid, passwd);
            String sql = "update dept set dname=?, loc=? where deptno=?";
            pstmt = con.prepareStatement(sql);
            pstmt.setString(1, dname);
            pstmt.setString(2, loc);
            pstmt.setInt(3, deptno);
            num = pstmt.executeUpdate();
        } catch (SQLException e) {
            e.printStackTrace();
        }finally  {
            try {
                if(rs!=null)rs.close();
                if(pstmt!=null)pstmt.close();
                if(con!=null)con.close();
            }catch(SQLException e) {
                e.getStackTrace();
            }
        }return num;
    }

selec문과 다르게 실행결과 갯수만 리턴한다.

System.out.println(test.updateDept("영업", "판교", 80));

 

 

6. insertDept 메소드 만들기

public int insertDept(String dname, String loc, int deptno){
    //부서하나 추가후 
    //insert 결과 갯수 리턴
}

부서번호와 부서명, 지역명을 받아서 새로운 레코드를 추가하는 메소드를 만든다.

 

public int insertDept(String dname, String loc, int deptno){
        int num = 0;
        try {
            con = DriverManager.getConnection(url, userid, passwd);
            String sql = "insert into dept (deptno, dname, loc) values(?,?,?)";
            pstmt = con.prepareStatement(sql);
            pstmt.setInt(1, deptno);
            pstmt.setString(2, dname);
            pstmt.setString(3, loc);
            num = pstmt.executeUpdate();
        } catch (SQLException e) {
            e.printStackTrace();
        }finally  {
            try {
                if(pstmt!=null)pstmt.close();
                if(con!=null)con.close();
            }catch(SQLException e) {
                e.getStackTrace();
            }
        }return num;
    }

System.out.println(test.insertDept("경영", "강남", 88));

 

 

 

JDBC Dept DTO연습

//  Dept테이블의 하나의 레코드 저장용도

package com.dto;

public class Dept {

    private int deptno;
    private String dname;
    private String loc;

    public Dept() {}

    public Dept(int deptno, String dname, String loc) {
        this.deptno = deptno;
        this.dname = dname;
        this.loc = loc;
    }

    public int getDeptno() {
        return deptno;
    }

    public void setDeptno(int deptno) {
        this.deptno = deptno;
    }

    public String getDname() {
        return dname;
    }

    public void setDname(String dname) {
        this.dname = dname;
    }

    public String getLoc() {
        return loc;
    }

    public void setLoc(String loc) {
        this.loc = loc;
    }

    @Override
    public String toString() {
        return "Dept [deptno=" + deptno + ", dname=" + dname + ", loc=" + loc + "]";
    }

}

Dept테이블을 하나의 모델 클래스로 만든다.

//사용자정의 예외클래스

package com.exception;

public class RecordNotFoundException extends Exception {
    public RecordNotFoundException(String mesg) {
        super(mesg);
        }
}

예외클래스를 하나 만든다.

package com.service;

import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.SQLException;
import java.util.ArrayList;

import com.dao.OracleTxDAO;
import com.dto.Dept;
import com.exception.RecordNotFoundException;

public class OracleTxService {
    String driver = "oracle.jdbc.driver.OracleDriver";
    String url = "jdbc:oracle:thin:@localhost:1521:orcl";
    String userid = "scott";
    String passwd = "tiger";

    OracleTxDAO dao;

    public OracleTxService() {
        dao = new OracleTxDAO(); // 전체에서 써야하니까 멤버변수로 생성
        try {
            Class.forName(driver);
        } catch (ClassNotFoundException e) {
            e.printStackTrace();
        }
    }// end OracleDAO

    public ArrayList<Dept> select() throws SQLException {
        Connection con = null;
        ArrayList<Dept> list = null;
        try {
            con = DriverManager.getConnection(url, userid, passwd);
            list = dao.select(con);  //커넥션을 서비스에서 처리하기로 했으니까 처리된거 dao로 보내
        } finally {  //여기서는 예외안잡음. 자원반납만 처리.
            if (con != null)
                con.close();
        }
        return list;
    }// end select

    public void insert(Dept xx) {
        Connection con = null;
        try {
            con = DriverManager.getConnection(url, userid, passwd);
            dao.insert(con, xx);
        } catch (SQLException e) {
            System.out.println(e.getMessage());
        } finally {
            try {
                if (con != null)
                    con.close();
            } catch (SQLException e) {
                e.printStackTrace();
            }
        }
    }// end insert

    public void update(Dept xx2) throws RecordNotFoundException {
        Connection con = null;
        try {
            con = DriverManager.getConnection(url, userid, passwd);
            dao.update(con, xx2);
        } catch (SQLException e) {
            System.out.println(e.getMessage());
        } finally {
            try {
                if (con != null)
                    con.close();
            } catch (SQLException e) {
                e.printStackTrace();
            }
        }
    }// end update

    public void delete(int i) {
        Connection con = null;
        try {
            con = DriverManager.getConnection(url, userid, passwd);
            dao.delete(con, i);
        } catch (SQLException e) {
            System.out.println(e.getMessage());
            // e.printStackTrace();
        } finally {
            try {
                if (con != null)
                    con.close();
            } catch (SQLException e) {
                // TODO Auto-generated catch block
                e.printStackTrace();
            }
        }
    }// end deletes

}

서비스 클래스를 만든다. 서비스 클래스에서는 주로 커넥션 처리를 담당한다.

package com.dao;

import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.ArrayList;

import com.dto.Dept;
import com.exception.RecordNotFoundException;

public class OracleTxDAO {
    String driver = "oracle.jdbc.driver.OracleDriver";
    String url = "jdbc:oracle:thin:@localhost:1521:orcl";
    String userid = "scott";
    String passwd = "tiger";

    public void update(Connection con, Dept xx) throws RecordNotFoundException {
        PreparedStatement pstmt = null;
        try {
            String sql = "update dept set " + " dname=? , loc=?  where deptno=?";
            pstmt = con.prepareStatement(sql);
            pstmt.setString(1, xx.getDname());
            pstmt.setString(2, xx.getLoc());
            pstmt.setInt(3, xx.getDeptno());
            int num = pstmt.executeUpdate();
            System.out.println("실행된 레코드 갯수:" + num);
            if (num == 0) {
                throw new RecordNotFoundException(xx.getDeptno() + " 레코드없다.");
            }
        } catch (SQLException e) {
            e.printStackTrace();
        } finally {
            try {
                if (pstmt != null)
                    pstmt.close();
            } catch (SQLException e) {
                e.printStackTrace();
            }
        }
    }// end upate

    public void delete(Connection con, int i) {
        PreparedStatement pstmt = null;
        try {
            String sql = "delete from dept where deptno=?";
            pstmt = con.prepareStatement(sql);
            pstmt.setInt(1, i);
            int num = pstmt.executeUpdate();
            System.out.println("실행된 레코드 갯수:" + num);
        } catch (SQLException e) {
            e.printStackTrace();
        } finally {
            try {
                if (pstmt != null)
                    pstmt.close();
            } catch (SQLException e) {
                e.printStackTrace();
            }
        }
    }

    public void insert(Connection con, Dept xx) {
        PreparedStatement pstmt = null;
        try {
            String sql = "insert into dept (deptno,dname,loc)" + " values (?,?,? )";
            pstmt = con.prepareStatement(sql);
            pstmt.setInt(1, xx.getDeptno());
            pstmt.setString(2, xx.getDname());
            pstmt.setString(3, xx.getLoc());
            int num = pstmt.executeUpdate();
            System.out.println("실행된 레코드 갯수:" + num);
        } catch (SQLException e) {
            e.printStackTrace();
        } finally {
            try {
                if (pstmt != null)
                    pstmt.close();
            } catch (SQLException e) {
                e.printStackTrace();
            }
        }
    }// end insert

    public ArrayList<Dept> select(Connection con) throws SQLException {
        ArrayList<Dept> list = new ArrayList<Dept>();
        PreparedStatement pstmt = null;
        ResultSet rs = null;
        String sql = "select deptno x,dname,loc from dept ";
        pstmt = con.prepareStatement(sql);
        rs = pstmt.executeQuery();
        while (rs.next()) {
            int deptno = rs.getInt("x");
            String dname = rs.getString(2);
            String loc = rs.getString("loc");
            Dept dto = new Dept(deptno, dname, loc);
            list.add(dto);
        }
        if (rs != null)
            rs.close();
        if (pstmt != null)
            pstmt.close();

        return list;
    }// end select
}

DAO클래스에선 서비스 클래스에서 받아온 커넥션을 사용해 각 함수에 해당하는 SQL문을 작성하고 결과를 메인으로 리턴한다.

//main

import java.sql.SQLException;
import java.util.ArrayList;

import com.dto.Dept;
import com.exception.RecordNotFoundException;
import com.service.OracleTxService;

public class OracleTxMain {

    public static void main(String[] args) {
        OracleTxService service = new OracleTxService();

        //4번
        service.delete(99);

        // 3번
        Dept xx2 = new Dept(99, "영업", "서울");
        try {
            service.update(xx2);
        } catch (RecordNotFoundException e1) {
            System.out.println(e1.getMessage());
        }

        // 2번
        int deptno = 99;
        String dname = "개발";
        String loc = "제주";
        Dept xx = new Dept(deptno, dname, loc);
        service.insert(xx);
        
        // 1번
        try {
            ArrayList<Dept> list = service.select();
            for (Dept dept : list) {
                System.out.println(dept);
            }
        } catch (SQLException e) {
            e.printStackTrace();
        }
    }
}

메인에서 서비스 클래스의 객체를 생성하고 메소드를 호출하여 결과를 출력한다.

 

 

select()함수만 하나하나 뜯어보기

public OracleTxService() {
        dao = new OracleTxDAO();
        try {
            Class.forName(driver);
        } catch (ClassNotFoundException e) {
            e.printStackTrace();
        }// end OracleDAO
    }

서비스 클래스에서 생성자를 먼저 만든다. 서비스 클래스의 객체를 생성하면서 dao 클래스도 쓸 수 있게 객체생성 해주고 드라이버를 로딩한다.

public ArrayList<Dept> select() throws SQLException {
        //db연결
        Connection con = null;
        ArrayList<Dept> list = null;
        try {
            con = DriverManager.getConnection(url, userid, passwd);
            list = dao.select(con);
        } finally {
            if(con!=null)con.close();
        }
        return list;
    }// end select

select 함수 안에서 커넥션 연결해주고, 어레이리스트를 만든다. 리스트에 dao의 셀렉트함수 호출하면서 커넥션 연결을 담아서 보내주고 자원반납 해주고 리스트 리턴하고 함수를 종료한다.

이제 dao로 간다.

public ArrayList<Dept> select(Connection con)throws SQLException {
    ArrayList<Dept> list = new ArrayList<Dept>();
    PreparedStatement pstmt = null;
    ResultSet rs = null;
    String sql = "select deptno x, dname, loc from dept";
    pstmt = con.prepareStatement(sql);
    rs = pstmt.executeQuery();
    while(rs.next()) {
        int deptno = rs.getInt("x");
        String dname = rs.getString("dname");
        String loc = rs.getString(3);
        Dept dto = new Dept(deptno, dname, loc);
        list.add(dto);
    }if(rs != null) rs.close();
    if(pstmt!=null) pstmt.close();
    return list;
    }//end select
}

셀렉함수 인자로  con을 받는다. 먼저 어레이리스트 객체 생성해주고 con은 받아왔으니까 나머지 pstmt, rs를 세팅한다. 시퀄문 작성하고, pstmt로 쿼리문 실행하고, 실행된 걸 rs에 저장해주고, 반복문 돌려서 Dept클래스의 생성자를 호출해서 dto라는 변수에 넣어준다. 그렇게 불러온 정보를 리스트에 추가한 후 자원반납하고 리스트 리턴하고 함수를 종료한다. 이제 리턴된 리스트는 이 함수를 호출한 서비스 클래스로 갔다가 메인으로 가게된다.

이제 메인으로 간다.

OracleTxService service = new OracleTxService();

메인에서는 서비스 클래스만 사용하면 되니까 서비스 객체를 먼저 생성한다.

try {
            ArrayList<Dept> list = service.select();
            for(Dept dept : list) {
                System.out.println(dept);
            }
        } catch (SQLException e) {
            e.printStackTrace();
        }

이제 어레이리스트 타입으로 서비스의 셀렉트함수를 호출해주고 받아온 리스트를 반복문 돌려서 다 출력한다. 예외를 잡고 셀렉트 함수 실행은 끝난다. 메인에서는 함수 호출만 하고 자원반납은 서비스랑 dao에서 다 처리해줬기때문에 예외만 잡고 간단하게 끝난다.

 

 

 

😜 NEXT : JDBC 문제 또 연습해보기