Initial commit

internal/db/models.go

@@ -0,0 +1,16 @@
+package db
+
+import "github.com/scylladb/gocqlx/v2/table"
+
+var UrlTable = table.New(table.Metadata{
+	Name:    "fivefeeteleven.urls",
+	Columns: []string{"id", "redirect_url", "secret"},
+	PartKey: []string{"id"},
+	SortKey: []string{},
+})
+
+type UrlModel struct {
+	Id          string
+	RedirectUrl string
+	Secret      *string
+}

internal/db/seed.go

@@ -0,0 +1,27 @@
+package db
+
+import (
+	"github.com/scylladb/gocqlx/v2"
+)
+
+func Seed(session gocqlx.Session) error {
+	err := session.ExecStmt(`
+		CREATE KEYSPACE IF NOT EXISTS fivefeeteleven 
+		WITH replication = {'class': 'SimpleStrategy', 'replication_factor': 1}`)
+	if err != nil {
+		return err
+	}
+
+	err = session.ExecStmt(`
+		CREATE TABLE IF NOT EXISTS fivefeeteleven.urls (
+			id text PRIMARY KEY,
+			redirect_url text,
+			secret text
+		)`)
+
+	if err != nil {
+		return err
+	}
+
+	return nil
+}