universe@34: /*
universe@34:  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
universe@34:  *
universe@34:  * Copyright 2018 Mike Becker. All rights reserved.
universe@34:  *
universe@34:  * Redistribution and use in source and binary forms, with or without
universe@34:  * modification, are permitted provided that the following conditions are met:
universe@34:  *
universe@34:  *   1. Redistributions of source code must retain the above copyright
universe@34:  *      notice, this list of conditions and the following disclaimer.
universe@34:  *
universe@34:  *   2. Redistributions in binary form must reproduce the above copyright
universe@34:  *      notice, this list of conditions and the following disclaimer in the
universe@34:  *      documentation and/or other materials provided with the distribution.
universe@34:  *
universe@34:  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
universe@34:  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
universe@34:  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
universe@34:  * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
universe@34:  * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
universe@34:  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
universe@34:  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
universe@34:  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
universe@34:  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
universe@34:  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
universe@34:  * POSSIBILITY OF SUCH DAMAGE.
universe@34:  *
universe@34:  */
universe@34: package de.uapcore.lightpit.dao.postgres;
universe@34: 
universe@34: import de.uapcore.lightpit.dao.UserDao;
universe@34: import de.uapcore.lightpit.entities.User;
universe@34: 
universe@34: import java.sql.Connection;
universe@34: import java.sql.PreparedStatement;
universe@34: import java.sql.ResultSet;
universe@34: import java.sql.SQLException;
universe@47: import java.util.ArrayList;
universe@47: import java.util.List;
universe@38: import java.util.Objects;
universe@34: 
universe@47: import static de.uapcore.lightpit.dao.Functions.setStringOrNull;
universe@34: 
universe@59: public final class PGUserDao implements UserDao {
universe@47: 
universe@47:     private final PreparedStatement insert, update, list, find;
universe@38: 
universe@38:     public PGUserDao(Connection connection) throws SQLException {
universe@47:         list = connection.prepareStatement(
universe@47:                 "select userid, username, lastname, givenname, mail " +
universe@47:                         "from lpit_user where userid >= 0 " +
universe@47:                         "order by username");
universe@47:         find = connection.prepareStatement(
universe@47:                 "select userid, username, lastname, givenname, mail " +
universe@47:                         "from lpit_user where userid = ? ");
universe@38: 
universe@38:         insert = connection.prepareStatement("insert into lpit_user (username, lastname, givenname, mail) values (?, ?, ?, ?)");
universe@38:         update = connection.prepareStatement("update lpit_user set lastname = ?, givenname = ?, mail = ? where userid = ?");
universe@38:     }
universe@38: 
universe@47:     public User mapColumns(ResultSet result) throws SQLException {
universe@47:         final int id = result.getInt("userid");
universe@47:         if (id == 0) return null;
universe@47:         final var user = new User(id);
universe@47:         user.setUsername(result.getString("username"));
universe@47:         user.setGivenname(result.getString("givenname"));
universe@47:         user.setLastname(result.getString("lastname"));
universe@47:         user.setMail(result.getString("mail"));
universe@34:         return user;
universe@34:     }
universe@34: 
universe@34:     @Override
universe@38:     public void save(User instance) throws SQLException {
universe@38:         Objects.requireNonNull(instance.getUsername());
universe@38:         insert.setString(1, instance.getUsername());
universe@38:         setStringOrNull(insert, 2, instance.getLastname());
universe@38:         setStringOrNull(insert, 3, instance.getGivenname());
universe@38:         setStringOrNull(insert, 4, instance.getMail());
universe@38:         insert.executeUpdate();
universe@38:     }
universe@38: 
universe@38:     @Override
universe@38:     public boolean update(User instance) throws SQLException {
universe@38:         setStringOrNull(update, 1, instance.getLastname());
universe@38:         setStringOrNull(update, 2, instance.getGivenname());
universe@38:         setStringOrNull(update, 3, instance.getMail());
universe@51:         update.setInt(4, instance.getId());
universe@38:         return update.executeUpdate() > 0;
universe@34:     }
universe@47: 
universe@47:     @Override
universe@47:     public List<User> list() throws SQLException {
universe@47:         List<User> users = new ArrayList<>();
universe@47:         try (var result = list.executeQuery()) {
universe@47:             while (result.next()) {
universe@47:                 users.add(mapColumns(result));
universe@47:             }
universe@47:         }
universe@47:         return users;
universe@47:     }
universe@47: 
universe@47:     @Override
universe@47:     public User find(int id) throws SQLException {
universe@47:         find.setInt(1, id);
universe@47:         try (var result = find.executeQuery()) {
universe@47:             if (result.next()) {
universe@47:                 return mapColumns(result);
universe@47:             } else {
universe@47:                 return null;
universe@47:             }
universe@47:         }
universe@47:     }
universe@34: }