728x90
반응형
SMALL
package miniprj.model;
import lombok.Data;
@Data
public class Book {
// 예약자성함
String name;
// 체크인날짜
String checkIn;
// 체크아웃 날짜
String checkOut;
// 예약한 객실(standard, superior, suite 셋 중 하나)
String room;
// 예약 비밀번호
String password;
}package miniprj.model;
import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.ArrayList;
public class BookDAO implements IBookDAO{
@Override
public int deleteBook(String name, String password) {
int rowCount=0;
String sql="delete from book where cust_name=? and pwd=?";
Connection con=null;
try {
con=BookDataSource.getConnection();
PreparedStatement stmt=con.prepareStatement(sql);
stmt.setString(1, name);
stmt.setString(2, password);
rowCount=stmt.executeUpdate();
} catch (Exception e) {
throw new RuntimeException(e);
} finally {
BookDataSource.closeConnection(con);
}
return rowCount;
}
@Override
public int insertBook(Book book) {
int rowCount=0;
String sql="insert into book "
+ "(cust_name, checkin, checkout, room, pwd) values(?,?,?,?,?)";
Connection con=null;
try {
con=BookDataSource.getConnection();
PreparedStatement stmt=con.prepareStatement(sql);
stmt.setString(1, book.getName());
stmt.setString(2, book.getCheckIn());
stmt.setString(3, book.getCheckOut());
stmt.setString(4, book.getRoom());
stmt.setString(5, book.getPassword());
rowCount=stmt.executeUpdate();
} catch (Exception e) {
throw new RuntimeException(e);
} finally {
BookDataSource.closeConnection(con);
}
return rowCount;
}
@Override
public int updateBook(Book book) {
String sql="update book set checkin=?, checkout=?, room=? "
+ "where cust_name=? and pwd=?";
int rowCount=0;
Connection con=null;
System.out.println(book);
try {
con=BookDataSource.getConnection();
PreparedStatement stmt=con.prepareStatement(sql);
stmt.setString(1, book.getCheckIn());
stmt.setString(2, book.getCheckOut());
stmt.setString(3, book.getRoom());
stmt.setString(4, book.getName());
stmt.setString(5, book.getPassword());
rowCount=stmt.executeUpdate();
} catch (Exception e) {
throw new RuntimeException();
}finally {
BookDataSource.closeConnection(con);
}
return rowCount;
}
public boolean login(String name, String password) {
String sql="select * from book where cust_name=? and pwd=?";
int rowCount=0;
Connection con=null;
try {
con=BookDataSource.getConnection();
PreparedStatement stmt=con.prepareStatement(sql);
stmt.setString(1, name);
stmt.setString(2, password);
ResultSet rs=stmt.executeQuery();
if (rs.next()) {
return true;
}else{
System.out.println("일치하는 정보가 없습니다.");
return false;
}
}catch (Exception e) {
return false;
}
}
@Override
public int getBookCount() {
String sql="select count(*) from book";
int rowCount=0;
Connection con=null;
try {
con=BookDataSource.getConnection();
PreparedStatement stmt=con.prepareStatement(sql);
ResultSet rs=stmt.executeQuery();
rs.next();
rowCount=rs.getInt(1);
} catch (Exception e) {
throw new RuntimeException();
}finally {
BookDataSource.closeConnection(con);
}
return rowCount;
}
@Override
public Book getOne(String name, String password) {
String sql="select * from book where cust_name=? and pwd=?";
Connection con=null;
Book book=new Book();
try {
con=BookDataSource.getConnection();
PreparedStatement stmt=con.prepareStatement(sql);
stmt.setString(1, name);
stmt.setString(2, password);
ResultSet rs=stmt.executeQuery();
while(rs.next()) {
book.setName(rs.getString("cust_name"));
book.setCheckIn(rs.getString("checkin"));
book.setCheckOut(rs.getString("checkout"));
book.setRoom(rs.getString("room"));
book.setPassword(rs.getString("pwd"));
}
} catch (Exception e) {
throw new RuntimeException();
}finally {
BookDataSource.closeConnection(con);
}
return book;
}
@Override
public ArrayList<Book> getAllBooks() {
// TODO Auto-generated method stub
ArrayList<Book> list = new ArrayList<>();
String sql = "select * from book";
Connection con = null;
try {
con = BookDataSource.getConnection();
PreparedStatement stmt = con.prepareStatement(sql);
ResultSet rs = stmt.executeQuery();
while(rs.next()) {
Book book = new Book();
book.setName(rs.getString("cust_name"));
book.setRoom(rs.getString("room"));
book.setPassword(rs.getString("pwd"));
book.setCheckIn(rs.getString("checkin"));
book.setCheckOut(rs.getString("checkout"));
list.add(book);
}
}catch(SQLException e) {
throw new RuntimeException(e);
} finally {
BookDataSource.closeConnection(con);
}
return list;
}
}package miniprj.model;
import java.sql.Connection;
import java.sql.SQLException;
import org.apache.commons.dbcp2.BasicDataSource;
public class BookDataSource {
private static final String DRIVER = "oracle.jdbc.OracleDriver";
private static final String URL = "jdbc:oracle:thin:@localhost:1521:xe";
private static final String USERNAME = "hr";
private static final String PASSWORD = "hr";
private static BasicDataSource dataSource;
static {
try {
dataSource = new BasicDataSource();
dataSource.setDriverClassName(DRIVER);
dataSource.setUrl(URL);
dataSource.setUsername(USERNAME);
dataSource.setPassword(PASSWORD);
dataSource.setInitialSize(10);
dataSource.setMaxTotal(10);
System.out.println("DataSource created");
} catch (Exception e) {
System.out.println(e.getMessage());
}
}
public static Connection getConnection() throws SQLException {
return dataSource.getConnection();
}
public static void closeConnection(Connection con) {
if(con!=null) {
try {
con.close();
}catch(Exception e) {}
}
}
}package miniprj.model;
import java.util.ArrayList;
public interface IBookDAO {
public int deleteBook(String name, String password);
public int insertBook(Book book);
public int updateBook(Book book);
public int getBookCount();
public Book getOne(String name, String password);
public ArrayList<Book> getAllBooks();
}package miniprj;
import java.util.List;
import java.util.Scanner;
import miniprj.model.Book;
import miniprj.model.BookDAO;
public class BookWin {
static boolean isCustomer;
static boolean isAdmin;
private static Scanner scan = new Scanner(System.in);
static BookDAO bookDao = new BookDAO();
static String name = null;
static String pwd = null;
// 호텔 예약 리스트
public static void list(Book list) {
System.out.printf("%-6s%-12s%-16s%-40s",list.getName(), list.getCheckIn(), list.getCheckOut(), list.getRoom());
System.out.println();
}
public static void mainMenu() {
System.out.println();
System.out.println("--------------------------------------------------------");
System.out.println("메인메뉴: 1.예약 | 2.조회 | 3.수정 | 4.삭제 | 5.로그아웃 | 6.종료");
System.out.println("메뉴선택: ");
String menuNo = scan.nextLine();
System.out.println();
switch(menuNo) {
case "1" -> {
Book book = new Book();
System.out.print("예약자: ");
book.setName(scan.nextLine());
System.out.print("체크인 날짜: ");
book.setCheckIn(scan.nextLine());
System.out.print("체크아웃 날짜: ");
book.setCheckOut(scan.nextLine());
System.out.print("객실: ");
book.setRoom(scan.nextLine());
System.out.println("비밀번호: ");
book.setPassword(scan.nextLine());
bookDao.insertBook(book);
break;
}
case "2" -> {
System.out.println("[호텔 예약관리 시스템]");
System.out.println("--------------------------------------------------------");
System.out.printf("%-6s%-12s%-16s%-40s", "성함","체크인 날짜","체크아웃 날짜","객실");
System.out.println("--------------------------------------------------------");
if (isAdmin) {
List<Book> list = bookDao.getAllBooks();
// System.out.println(list);
for(int i=0;i<list.size();i++) {
list(list.get(i));
}
} else if(isCustomer) {
Book book = bookDao.getOne(name, pwd);
list(book);
}
break;
}
case "3" -> {
Book book = bookDao.getOne(name, pwd);
System.out.print("체크인 날짜: ");
book.setCheckIn(scan.nextLine());
System.out.print("체크아웃 날짜: ");
book.setCheckOut(scan.nextLine());
System.out.print("객실: ");
book.setRoom(scan.nextLine());
bookDao.updateBook(book);
break;
}
case "4" -> {
Book onesBook = bookDao.getOne(name, pwd);
if(!onesBook.getName().equals(name)) {
System.out.println("예약건이 없습니다.");
break;
}
int rowCount = bookDao.deleteBook(name, pwd);
System.out.println(rowCount + "건의 에약이 취소되었습니다.");
break;
}
case "5" -> {
isAdmin = false;
isCustomer = false;
name = null;
pwd = null;
loginMenu();
break;
}
case "6" -> {
System.out.println("프로그램이 종료되었습니다.");
System.exit(0);
}
}
}
public static boolean loginMenu() {
System.out.println();
System.out.println("--------------------------------------------------------");
System.out.println("성함을 입력하세요: ");
name = scan.nextLine();
System.out.println("비밀번호를 입력하세요: ");
pwd = scan.nextLine();
if(name.equals("admin") && pwd.equals("admin")) {
isAdmin = true;
return true;
} else if(bookDao.login(name, pwd)){
isCustomer = true;
mainMenu();
return true;
} else {
System.out.println("회원정보가 일치하지 않습니다.");
return false;
}
}
public static void main(String[] args) {
while(true) {
if(loginMenu()) {
while(isAdmin || isCustomer) {
mainMenu();
}
}
}
}
}728x90
반응형
LIST
'IT > JAVA' 카테고리의 다른 글
| [JAVA] #19 데이터베이스 입출력(게시판 프로그램) (1) | 2023.05.02 |
|---|---|
| [JAVA] #18 입출력 스트림 (0) | 2023.05.01 |
| [JAVA] #17 스트림 요소 처리 (0) | 2023.05.01 |
| [JAVA] #16 람다식 (0) | 2023.04.27 |
| [JAVA] #15 컬렉션 자료구조 (0) | 2023.04.27 |
