src/main/java/com/carpooling/dao/csv/CsvBookingDao.java - master - Buka/CarPooling

src/main/java/com/carpooling/dao/csv/CsvBookingDao.java
93 строки4 KB
BUka228
После рефакторинга пакетов
18 янв 2025, 17:08
18 янв 2025, 17:08a4d4c33
package com.carpooling.dao.csv;

import com.carpooling.dao.base.BookingDao;
import com.carpooling.entities.record.BookingRecord;
import com.carpooling.exceptions.dao.DataAccessException;
import com.opencsv.exceptions.CsvDataTypeMismatchException;
import com.opencsv.exceptions.CsvRequiredFieldEmptyException;
import lombok.extern.slf4j.Slf4j;
import org.jetbrains.annotations.NotNull;

import java.io.IOException;
import java.util.List;
import java.util.Optional;

import static com.carpooling.constants.ErrorMessages.BOOKING_CREATION_ERROR;
import static com.carpooling.constants.ErrorMessages.BOOKING_UPDATE_ERROR;
import static com.carpooling.constants.ErrorMessages.*;
import static com.carpooling.constants.LogMessages.*;

@Slf4j
public class CsvBookingDao extends AbstractCsvDao<BookingRecord> implements BookingDao {

    public CsvBookingDao(String filePath) {
        super(BookingRecord.class, filePath);
    }

    @Override
    public String createBooking(@NotNull BookingRecord bookingRecord) throws DataAccessException {
        log.info(CREATE_BOOKING_START, bookingRecord.getTripId(), bookingRecord.getUserId());

        // Генерация UUID для ID
        String bookingId = generateId();
        bookingRecord.setId(bookingId);

        try {
            List<BookingRecord> bookings = readAll();
            bookings.add(bookingRecord);
            writeAll(bookings);

            log.info(CREATE_BOOKING_SUCCESS, bookingId);
            return bookingId;
        } catch (IOException | CsvDataTypeMismatchException | CsvRequiredFieldEmptyException e) {
            log.error(ERROR_CREATE_BOOKING, bookingRecord.getTripId(), bookingRecord.getUserId(), e);
            throw new DataAccessException(BOOKING_CREATION_ERROR, e);
        }
    }

    @Override
    public Optional<BookingRecord> getBookingById(String id) throws DataAccessException {
        log.info(GET_BOOKING_START, id);
        try {
            return findById(record -> record.getId().equals(id));
        } catch (IOException e) {
            log.error(ERROR_GET_BOOKING, id, e);
            throw new DataAccessException(String.format(BOOKING_NOT_FOUND_ERROR, id), e);
        }
    }

    @Override
    public void updateBooking(@NotNull BookingRecord bookingRecord) throws DataAccessException {
        try {
            List<BookingRecord> bookings = readAll();
            boolean found = false;
            for (int i = 0; i < bookings.size(); i++) {
                if (bookings.get(i).getId().equals(bookingRecord.getId())) {
                    bookings.set(i, bookingRecord);
                    found = true;
                    break;
                }
            }
            if (!found) {
                log.warn(WARN_BOOKING_NOT_FOUND, bookingRecord.getId());
                throw new DataAccessException(String.format(BOOKING_NOT_FOUND_ERROR, bookingRecord.getId()));
            }
            writeAll(bookings);
            log.info(UPDATE_BOOKING_SUCCESS, bookingRecord.getId());
        } catch (IOException | CsvDataTypeMismatchException | CsvRequiredFieldEmptyException e) {
            log.error(ERROR_UPDATE_BOOKING, bookingRecord.getId(), e);
            throw new DataAccessException(BOOKING_UPDATE_ERROR, e);
        }
    }

    @Override
    public void deleteBooking(String id) throws DataAccessException {
        try {
            deleteById(record -> record.getId().equals(id));
            log.info(DELETE_BOOKING_SUCCESS, id);
        } catch (IOException | CsvDataTypeMismatchException | CsvRequiredFieldEmptyException e) {
            log.error(ERROR_DELETE_BOOKING, id, e);
            throw new DataAccessException(BOOKING_DELETE_ERROR, e);
        }
    }
}