feat: use embed.FS (go1.16) instead of go-rice

dosco · Mar 27, 2021 · db88f10 · db88f10
1 parent 3fe3b61
commit db88f10
Show file tree

Hide file tree

Showing 13 changed files with 385 additions and 423 deletions.
diff --git a/core/cockroach.sql b/core/cockroach.sql
@@ -0,0 +1,179 @@
+CREATE TABLE users (
+  id BIGSERIAL PRIMARY KEY,
+  full_name TEXT NOT NULL,
+  phone TEXT,
+  avatar TEXT,
+  stripe_id TEXT,
+  email TEXT UNIQUE NOT NULL,
+  category_counts JSON,
+  disabled bool DEFAULT false,
+  created_at TIMESTAMPTZ NOT NULL DEFAULT NOW(),
+  updated_at TIMESTAMPTZ
+);
+
+CREATE TABLE categories (
+  id BIGSERIAL PRIMARY KEY,
+  name TEXT NOT NULL CHECK (length(name) < 100),
+  description TEXT CHECK (length(description) < 300),
+  created_at TIMESTAMPTZ NOT NULL DEFAULT NOW(),
+  updated_at TIMESTAMPTZ
+);
+
+CREATE TABLE products (
+  id BIGSERIAL PRIMARY KEY,
+  name TEXT CHECK (length(name) > 1 AND length(name) < 50),
+  description TEXT CHECK (length(name) > 1 AND length(name) < 200),
+  tags  TEXT[],  
+  price NUMERIC(7,1),
+  owner_id BIGINT REFERENCES users(id),
+
+  category_ids BIGINT[] NOT NULL,
+
+  created_at TIMESTAMPTZ NOT NULL DEFAULT NOW(),
+  updated_at TIMESTAMPTZ
+);
+
+CREATE TABLE purchases (
+  id BIGSERIAL PRIMARY KEY,
+  customer_id BIGINT REFERENCES users(id),
+  product_id BIGINT REFERENCES products(id),
+  quantity integer,
+  returned_at TIMESTAMPTZ,
+  created_at TIMESTAMPTZ NOT NULL DEFAULT NOW(),
+  updated_at TIMESTAMPTZ
+);
+
+CREATE TABLE notifications (
+  id BIGSERIAL PRIMARY KEY,
+  verb TEXT,
+  subject_type TEXT,
+  subject_id BIGINT,
+  user_id BIGINT REFERENCES users(id),
+  created_at TIMESTAMPTZ NOT NULL DEFAULT NOW(),
+  updated_at TIMESTAMPTZ
+);
+
+CREATE TABLE comments (
+  id BIGSERIAL PRIMARY KEY,
+  body TEXT CHECK (length(body) > 1 AND length(body) < 200),
+  product_id BIGINT REFERENCES products(id),
+  commenter_id BIGINT REFERENCES users(id),
+  reply_to_id BIGINT REFERENCES comments(id),
+  created_at TIMESTAMPTZ NOT NULL DEFAULT NOW(),
+  updated_at TIMESTAMPTZ
+);
+
+CREATE TABLE chats (
+  id BIGSERIAL PRIMARY KEY,
+  body text,
+  reply_to_id  BIGINT REFERENCES chats(id),
+  created_at TIMESTAMPTZ NOT NULL DEFAULT NOW(),
+  updated_at TIMESTAMPTZ
+);
+
+CREATE VIEW 
+  hot_products
+AS (
+  SELECT 
+      id as product_id
+  FROM
+      products
+  WHERE
+      id > 50
+);
+
+-- CREATE TABLE chats (
+--   id BIGSERIAL PRIMARY KEY,
+--   body TEXT,
+--   reply_to_id BIGINT[],
+--   created_at TIMESTAMPTZ NOT NULL DEFAULT NOW(),
+--   updated_at TIMESTAMPTZ
+-- );
+
+-- insert users
+INSERT INTO 
+  users (id, full_name, email, stripe_id, category_counts, disabled, created_at) 
+SELECT 
+  i, 
+  'User ' || i::string,
+  'user' || i::string || '@test.com',
+  'payment_id_' || (i + 1000)::string,
+  '[{"category_id": 1, "count": 400},{"category_id": 2, "count": 600}]',
+  (CASE WHEN i = 50 THEN true ELSE false END),
+  '2021-01-09 16:37:01'
+ FROM 
+  GENERATE_SERIES(1, 100) i;
+
+-- insert categories
+INSERT INTO 
+  categories (id, name, description, created_at) 
+SELECT 
+  i, 
+  'Category ' || i::string,
+  'Description for category ' || i::string,
+  '2021-01-09 16:37:01'
+FROM 
+  GENERATE_SERIES(1, 5) i;
+
+-- insert products
+INSERT INTO 
+  products (id, name, description, tags, category_ids, price, owner_id, created_at)
+SELECT 
+  i, 
+  'Product ' || i::string, 
+  'Description for product ' || i::string,
+ 	(SELECT array_agg(('Tag ' || i::string)) FROM GENERATE_SERIES(1, 5) i),
+  (SELECT array_agg(i) FROM GENERATE_SERIES(1, 5) i),
+  (i + 10.5),
+  i,
+  '2021-01-09 16:37:01'
+FROM 
+  GENERATE_SERIES(1, 100) i;
+
+-- insert purchases
+INSERT INTO 
+  purchases (id, customer_id, product_id, quantity, created_at)
+SELECT 
+  i, 
+  (CASE WHEN i >= 100 THEN 1 ELSE (i+1) END),
+  i,
+  (i * 10),
+  '2021-01-09 16:37:01'
+FROM 
+  GENERATE_SERIES(1, 100) i;
+
+-- insert notifications
+INSERT INTO 
+  notifications (id, verb, subject_type, subject_id, user_id, created_at)
+SELECT 
+  i, 
+  (CASE WHEN ((i % 2) = 0) THEN 'Bought' ELSE 'Joined' END),
+  (CASE WHEN ((i % 2) = 0) THEN 'products' ELSE 'users' END),
+  i,
+  (CASE WHEN i >= 2 THEN i - 1 ELSE NULL END),
+  '2021-01-09 16:37:01'
+FROM 
+  GENERATE_SERIES(1, 100) i;
+
+-- insert comments
+INSERT INTO 
+  comments (id, body, product_id, commenter_id, reply_to_id, created_at)
+SELECT 
+  i, 
+  'This is comment number ' || i::string,
+  i,
+  i,
+  (CASE WHEN i >= 2 THEN i - 1 ELSE NULL END),
+  '2021-01-09 16:37:01'
+FROM 
+  GENERATE_SERIES(1, 100) i;
+
+-- insert chats
+INSERT INTO 
+  chats (id, body, created_at)
+SELECT 
+  i, 
+  'This is chat message number ' || i::string,
+  '2021-01-09 16:37:01'
+FROM 
+  GENERATE_SERIES(1, 5) i;
diff --git a/core/core.go b/core/core.go
@@ -62,9 +62,12 @@ type qres struct {
 }
 
 func (gj *GraphJin) initDiscover() error {
-	if gj.conf.DBType == "" {
+	switch gj.conf.DBType {
+	case "":
 		gj.dbtype = "postgres"
-	} else {
+	case "mssql":
+		gj.dbtype = "mysql"
+	default:
 		gj.dbtype = gj.conf.DBType
 	}
 

diff --git a/core/core_test.go b/core/core_test.go
@@ -8,6 +8,8 @@ import (
 	"testing"
 
 	"github.com/orlangure/gnomock"
+	"github.com/orlangure/gnomock/preset/cockroachdb"
+	"github.com/orlangure/gnomock/preset/mssql"
 	"github.com/orlangure/gnomock/preset/mysql"
 	"github.com/orlangure/gnomock/preset/postgres"
 )
@@ -42,7 +44,18 @@ func TestMain(m *testing.M) {
 				postgres.WithUser("tester", "tester"),
 				postgres.WithDatabase("db"),
 				postgres.WithQueriesFile("./postgres.sql"),
-			)},
+			),
+		},
+		{
+			disable: true,
+			name:    "cockroach",
+			driver:  "postgres",
+			connstr: "postgres://root:@%s/db?sslmode=disable",
+			preset: cockroachdb.Preset(
+				cockroachdb.WithDatabase("db"),
+				cockroachdb.WithQueriesFile("./cockroach.sql"),
+			),
+		},
 		{
 			disable: true,
 			name:    "mysql",
@@ -52,7 +65,20 @@ func TestMain(m *testing.M) {
 				mysql.WithUser("user", "user"),
 				mysql.WithDatabase("db"),
 				mysql.WithQueriesFile("./mysql.sql"),
-			)},
+			),
+		},
+		{
+			disable: true,
+			name:    "mssql",
+			driver:  "sqlserver",
+			connstr: "sqlserver://sa:password@%s?database=db",
+			preset: mssql.Preset(
+				mssql.WithLicense(true),
+				mssql.WithAdminPassword("password"),
+				mssql.WithDatabase("db"),
+				mssql.WithQueriesFile("./mysql.sql"),
+			),
+		},
 	}
 
 	for _, v := range dbinfoList {
@@ -74,7 +100,11 @@ func TestMain(m *testing.M) {
 		if err != nil {
 			panic(err)
 		}
-		defer func() { _ = gnomock.Stop(con) }()
+		defer func() {
+			if err := gnomock.Stop(con); err != nil {
+				panic(err)
+			}
+		}()
 
 		db, err = sql.Open(v.driver, fmt.Sprintf(v.connstr, con.DefaultAddress()))
 		if err != nil {