diff -r 4f912cd42876 -r 86b5d8a1662f src/main/java/de/uapcore/lightpit/dao/postgres/PGComponentDao.java --- a/src/main/java/de/uapcore/lightpit/dao/postgres/PGComponentDao.java Fri Nov 06 10:50:32 2020 +0100 +++ /dev/null Thu Jan 01 00:00:00 1970 +0000 @@ -1,130 +0,0 @@ -/* - * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER. - * - * Copyright 2018 Mike Becker. All rights reserved. - * - * Redistribution and use in source and binary forms, with or without - * modification, are permitted provided that the following conditions are met: - * - * 1. Redistributions of source code must retain the above copyright - * notice, this list of conditions and the following disclaimer. - * - * 2. Redistributions in binary form must reproduce the above copyright - * notice, this list of conditions and the following disclaimer in the - * documentation and/or other materials provided with the distribution. - * - * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" - * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE - * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE - * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE - * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR - * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF - * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS - * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN - * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) - * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE - * POSSIBILITY OF SUCH DAMAGE. - * - */ -package de.uapcore.lightpit.dao.postgres; - -import de.uapcore.lightpit.dao.ComponentDao; -import de.uapcore.lightpit.dao.Functions; -import de.uapcore.lightpit.entities.Component; -import de.uapcore.lightpit.entities.Project; -import de.uapcore.lightpit.entities.User; -import de.uapcore.lightpit.types.WebColor; - -import java.sql.Connection; -import java.sql.PreparedStatement; -import java.sql.ResultSet; -import java.sql.SQLException; -import java.util.List; - -public final class PGComponentDao implements ComponentDao { - - private final PreparedStatement insert, update, list, find, findByNode; - - public PGComponentDao(Connection connection) throws SQLException { - final var query = "select id, name, node, color, ordinal, description, " + - "userid, username, givenname, lastname, mail " + - "from lpit_component " + - "left join lpit_user on lead = userid"; - - list = connection.prepareStatement(query + " where project = ? order by ordinal, lower(name)"); - - find = connection.prepareStatement(query + " where id = ? "); - - findByNode = connection.prepareStatement(query + " where project = ? and node = ?"); - - insert = connection.prepareStatement( - "insert into lpit_component (name, node, color, ordinal, description, lead, project) values (?, ?, ?, ?, ?, ?, ?)" - ); - - update = connection.prepareStatement( - "update lpit_component set name = ?, node = ?, color = ?, ordinal = ?, description = ?, lead = ? where id = ?" - ); - } - - private static Component mapColumns(ResultSet result) throws SQLException { - final var component = new Component(result.getInt("id")); - component.setName(result.getString("name")); - component.setNode(result.getString("node")); - try { - component.setColor(new WebColor(result.getString("color"))); - } catch (IllegalArgumentException ex) { - // if someone tempered with the database we default the color to black - component.setColor(new WebColor("000000")); - } - component.setOrdinal(result.getInt("ordinal")); - component.setDescription(result.getString("description")); - component.setLead(PGUserDao.mapColumns(result)); - return component; - } - - private static int setColumns(PreparedStatement stmt, Component instance) throws SQLException { - int column = 0; - stmt.setString(++column, instance.getName()); - stmt.setString(++column, instance.getNode()); - stmt.setString(++column, instance.getColor().getHex()); - stmt.setInt(++column, instance.getOrdinal()); - Functions.setStringOrNull(stmt, ++column, instance.getDescription()); - Functions.setForeignKeyOrNull(stmt, ++column, instance.getLead(), User::getId); - return column; - } - - @Override - public void save(Component instance, Project project) throws SQLException { - int column = setColumns(insert, instance); - insert.setInt(++column, project.getId()); - insert.executeUpdate(); - } - - @Override - public boolean update(Component instance) throws SQLException { - if (instance.getId() < 0) return false; - int column = setColumns(update, instance); - update.setInt(++column, instance.getId()); - return update.executeUpdate() > 0; - } - - - @Override - public List list(Project project) throws SQLException { - list.setInt(1, project.getId()); - return Functions.list(list, PGComponentDao::mapColumns); - } - - @Override - public Component find(int id) throws SQLException { - find.setInt(1, id); - return Functions.find(find, PGComponentDao::mapColumns); - } - - @Override - public Component findByNode(Project project, String node) throws SQLException { - findByNode.setInt(1, project.getId()); - findByNode.setString(2, node);; - return Functions.find(findByNode, PGComponentDao::mapColumns); - } -}