split classes into separate files

1 files changed, 86 insertions, 0 deletions

diff --git a/src/com.example.portfolio2/com/example/portfolio2/MyDb.java b/src/com.example.portfolio2/com/example/portfolio2/MyDb.java
new file mode 100644
index 0000000..f85f391
--- /dev/null
+++ b/src/com.example.portfolio2/com/example/portfolio2/MyDb.java
@@ -0,0 +1,86 @@
+package com.example.portfolio2;
+
+import java.sql.*;
+import java.util.ArrayList;
+
+class MyDB { // MyDB is all standard Database configuration that was gotten from Mads
+    Connection conn = null;
+
+    MyDB() {
+        if (conn == null) open();
+    }
+
+    public void open() {
+        try {
+            String url = "jdbc:sqlite:identifier.sqlite";
+            conn = DriverManager.getConnection(url);
+        } catch (SQLException e) {
+            System.out.println("cannot open");
+            if (conn != null) close();
+            throw new RuntimeException(e);
+        }
+        ;
+    }
+
+    public void close() {
+        try {
+            if (conn != null) conn.close();
+        } catch (SQLException e) {
+            throw new RuntimeException(e);
+        }
+        conn = null;
+    }
+    public void cmd(String sql) {
+        if (conn == null) open();
+        if (conn == null) {
+            System.out.println("No connection");
+            return;
+        }
+        Statement stmt = null;
+        try {
+            stmt = conn.createStatement();
+            stmt.executeUpdate(sql);
+        } catch (SQLException e) {
+            System.out.println("Error in statement " + sql);
+            throw new RuntimeException(e);
+        }
+        try {
+            if (stmt != null) {
+                stmt.close();
+            }
+        } catch (SQLException e) {
+            System.out.println("Error in statement " + sql);
+            throw new RuntimeException(e);
+        }
+    }
+
+    public ArrayList<String> query(String query, String fld) {
+        ArrayList<String> res = new ArrayList<>();
+        if (conn == null) open();
+        if (conn == null) {
+            System.out.println("No connection");
+            throw new RuntimeException("No connection");
+        }
+        Statement stmt = null;
+        try {
+            stmt = conn.createStatement();
+            ResultSet rs = stmt.executeQuery(query);
+            while (rs.next()) {
+                String name = rs.getString(fld);
+                res.add(name);
+            }
+        } catch (SQLException e) {
+            System.out.println("Error in statement " + query + " " + fld);
+            throw new RuntimeException(e);
+        }
+        try {
+            if (stmt != null) {
+                stmt.close();
+            }
+        } catch (SQLException e) {
+            System.out.println("Error in statement " + query + " " + fld);
+            throw new RuntimeException(e);
+        }
+        return res;
+    }
+}