1 files changed, 38 insertions, 0 deletions
diff --git a/research/lesci/database.go b/research/lesci/database.go
new file mode 100644
index 0000000..69c8d49
--- /dev/null
+++ b/research/lesci/database.go
@@ -0,0 +1,38 @@
+package lesci
+
+import (
+	"context"
+	"database/sql"
+	"errors"
+	"fmt"
+)
+
+func EnsureTable(db *sql.DB, table string, query string) (bool, error) {
+	ctx := context.Background()
+
+	var exists bool
+
+	queryTableExists := `SELECT COUNT(*) > 0 FROM information_schema.tables WHERE table_name = ?`
+
+	if err := db.QueryRowContext(ctx, queryTableExists, table).Scan(&exists); err != nil {
+		return false, err
+	}
+
+	if !exists {
+		if _, err := db.ExecContext(ctx, query); err != nil {
+			return false, err
+		}
+
+		return true, nil
+	}
+
+	var count int64
+
+	queryNumRows := fmt.Sprintf(`SELECT COUNT(*) FROM "%s"`, table)
+
+	if err := db.QueryRowContext(ctx, queryNumRows).Scan(&count); err != nil {
+		return false, errors.New("table not empty")
+	}
+
+	return count == 0, nil
+}