java-design-patterns/transaction-script/src/main/java/com/ashishtrivedi16/transactionscript/TransactionScriptApp.java

package com.ashishtrivedi16.transactionscript;

import com.ashishtrivedi16.transactionscript.db.CustomException;
import com.ashishtrivedi16.transactionscript.db.HotelDAOImpl;
import com.ashishtrivedi16.transactionscript.db.RoomSchemaSql;

import java.sql.SQLException;
import java.util.List;
import javax.sql.DataSource;

import org.h2.jdbc.JdbcSQLException;
import org.h2.jdbcx.JdbcDataSource;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

public class TransactionScriptApp {

    private static final String H2_DB_URL = "jdbc:h2:~/test";
    private static final Logger LOGGER = LoggerFactory.getLogger(TransactionScriptApp.class);
    private static final String ALL_ROOMS = "customerDao.getAllRooms(): ";

    public static void main(String[] args) throws Exception {

        /*
            TODO
                -> Create in memory database and some sample tables for demo
        */
        final var dataSource = createDataSource();
        createSchema(dataSource);
        final var DAO = new HotelDAOImpl(dataSource);

        addRooms(DAO);
        try (var customerStream = DAO.getAll()) {
            customerStream.forEach((customer) -> LOGGER.info(customer.toString()));
        }

        Hotel hotel = new Hotel(DAO);

        hotel.bookRoom(1);
        hotel.bookRoom(2);
        hotel.bookRoom(3);
        hotel.bookRoom(4);
        hotel.bookRoom(5);
        hotel.bookRoom(6);

        hotel.cancelRoomBooking(3);
        hotel.cancelRoomBooking(4);

        deleteSchema(dataSource);

    }

    private static void deleteSchema(DataSource dataSource) throws SQLException {
        try (var connection = dataSource.getConnection();
             var statement = connection.createStatement()) {
            statement.execute(RoomSchemaSql.DELETE_SCHEMA_SQL);
        }
    }

    private static void createSchema(DataSource dataSource) throws Exception {
        try (var connection = dataSource.getConnection();
             var statement = connection.createStatement()) {
            statement.execute(RoomSchemaSql.CREATE_SCHEMA_SQL);
        } catch (JdbcSQLException e) {
            throw new CustomException(e.getMessage(), e);
        }
    }

    public static DataSource createDataSource() {
        JdbcDataSource dataSource = new JdbcDataSource();
        dataSource.setUrl(H2_DB_URL);
        return dataSource;
    }

    private static void addRooms(HotelDAOImpl hotelDAO) throws Exception {
        for (var room : generateSampleRooms()) {
            hotelDAO.add(room);
        }
    }

    public static List<Room> generateSampleRooms() {
        final var room1 = new Room(1, "Single", 50, false);
        final var room2 = new Room(2, "Double", 80, false);
        final var room3 = new Room(3, "Queen", 120, false);
        final var room4 = new Room(4, "King", 150, false);
        final var room5 = new Room(5, "Single", 50, false);
        final var room6 = new Room(6, "Double", 80, false);
        return List.of(room1, room2, room3, room4, room5, room6);
    }
}