1 files changed, 98 insertions, 0 deletions
diff --git a/dozentenmodulserver/src/main/java/org/openslx/bwlp/sat/database/mappers/DbLectureNetworkRules.java b/dozentenmodulserver/src/main/java/org/openslx/bwlp/sat/database/mappers/DbLectureNetworkRules.java
new file mode 100644
index 00000000..61b0c50d
--- /dev/null
+++ b/dozentenmodulserver/src/main/java/org/openslx/bwlp/sat/database/mappers/DbLectureNetworkRules.java
@@ -0,0 +1,98 @@
+package org.openslx.bwlp.sat.database.mappers;
+
+import java.sql.ResultSet;
+import java.sql.SQLException;
+import java.util.ArrayList;
+import java.util.Arrays;
+import java.util.List;
+
+import org.apache.log4j.Logger;
+import org.openslx.bwlp.sat.database.Database;
+import org.openslx.bwlp.sat.database.MysqlConnection;
+import org.openslx.bwlp.sat.database.MysqlStatement;
+import org.openslx.bwlp.thrift.iface.NetRule;
+import org.openslx.bwlp.thrift.iface.PresetNetRule;
+import org.openslx.util.Json;
+
+public class DbLectureNetworkRules {
+	
+	private static final Logger LOGGER = Logger.getLogger(DbLectureNetworkRules.class);
+
+	public static void writeLectureNetworkExceptions(MysqlConnection connection, String lectureId,
+			List<Integer> ruleIds) throws SQLException {
+		if (lectureId == null || lectureId.isEmpty()) {
+			return;
+		}
+		MysqlStatement delStmt = connection
+				.prepareStatement("DELETE FROM lecture_x_networkrule WHERE lectureid = :lectureid");
+		delStmt.setString("lectureid", lectureId);
+		delStmt.executeUpdate();
+		if (ruleIds == null || ruleIds.isEmpty()) {
+			return;
+		}
+		MysqlStatement addStmt = connection
+				.prepareStatement("INSERT IGNORE INTO lecture_x_networkrule (lectureid, ruleid)"
+						+ " VALUES (:lectureid, :ruleid)");
+		addStmt.setString("lectureid", lectureId);
+		for (int ruleId : ruleIds) {
+			addStmt.setInt("ruleid", ruleId);
+			addStmt.executeUpdate();
+		}
+	}
+
+	static List<NetRule> getForStartup(MysqlConnection connection, String lectureId) throws SQLException {
+		MysqlStatement stmt = connection.prepareStatement("SELECT p.ruledata FROM presetnetworkrule p"
+				+ " INNER JOIN `lecture_x_networkrule` lxr"
+				+ " ON (lxr.lectureid = :lectureid AND lxr.ruleid = p.ruleid)");
+		stmt.setString("lectureid", lectureId);
+		ResultSet rs = stmt.executeQuery();
+		List<NetRule> result = new ArrayList<>();
+		while (rs.next()) {
+			NetRule[] lst = Json.deserialize(rs.getString("ruledata"), NetRule[].class);
+			if (lst == null)
+				continue;
+			for (NetRule r : lst) {
+				result.add(r);
+			}
+		}
+		return result;
+	}
+	
+	public static List<Integer> getForEdit(MysqlConnection connection, String lectureId) throws SQLException
+	{
+		MysqlStatement stmt = connection.prepareStatement("SELECT ruleid FROM lecture_x_networkrule"
+				+ " WHERE lectureid = :lectureid");
+		stmt.setString("lectureid", lectureId);
+		ResultSet rs = stmt.executeQuery();
+		List<Integer> result = new ArrayList<>();
+		while (rs.next()) {
+			result.add(rs.getInt("ruleid"));
+		}
+		return result;
+	}
+
+	public static List<PresetNetRule> getPredefined() throws SQLException {
+		try (MysqlConnection connection = Database.getConnection()) {
+			List<PresetNetRule> list = new ArrayList<>();
+			MysqlStatement stmt = connection
+					.prepareStatement("SELECT ruleid, rulename, ruledata"
+							+ " FROM presetnetworkrule");
+			ResultSet rs = stmt.executeQuery();
+			while (rs.next()) {
+				NetRule[] ruleArray = Json.deserialize(rs.getString("ruledata"), NetRule[].class);
+				List<NetRule> rules;
+				if (ruleArray == null || ruleArray.length == 0) {
+					rules = new ArrayList<>(0);
+				} else {
+					rules = Arrays.asList(ruleArray);
+				}
+				list.add(new PresetNetRule(rs.getInt("ruleid"), rs.getString("rulename"), rules));
+			}
+			return list;
+		} catch (SQLException e) {
+			LOGGER.error("Query failed in getPredefinedNetshares()", e);
+			throw e;
+		}
+	}
+
+}