LightPIT: src/main/java/de/uapcore/lightpit/dao/postgres/PGUserDao.java@57cfb94ab99f

     1 /*

     2  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.

     3  *

     4  * Copyright 2018 Mike Becker. All rights reserved.

     5  *

     6  * Redistribution and use in source and binary forms, with or without

     7  * modification, are permitted provided that the following conditions are met:

     8  *

     9  *   1. Redistributions of source code must retain the above copyright

    10  *      notice, this list of conditions and the following disclaimer.

    11  *

    12  *   2. Redistributions in binary form must reproduce the above copyright

    13  *      notice, this list of conditions and the following disclaimer in the

    14  *      documentation and/or other materials provided with the distribution.

    15  *

    16  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"

    17  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE

    18  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE

    19  * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE

    20  * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR

    21  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF

    22  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS

    23  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN

    24  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)

    25  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE

    26  * POSSIBILITY OF SUCH DAMAGE.

    27  *

    28  */

    29 package de.uapcore.lightpit.dao.postgres;

    31 import de.uapcore.lightpit.dao.GenericDao;

    32 import de.uapcore.lightpit.dao.UserDao;

    33 import de.uapcore.lightpit.entities.User;

    35 import java.sql.Connection;

    36 import java.sql.PreparedStatement;

    37 import java.sql.ResultSet;

    38 import java.sql.SQLException;

    39 import java.util.ArrayList;

    40 import java.util.List;

    41 import java.util.Objects;

    43 import static de.uapcore.lightpit.dao.Functions.setStringOrNull;

    45 public final class PGUserDao implements UserDao, GenericDao<User> {

    47     public static final String[] COLUMNS = {

    48             "id", "username", "lastname", "givenname", "mail"

    49     };

    51     private final PreparedStatement insert, update, list, find;

    53     public PGUserDao(Connection connection) throws SQLException {

    54         list = connection.prepareStatement(

    55                 "select userid, username, lastname, givenname, mail " +

    56                         "from lpit_user where userid >= 0 " +

    57                         "order by username");

    58         find = connection.prepareStatement(

    59                 "select userid, username, lastname, givenname, mail " +

    60                         "from lpit_user where userid = ? ");

    62         insert = connection.prepareStatement("insert into lpit_user (username, lastname, givenname, mail) values (?, ?, ?, ?)");

    63         update = connection.prepareStatement("update lpit_user set lastname = ?, givenname = ?, mail = ? where userid = ?");

    64     }

    66     public User mapColumns(ResultSet result) throws SQLException {

    67         final int id = result.getInt("userid");

    68         if (id == 0) return null;

    69         final var user = new User(id);

    70         user.setUsername(result.getString("username"));

    71         user.setGivenname(result.getString("givenname"));

    72         user.setLastname(result.getString("lastname"));

    73         user.setMail(result.getString("mail"));

    74         return user;

    75     }

    77     @Override

    78     public void save(User instance) throws SQLException {

    79         Objects.requireNonNull(instance.getUsername());

    80         insert.setString(1, instance.getUsername());

    81         setStringOrNull(insert, 2, instance.getLastname());

    82         setStringOrNull(insert, 3, instance.getGivenname());

    83         setStringOrNull(insert, 4, instance.getMail());

    84         insert.executeUpdate();

    85     }

    87     @Override

    88     public boolean update(User instance) throws SQLException {

    89         setStringOrNull(update, 1, instance.getLastname());

    90         setStringOrNull(update, 2, instance.getGivenname());

    91         setStringOrNull(update, 3, instance.getMail());

    92         update.setInt(4, instance.getUserID());

    93         return update.executeUpdate() > 0;

    94     }

    96     @Override

    97     public List<User> list() throws SQLException {

    98         List<User> users = new ArrayList<>();

    99         try (var result = list.executeQuery()) {

   100             while (result.next()) {

   101                 users.add(mapColumns(result));

   102             }

   103         }

   104         return users;

   105     }

   107     @Override

   108     public User find(int id) throws SQLException {

   109         find.setInt(1, id);

   110         try (var result = find.executeQuery()) {

   111             if (result.next()) {

   112                 return mapColumns(result);

   113             } else {

   114                 return null;

   115             }

   116         }

   117     }

   118 }

Mercurial > hg > LightPIT / file revision