Thu, 22 Oct 2020 13:03:26 +0200
adds custom node names - fixes #27
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.UserDao;
32 import de.uapcore.lightpit.entities.User;
34 import java.sql.Connection;
35 import java.sql.PreparedStatement;
36 import java.sql.ResultSet;
37 import java.sql.SQLException;
38 import java.util.ArrayList;
39 import java.util.List;
40 import java.util.Objects;
41 import java.util.Optional;
43 import static de.uapcore.lightpit.dao.Functions.setStringOrNull;
45 public final class PGUserDao implements UserDao {
47 private final PreparedStatement insert, update, list, find, findByUsername;
49 public PGUserDao(Connection connection) throws SQLException {
50 list = connection.prepareStatement(
51 "select userid, username, lastname, givenname, mail " +
52 "from lpit_user where userid >= 0 " +
53 "order by username");
54 find = connection.prepareStatement(
55 "select userid, username, lastname, givenname, mail " +
56 "from lpit_user where userid = ? ");
58 findByUsername = connection.prepareStatement(
59 "select userid, username, lastname, givenname, mail " +
60 "from lpit_user where lower(username) = lower(?) ");
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 static 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 if (instance.getId() < 0) return false;
90 setStringOrNull(update, 1, instance.getLastname());
91 setStringOrNull(update, 2, instance.getGivenname());
92 setStringOrNull(update, 3, instance.getMail());
93 update.setInt(4, instance.getId());
94 return update.executeUpdate() > 0;
95 }
97 @Override
98 public List<User> list() throws SQLException {
99 List<User> users = new ArrayList<>();
100 try (var result = list.executeQuery()) {
101 while (result.next()) {
102 users.add(mapColumns(result));
103 }
104 }
105 return users;
106 }
108 @Override
109 public User find(int id) throws SQLException {
110 find.setInt(1, id);
111 try (var result = find.executeQuery()) {
112 if (result.next()) {
113 return mapColumns(result);
114 } else {
115 return null;
116 }
117 }
118 }
120 @Override
121 public Optional<User> findByUsername(String username) throws SQLException {
122 findByUsername.setString(1, username);
123 try (var result = findByUsername.executeQuery()) {
124 if (result.next()) {
125 return Optional.of(mapColumns(result));
126 } else {
127 return Optional.empty();
128 }
129 }
130 }
131 }