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