From 0650161d8faa675bff498dd881316f3caed85677 Mon Sep 17 00:00:00 2001
From: Dipesh Dulal <nilkantha.dipesh@gmail.com>
Date: Sat, 16 Jan 2021 10:20:43 +0545
Subject: [PATCH] feat: added db seeding

---
 main.go    |  4 +++-
 seed_db.go | 17 +++++++++++++++++
 2 files changed, 20 insertions(+), 1 deletion(-)
 create mode 100644 seed_db.go

diff --git a/main.go b/main.go
index db6d6f5..87678ce 100644
--- a/main.go
+++ b/main.go
@@ -25,6 +25,8 @@ func main() {
 
 	db := initDBConnection()
 
+	seedDB(db)
+
 	rows, err := db.Query("SELECT id, name FROM public.test")
 
 	if err != nil {
@@ -43,7 +45,7 @@ func main() {
 
 	go func() {
 		for range interrupt {
-			log.Print("Interrupt received closing...")
+			log.Println("\n❌ Interrupt received closing...")
 			cancel()
 		}
 	}()
diff --git a/seed_db.go b/seed_db.go
new file mode 100644
index 0000000..187d259
--- /dev/null
+++ b/seed_db.go
@@ -0,0 +1,17 @@
+package main
+
+import (
+	"database/sql"
+	"log"
+)
+
+func seedDB(db *sql.DB) error {
+	log.Print("💾 Seeding database with table...")
+	_, err := db.Exec(`
+		CREATE TABLE IF NOT EXISTS "public"."test" (
+			"id" integer NOT NULL,
+			"name" text NOT NULL
+		)
+	`)
+	return err
+}