Fri, 23 Oct 2020 20:34:57 +0200
migrate entities package
universe@34 | 1 | /* |
universe@34 | 2 | * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER. |
universe@34 | 3 | * |
universe@34 | 4 | * Copyright 2018 Mike Becker. All rights reserved. |
universe@34 | 5 | * |
universe@34 | 6 | * Redistribution and use in source and binary forms, with or without |
universe@34 | 7 | * modification, are permitted provided that the following conditions are met: |
universe@34 | 8 | * |
universe@34 | 9 | * 1. Redistributions of source code must retain the above copyright |
universe@34 | 10 | * notice, this list of conditions and the following disclaimer. |
universe@34 | 11 | * |
universe@34 | 12 | * 2. Redistributions in binary form must reproduce the above copyright |
universe@34 | 13 | * notice, this list of conditions and the following disclaimer in the |
universe@34 | 14 | * documentation and/or other materials provided with the distribution. |
universe@34 | 15 | * |
universe@34 | 16 | * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" |
universe@34 | 17 | * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE |
universe@34 | 18 | * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE |
universe@34 | 19 | * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE |
universe@34 | 20 | * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR |
universe@34 | 21 | * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF |
universe@34 | 22 | * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS |
universe@34 | 23 | * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN |
universe@34 | 24 | * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) |
universe@34 | 25 | * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE |
universe@34 | 26 | * POSSIBILITY OF SUCH DAMAGE. |
universe@34 | 27 | * |
universe@34 | 28 | */ |
universe@34 | 29 | package de.uapcore.lightpit.dao.postgres; |
universe@34 | 30 | |
universe@34 | 31 | import de.uapcore.lightpit.dao.UserDao; |
universe@34 | 32 | import de.uapcore.lightpit.entities.User; |
universe@34 | 33 | |
universe@34 | 34 | import java.sql.Connection; |
universe@34 | 35 | import java.sql.PreparedStatement; |
universe@34 | 36 | import java.sql.ResultSet; |
universe@34 | 37 | import java.sql.SQLException; |
universe@47 | 38 | import java.util.ArrayList; |
universe@47 | 39 | import java.util.List; |
universe@38 | 40 | import java.util.Objects; |
universe@124 | 41 | import java.util.Optional; |
universe@34 | 42 | |
universe@150 | 43 | import static de.uapcore.lightpit.dao.Functions.getSafeString; |
universe@47 | 44 | import static de.uapcore.lightpit.dao.Functions.setStringOrNull; |
universe@34 | 45 | |
universe@59 | 46 | public final class PGUserDao implements UserDao { |
universe@47 | 47 | |
universe@124 | 48 | private final PreparedStatement insert, update, list, find, findByUsername; |
universe@38 | 49 | |
universe@38 | 50 | public PGUserDao(Connection connection) throws SQLException { |
universe@47 | 51 | list = connection.prepareStatement( |
universe@47 | 52 | "select userid, username, lastname, givenname, mail " + |
universe@47 | 53 | "from lpit_user where userid >= 0 " + |
universe@47 | 54 | "order by username"); |
universe@47 | 55 | find = connection.prepareStatement( |
universe@47 | 56 | "select userid, username, lastname, givenname, mail " + |
universe@47 | 57 | "from lpit_user where userid = ? "); |
universe@38 | 58 | |
universe@124 | 59 | findByUsername = connection.prepareStatement( |
universe@124 | 60 | "select userid, username, lastname, givenname, mail " + |
universe@124 | 61 | "from lpit_user where lower(username) = lower(?) "); |
universe@124 | 62 | |
universe@38 | 63 | insert = connection.prepareStatement("insert into lpit_user (username, lastname, givenname, mail) values (?, ?, ?, ?)"); |
universe@38 | 64 | update = connection.prepareStatement("update lpit_user set lastname = ?, givenname = ?, mail = ? where userid = ?"); |
universe@38 | 65 | } |
universe@38 | 66 | |
universe@128 | 67 | static User mapColumns(ResultSet result) throws SQLException { |
universe@47 | 68 | final int id = result.getInt("userid"); |
universe@47 | 69 | if (id == 0) return null; |
universe@47 | 70 | final var user = new User(id); |
universe@47 | 71 | user.setUsername(result.getString("username")); |
universe@150 | 72 | user.setGivenname(getSafeString(result, "givenname")); |
universe@150 | 73 | user.setLastname(getSafeString(result, "lastname")); |
universe@150 | 74 | user.setMail(getSafeString(result, "mail")); |
universe@34 | 75 | return user; |
universe@34 | 76 | } |
universe@34 | 77 | |
universe@34 | 78 | @Override |
universe@38 | 79 | public void save(User instance) throws SQLException { |
universe@38 | 80 | Objects.requireNonNull(instance.getUsername()); |
universe@38 | 81 | insert.setString(1, instance.getUsername()); |
universe@38 | 82 | setStringOrNull(insert, 2, instance.getLastname()); |
universe@38 | 83 | setStringOrNull(insert, 3, instance.getGivenname()); |
universe@38 | 84 | setStringOrNull(insert, 4, instance.getMail()); |
universe@38 | 85 | insert.executeUpdate(); |
universe@38 | 86 | } |
universe@38 | 87 | |
universe@38 | 88 | @Override |
universe@38 | 89 | public boolean update(User instance) throws SQLException { |
universe@75 | 90 | if (instance.getId() < 0) return false; |
universe@38 | 91 | setStringOrNull(update, 1, instance.getLastname()); |
universe@38 | 92 | setStringOrNull(update, 2, instance.getGivenname()); |
universe@38 | 93 | setStringOrNull(update, 3, instance.getMail()); |
universe@51 | 94 | update.setInt(4, instance.getId()); |
universe@38 | 95 | return update.executeUpdate() > 0; |
universe@34 | 96 | } |
universe@47 | 97 | |
universe@47 | 98 | @Override |
universe@47 | 99 | public List<User> list() throws SQLException { |
universe@47 | 100 | List<User> users = new ArrayList<>(); |
universe@47 | 101 | try (var result = list.executeQuery()) { |
universe@47 | 102 | while (result.next()) { |
universe@47 | 103 | users.add(mapColumns(result)); |
universe@47 | 104 | } |
universe@47 | 105 | } |
universe@47 | 106 | return users; |
universe@47 | 107 | } |
universe@47 | 108 | |
universe@47 | 109 | @Override |
universe@47 | 110 | public User find(int id) throws SQLException { |
universe@47 | 111 | find.setInt(1, id); |
universe@47 | 112 | try (var result = find.executeQuery()) { |
universe@47 | 113 | if (result.next()) { |
universe@47 | 114 | return mapColumns(result); |
universe@47 | 115 | } else { |
universe@47 | 116 | return null; |
universe@47 | 117 | } |
universe@47 | 118 | } |
universe@47 | 119 | } |
universe@124 | 120 | |
universe@124 | 121 | @Override |
universe@124 | 122 | public Optional<User> findByUsername(String username) throws SQLException { |
universe@124 | 123 | findByUsername.setString(1, username); |
universe@124 | 124 | try (var result = findByUsername.executeQuery()) { |
universe@124 | 125 | if (result.next()) { |
universe@124 | 126 | return Optional.of(mapColumns(result)); |
universe@124 | 127 | } else { |
universe@124 | 128 | return Optional.empty(); |
universe@124 | 129 | } |
universe@124 | 130 | } |
universe@124 | 131 | } |
universe@34 | 132 | } |