자바로 만든 SQLPLUS
D19_SQLPLUS.java
//---------------------------------------------------------------------------------------------------
import java.sql.Statement;
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.sql.Connection;
import java.sql.ResultSet;
import java.sql.ResultSetMetaData;
import java.sql.SQLException;
/*
2. SQLPLUS 프로그램을 제작해보자
(기능:테이블생성, 삭제, 프로시져 수행, select, delete, update....)
*/
public class D19_SQLPLUS {
static Statement stmt;
static Connection con;
static ResultSet rs;
static ResultSetMetaData rsmd;
static BufferedReader br;
static boolean exitCheck;
public static void main(String[] args) {
String id=null;
String pwd=null;
int chanceToLogin=0;
while(true){
br = new BufferedReader(
new InputStreamReader(System.in));
//사용자 로그인
try{
//3번 이상 로그인에 실패햐면 프로그램 종료
if(chanceToLogin==3){
System.out.println("프로그램을 종료합니다.");
break;
}
System.out.print("사용자명 입력 : ");
id = br.readLine();
System.out.print("암호 입력 : ");
pwd = br.readLine();
//입력된 id와 password로 connection 객체 생성
DBConnetion dbc = new DBConnetion(id, pwd);
con = dbc.getCon();
//sql쿼리를 입력받는 부분을 처리하는 input()메소드 호출
input();
}catch(IOException ioe){
System.out.println(ioe.getMessage());
}catch(SQLException se){
//DBConnetion()객체 생성시 ID나 PWD가 틀리면 while루프 처음으로 이동
System.out.print(se.getMessage());
System.out.println("아이디 또는 비밀번호가 틀립니다.!");
System.out.println();
chanceToLogin++;
continue;
}catch(Exception e){
System.out.println(e.getMessage());
}finally{
try{
if(rs != null) rs.close();
if(stmt != null) stmt.close();
if(con != null) con.close();
if(exitCheck){
System.exit(0);
}
}catch(SQLException se){
System.out.println(se.getMessage());
}
}
}
}
public static void input()throws IOException, Exception{
br = new BufferedReader(
new InputStreamReader(System.in));
String inQuery=""; //입력받은 쿼리
String[] splitedQuery=null;//공백을 기준으로 분할한 쿼리
String query = null; //실제로 실행할 쿼리
String tempQuery = null;//Enter가 입력되기까지의 입력값
String startString = null;//쿼리의 첫번째 문자열
int lineNum=2;
boolean flag=false;
System.out.print("SQL> ");
while(true){
tempQuery = br.readLine();
inQuery += tempQuery;
if(inQuery.equals("exit")){
//System.out.println("프로그램을 종료합니다.");
exitCheck = true;
throw new Exception("프로그램을 종료합니다.");
//System.exit(0);
}
//입력값 없이 enter가 들어오면 SQL> 프롬프트로 빠져나옴
if(tempQuery.equals("")){
lineNum=2;
inQuery = "";
System.out.print("SQL> ");
continue;
}
//공백을 기준으로 입력받은 쿼리문자열을 자른다.
splitedQuery = inQuery.split(" ");
for(int i=0; i<splitedQuery.length; i++){
if(splitedQuery[i].trim().equals("begin")){
procedure(inQuery, lineNum);
flag = true;
break;
}
}
//프로시저의 경우 바로 위의 for에서 처리 했으므로while의 처음으로 이동
if(flag){
flag=false;
lineNum=2;
inQuery = "";
System.out.print("SQL> ");
continue;
}
if(!inQuery.endsWith(";")){
System.out.print(" "+lineNum+" ");
inQuery += " ";
lineNum++;
continue;
}
//공백을 기준으로 입력받은 쿼리문자열을 자른다.
splitedQuery = inQuery.split(" ");
//첫문자로 " "(공백)을 넣어서 시작문자는 인덱스[1]이 된다.
startString = splitedQuery[0];
//끝문자인 ';'를 공백으로 replace
query = inQuery.replace(';',' ');
if(startString.equals("select")){
exeQuery(query);
}else if(startString.equals("update") || startString.equals("delete") ||
startString.equals("insert")){
exeUpdate(query);
}else{
exe(query);
}
//다음 쿼리 입력을 위한 변수 셋팅
lineNum=2;
inQuery = "";
System.out.print("SQL> ");
}
}
public static void exeQuery(String query){
int count=0;//수행한 열의 수를 세는 변수
try{
stmt = con.createStatement();
rs = stmt.executeQuery(query);
rsmd = rs.getMetaData();
//컬럼의 수를 알아온다.
int colNum = rsmd.getColumnCount();
//컬럼의 수 만큼 루프를 돌면서 컬럼명을 출력한다.
for(int i=1; i<=colNum; i++){
String colName = rsmd.getColumnName(i);
System.out.print(colName +"\t");
}
System.out.println();
System.out.println("=============================");
//컬럼의 내용을 각 컬럼의 변수형에 따라 얻어오고, 이를 출력한다.
while(rs.next()){
for(int i=1; i<=colNum; i++){
if(rsmd.getColumnTypeName(i).equals("NUMBER")){
System.out.print(rs.getInt(i)+"\t");
}else if(rsmd.getColumnTypeName(i).equals("VARCHAR2")){
System.out.print(rs.getString(i)+"\t");
}else{
System.out.print(rs.getObject(i)+"\t");
}
}
System.out.println();
count++;
}
System.out.println(count+"개의 행이 선택 되었습니다.");
}catch(SQLException se){
System.out.println(se.getMessage());
}
}
public static void exeUpdate(String query){
try{
stmt = con.createStatement();
int num = stmt.executeUpdate(query);
System.out.println(num+"개의 행이 수행 되었습니다.");
}catch(SQLException se){
System.out.println(se.getMessage());
}
}
public static void exe(String query){
try{
stmt = con.createStatement();
stmt.execute(query);
System.out.println("SQL 구문이 수행되었습니다.");
}catch(SQLException se){
System.out.println(se.getMessage());
}
}
public static void procedure(String query, int lineNum)throws IOException{
br = new BufferedReader(
new InputStreamReader(System.in));
String tempQuery =null;
do{
System.out.print(" "+lineNum+" ");
tempQuery = br.readLine();
if(tempQuery.trim().equals("/")){
try{
query += tempQuery+"\n";
stmt = con.createStatement();
stmt.execute(query);
System.out.println("프로시저가 생성되었습니다.");
break;
}catch(SQLException se){
System.out.println(se.getMessage());
break;
}
}
lineNum++;
query += tempQuery;
}while(true);
}
}
//---------------------------------------------------------------------------------------------------
DBConnection.java
//---------------------------------------------------------------------------------------------------
import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.SQLException;
class DBConnetion{
private String id;
private String pwd;
private String url="jdbc:oracle:thin:@localhost:1521:orcl";
private String driver="oracle.jdbc.driver.OracleDriver";
private Connection con;
//SQLException은 호출부에서 처리하도록 throws 한다.
public DBConnetion(String id, String pwd) throws SQLException{
this.id = id;
this.pwd = pwd;
try{
Class.forName(driver);
con = DriverManager.getConnection(url, id, pwd);
System.out.println(id+"님 안녕하세요~^^");
}catch(ClassNotFoundException ce){
System.out.println(ce.getMessage());
}
}
public Connection getCon(){
return con;
}
}
//---------------------------------------------------------------------------------------------------
출처: https://invincure.tistory.com/entry/자바로-만든-SQLPLUS?category=299957 [Do you want it? Come get it.]
