1 files changed, 65 insertions, 0 deletions
diff --git a/src/database/ConnectionBuilder.java b/src/database/ConnectionBuilder.java
new file mode 100644
index 0000000..74e5c33
--- /dev/null
+++ b/src/database/ConnectionBuilder.java
@@ -0,0 +1,65 @@
+package database;
+
+import java.sql.Connection;
+import java.sql.DriverManager;
+import java.sql.SQLException;
+
+/**
+ * Builder for connection.
+ *
+ * @author Peter Wu
+ */
+public class ConnectionBuilder {
+
+    private String dbms;
+    private String serverName;
+    private int port;
+    private String dbName;
+    private String username;
+    private String password;
+
+    /**
+     * Sets up a ConnectionBuilder for PostgreSQL and serverName localhost. The
+     * username, password and database name must still be supplied.
+     */
+    public ConnectionBuilder() {
+        dbms = "postgresql";
+        serverName = "localhost";
+        port = 5432;
+    }
+
+    public ConnectionBuilder setDbms(String dbms) {
+        this.dbms = dbms;
+        return this;
+    }
+
+    public ConnectionBuilder setServerName(String serverName) {
+        this.serverName = serverName;
+        return this;
+    }
+
+    public ConnectionBuilder setPort(int port) {
+        this.port = port;
+        return this;
+    }
+
+    public ConnectionBuilder setDbName(String dbName) {
+        this.dbName = dbName;
+        return this;
+    }
+
+    public ConnectionBuilder setUsername(String username) {
+        this.username = username;
+        return this;
+    }
+
+    public ConnectionBuilder setPassword(String password) {
+        this.password = password;
+        return this;
+    }
+
+    public Connection create() throws SQLException {
+        String url = "jdbc:" + dbms + "://" + serverName + ":" + port + "/" + dbName;
+        return DriverManager.getConnection(url, username, password);
+    }
+}