Fri, 23 Oct 2020 18:40:50 +0200
migrate WebColor
/* * 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 desc, lower(name) desc"); 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<Component> 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); } }