Mercurial > hg > LightPIT / file revision

 /*

  * 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.LightPITModule;

 import de.uapcore.lightpit.dao.AbstractDao;

 import de.uapcore.lightpit.dao.ModuleDao;

 import de.uapcore.lightpit.entities.Module;

 import java.sql.Connection;

 import java.sql.PreparedStatement;

 import java.sql.ResultSet;

 import java.sql.SQLException;

 import java.util.Map;

 import java.util.Set;

 public final class PGModuleDao extends AbstractDao<Module> implements ModuleDao {

     @Override

     protected PreparedStatement listQuery(Connection connection) throws SQLException {

         return connection.prepareStatement("select * from lpitcore_module");

     }

     @Override

     protected Module mapColumns(ResultSet result) throws SQLException {

         final var mod = new Module();

         mod.setModID(result.getInt("modid"));

         mod.setClassname(result.getString("classname"));

         mod.setVisible(result.getBoolean("visible"));

         mod.setPriority(result.getInt("priority"));

         return mod;

     }

     @Override

     public void syncRegisteredModuleClasses(Connection conn, Set<Map.Entry<String, LightPITModule>> moduleSet) throws SQLException {

         var check = conn.prepareStatement("select visible from lpitcore_module where classname = ?");

         var insert = conn.prepareStatement("insert into lpitcore_module (classname, visible, priority) values (?, ?, ?)");

         insert.setBoolean(2, true);

         // update/delete not required, we do this in the module management UI

         for (Map.Entry<String, LightPITModule> modEntry : moduleSet) {

             if (modEntry.getValue().systemModule()) continue;

             check.setString(1, modEntry.getKey());

             try (ResultSet r = check.executeQuery()) {

                 if (!r.next()) {

                     insert.setString(1, modEntry.getKey());

                     insert.setInt(3, modEntry.getValue().defaultPriority());

                     insert.executeUpdate();

                 }

             }

         }

     }

 }