|
1 /* |
|
2 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER. |
|
3 * |
|
4 * Copyright 2018 Mike Becker. All rights reserved. |
|
5 * |
|
6 * Redistribution and use in source and binary forms, with or without |
|
7 * modification, are permitted provided that the following conditions are met: |
|
8 * |
|
9 * 1. Redistributions of source code must retain the above copyright |
|
10 * notice, this list of conditions and the following disclaimer. |
|
11 * |
|
12 * 2. Redistributions in binary form must reproduce the above copyright |
|
13 * notice, this list of conditions and the following disclaimer in the |
|
14 * documentation and/or other materials provided with the distribution. |
|
15 * |
|
16 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" |
|
17 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE |
|
18 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE |
|
19 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE |
|
20 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR |
|
21 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF |
|
22 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS |
|
23 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN |
|
24 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) |
|
25 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE |
|
26 * POSSIBILITY OF SUCH DAMAGE. |
|
27 * |
|
28 */ |
|
29 package de.uapcore.lightpit.dao.postgres; |
|
30 |
|
31 import de.uapcore.lightpit.LightPITModule; |
|
32 import de.uapcore.lightpit.dao.AbstractDao; |
|
33 import de.uapcore.lightpit.dao.ModuleDao; |
|
34 import de.uapcore.lightpit.entities.Module; |
|
35 |
|
36 import java.sql.Connection; |
|
37 import java.sql.PreparedStatement; |
|
38 import java.sql.ResultSet; |
|
39 import java.sql.SQLException; |
|
40 import java.util.Map; |
|
41 import java.util.Set; |
|
42 |
|
43 public final class PGModuleDao extends AbstractDao<Module> implements ModuleDao { |
|
44 |
|
45 @Override |
|
46 protected PreparedStatement listQuery(Connection connection) throws SQLException { |
|
47 return connection.prepareStatement("select * from lpitcore_module"); |
|
48 } |
|
49 |
|
50 @Override |
|
51 protected Module mapColumns(ResultSet result) throws SQLException { |
|
52 final var mod = new Module(); |
|
53 mod.setModID(result.getInt("modid")); |
|
54 mod.setClassname(result.getString("classname")); |
|
55 mod.setVisible(result.getBoolean("visible")); |
|
56 mod.setPriority(result.getInt("priority")); |
|
57 return mod; |
|
58 } |
|
59 |
|
60 @Override |
|
61 public void syncRegisteredModuleClasses(Connection conn, Set<Map.Entry<String, LightPITModule>> moduleSet) throws SQLException { |
|
62 |
|
63 var check = conn.prepareStatement("select visible from lpitcore_module where classname = ?"); |
|
64 var insert = conn.prepareStatement("insert into lpitcore_module (classname, visible, priority) values (?, ?, ?)"); |
|
65 insert.setBoolean(2, true); |
|
66 // update/delete not required, we do this in the module management UI |
|
67 |
|
68 for (Map.Entry<String, LightPITModule> modEntry : moduleSet) { |
|
69 if (modEntry.getValue().systemModule()) continue; |
|
70 |
|
71 check.setString(1, modEntry.getKey()); |
|
72 try (ResultSet r = check.executeQuery()) { |
|
73 if (!r.next()) { |
|
74 insert.setString(1, modEntry.getKey()); |
|
75 insert.setInt(3, modEntry.getValue().defaultPriority()); |
|
76 insert.executeUpdate(); |
|
77 } |
|
78 } |
|
79 } |
|
80 } |
|
81 } |