1.1 --- /dev/null Thu Jan 01 00:00:00 1970 +0000
1.2 +++ b/src/main/java/de/uapcore/lightpit/dao/postgres/PGModuleDao.java Sat May 09 17:01:29 2020 +0200
1.3 @@ -0,0 +1,81 @@
1.4 +/*
1.5 + * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
1.6 + *
1.7 + * Copyright 2018 Mike Becker. All rights reserved.
1.8 + *
1.9 + * Redistribution and use in source and binary forms, with or without
1.10 + * modification, are permitted provided that the following conditions are met:
1.11 + *
1.12 + * 1. Redistributions of source code must retain the above copyright
1.13 + * notice, this list of conditions and the following disclaimer.
1.14 + *
1.15 + * 2. Redistributions in binary form must reproduce the above copyright
1.16 + * notice, this list of conditions and the following disclaimer in the
1.17 + * documentation and/or other materials provided with the distribution.
1.18 + *
1.19 + * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
1.20 + * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
1.21 + * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
1.22 + * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
1.23 + * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
1.24 + * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
1.25 + * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
1.26 + * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
1.27 + * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
1.28 + * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
1.29 + * POSSIBILITY OF SUCH DAMAGE.
1.30 + *
1.31 + */
1.32 +package de.uapcore.lightpit.dao.postgres;
1.33 +
1.34 +import de.uapcore.lightpit.LightPITModule;
1.35 +import de.uapcore.lightpit.dao.AbstractDao;
1.36 +import de.uapcore.lightpit.dao.ModuleDao;
1.37 +import de.uapcore.lightpit.entities.Module;
1.38 +
1.39 +import java.sql.Connection;
1.40 +import java.sql.PreparedStatement;
1.41 +import java.sql.ResultSet;
1.42 +import java.sql.SQLException;
1.43 +import java.util.Map;
1.44 +import java.util.Set;
1.45 +
1.46 +public final class PGModuleDao extends AbstractDao<Module> implements ModuleDao {
1.47 +
1.48 + @Override
1.49 + protected PreparedStatement listQuery(Connection connection) throws SQLException {
1.50 + return connection.prepareStatement("select * from lpitcore_module");
1.51 + }
1.52 +
1.53 + @Override
1.54 + protected Module mapColumns(ResultSet result) throws SQLException {
1.55 + final var mod = new Module();
1.56 + mod.setModID(result.getInt("modid"));
1.57 + mod.setClassname(result.getString("classname"));
1.58 + mod.setVisible(result.getBoolean("visible"));
1.59 + mod.setPriority(result.getInt("priority"));
1.60 + return mod;
1.61 + }
1.62 +
1.63 + @Override
1.64 + public void syncRegisteredModuleClasses(Connection conn, Set<Map.Entry<String, LightPITModule>> moduleSet) throws SQLException {
1.65 +
1.66 + var check = conn.prepareStatement("select visible from lpitcore_module where classname = ?");
1.67 + var insert = conn.prepareStatement("insert into lpitcore_module (classname, visible, priority) values (?, ?, ?)");
1.68 + insert.setBoolean(2, true);
1.69 + // update/delete not required, we do this in the module management UI
1.70 +
1.71 + for (Map.Entry<String, LightPITModule> modEntry : moduleSet) {
1.72 + if (modEntry.getValue().systemModule()) continue;
1.73 +
1.74 + check.setString(1, modEntry.getKey());
1.75 + try (ResultSet r = check.executeQuery()) {
1.76 + if (!r.next()) {
1.77 + insert.setString(1, modEntry.getKey());
1.78 + insert.setInt(3, modEntry.getValue().defaultPriority());
1.79 + insert.executeUpdate();
1.80 + }
1.81 + }
1.82 + }
1.83 + }
1.84 +}
