o First badge: UNBREAKABLE.

1 files changed, 65 insertions, 0 deletions

diff --git a/src/main/java/io/trygvis/esper/testing/core/db/BuildDao.java b/src/main/java/io/trygvis/esper/testing/core/db/BuildDao.java
new file mode 100644
index 0000000..24ecfd3
--- /dev/null
+++ b/src/main/java/io/trygvis/esper/testing/core/db/BuildDao.java
@@ -0,0 +1,65 @@
+package io.trygvis.esper.testing.core.db;
+
+import io.trygvis.esper.testing.*;
+import io.trygvis.esper.testing.util.sql.*;
+import org.joda.time.*;
+
+import java.sql.*;
+import java.util.*;
+
+import static io.trygvis.esper.testing.util.sql.ResultSetF.*;
+import static java.lang.System.*;
+
+public class BuildDao {
+    private final Connection c;
+
+    public static final String BUILD = "uuid, created_date, timestamp, success, reference_type, reference_uuid";
+
+    public static final SqlF<ResultSet, BuildDto> build = new SqlF<ResultSet, BuildDto>() {
+        public BuildDto apply(ResultSet rs) throws SQLException {
+            int i = 1;
+            return new BuildDto(
+                    UUID.fromString(rs.getString(i++)),
+                    new DateTime(rs.getTimestamp(i++).getTime()),
+                    new DateTime(rs.getTimestamp(i++).getTime()),
+                    rs.getBoolean(i++),
+                    EntityRef.fromRs(rs, i));
+        }
+    };
+
+    public BuildDao(Connection c) {
+        this.c = c;
+    }
+
+    public UUID insertBuild(DateTime timestamp, boolean success, EntityRef ref) throws SQLException {
+        try (PreparedStatement s = c.prepareStatement("INSERT INTO build(" + BUILD + ") VALUES(?, ?, ?, ?, ?, ?)")) {
+            UUID uuid = UUID.randomUUID();
+            int i = 1;
+            s.setString(i++, uuid.toString());
+            s.setTimestamp(i++, new Timestamp(currentTimeMillis()));
+            s.setTimestamp(i++, new Timestamp(timestamp.getMillis()));
+            s.setBoolean(i++, success);
+            s.setString(i++, ref.type);
+            s.setString(i, ref.uuid.toString());
+            s.executeUpdate();
+            return uuid;
+        }
+    }
+
+    public void insertBuildParticipant(UUID build, UUID person) throws SQLException {
+        try (PreparedStatement s = c.prepareStatement("INSERT INTO build_participant(build, person) VALUES(?, ?)")) {
+            int i = 1;
+            s.setString(i++, build.toString());
+            s.setString(i, person.toString());
+            s.executeUpdate();
+        }
+    }
+
+    public List<UUID> selectPersonsFromBuildParticipant(UUID build) throws SQLException {
+        try (PreparedStatement s = c.prepareStatement("SELECT person FROM build_participant WHERE build=?")) {
+            int i = 1;
+            s.setString(i, build.toString());
+            return Util.toList(s, getUuid);
+        }
+    }
+}