Add database adapter pkg

.circleci/config.yml

@@ -7,6 +7,11 @@ jobs:
       environment:
         GO111MODULE: "on"
     - image: circleci/mysql:5.7
+    - image: circleci/postgres:12.2
+      environment:
+        POSTGRES_USER: root
+        POSTGRES_DB: rapidash
+        POSTGRES_HOST_AUTH_METHOD: trust
     - image: memcached:1.5
     - image: redis:5.0
     steps:
@@ -25,11 +30,15 @@ jobs:
             golangci-lint run
       - run:
           name: Wait for DB
-          command: dockerize -wait tcp://127.0.0.1:3306 -timeout 30s
+          command: dockerize -wait tcp://127.0.0.1:3306 -wait tcp://127.0.0.1:5432 -timeout 30s
+      - run:
+          name: Run unit tests for postgres
+          command: |
+            RAPIDASH_DB_DRIVER=postgres go test -v ./...
       - run:
           name: Run unit tests and measure coverage
           command: |
-            go test -v -coverprofile=coverage.out ./...
+            RAPIDASH_DB_DRIVER=mysql go test -v -coverprofile=coverage.out ./...
             bash <(curl -s https://codecov.io/bash) -P ${CIRCLE_PULL_REQUEST##*/}
 workflows:
   version: 2

benchmark_test.go

@@ -202,7 +202,7 @@ func BenchmarkGetByPrimaryKey_RapidashWorst(b *testing.B) {
 		if err != nil {
 			panic(err)
 		}
-		builder := NewQueryBuilder("a").Eq("id", id)
+		builder := NewQueryBuilder("a", driver.Adapter).Eq("id", id)
 		var a A
 		if err := tx.FindByQueryBuilder(builder, &a); err != nil {
 			panic(err)
@@ -243,7 +243,7 @@ func BenchmarkGetByPrimaryKey_RapidashBest(b *testing.B) {
 		if err != nil {
 			panic(err)
 		}
-		builder := NewQueryBuilder("a").Eq("id", id)
+		builder := NewQueryBuilder("a", driver.Adapter).Eq("id", id)
 		var a A
 		if err := tx.FindByQueryBuilder(builder, &a); err != nil {
 			panic(err)
@@ -261,7 +261,7 @@ func BenchmarkGetByPrimaryKey_RapidashBest(b *testing.B) {
 		if err != nil {
 			panic(err)
 		}
-		builder := NewQueryBuilder("a").Eq("id", id)
+		builder := NewQueryBuilder("a", driver.Adapter).Eq("id", id)
 		var a A
 		if err := tx.FindByQueryBuilder(builder, &a); err != nil {
 			panic(err)
@@ -462,7 +462,7 @@ func BenchmarkUpdateByPrimaryKey_RapidashWorst(b *testing.B) {
 		if err != nil {
 			panic(err)
 		}
-		builder := NewQueryBuilder("a").Eq("id", id)
+		builder := NewQueryBuilder("a", driver.Adapter).Eq("id", id)
 		if err := tx.UpdateByQueryBuilder(builder, map[string]interface{}{
 			"name": "bench2",
 		}); err != nil {
@@ -503,7 +503,7 @@ func BenchmarkUpdateByPrimaryKey_RapidashBest(b *testing.B) {
 		if err != nil {
 			panic(err)
 		}
-		builder := NewQueryBuilder("a").Eq("id", id)
+		builder := NewQueryBuilder("a", driver.Adapter).Eq("id", id)
 		var a A
 		if err := tx.FindByQueryBuilder(builder, &a); err != nil {
 			panic(err)
@@ -524,7 +524,7 @@ func BenchmarkUpdateByPrimaryKey_RapidashBest(b *testing.B) {
 		if err != nil {
 			panic(err)
 		}
-		builder := NewQueryBuilder("a").Eq("id", id)
+		builder := NewQueryBuilder("a", driver.Adapter).Eq("id", id)
 		if err := tx.UpdateByQueryBuilder(builder, map[string]interface{}{
 			"name": "bench2",
 		}); err != nil {
@@ -622,7 +622,7 @@ func BenchmarkDeleteByPrimaryKey_Rapidash(b *testing.B) {
 		if err != nil {
 			panic(err)
 		}
-		builder := NewQueryBuilder("a").Eq("id", id)
+		builder := NewQueryBuilder("a", driver.Adapter).Eq("id", id)
 		if err := tx.DeleteByQueryBuilder(builder); err != nil {
 			panic(err)
 		}

config_test.go

@@ -10,10 +10,10 @@ import (
 func TestConfig(t *testing.T) {
 	cfg, err := NewConfig("testdata/cache.yml")
 	NoError(t, err)
-	cache, err := New(cfg.Options()...)
+	cache, err := New(append([]OptionFunc{DatabaseAdapter(driver.DBType)}, cfg.Options()...)...)
 	NoError(t, err)
 	NoError(t, cache.Flush())
-	conn, err := sql.Open("mysql", "root:@tcp(localhost:3306)/rapidash?parseTime=true")
+	conn, err := sql.Open(driver.Name, driver.Source)
 	NoError(t, err)
 	NoError(t, cache.WarmUp(conn, userLoginType(), false))
 	t.Run("create new records", func(t *testing.T) {
@@ -43,7 +43,7 @@ func TestConfig(t *testing.T) {
 		tx, err := cache.Begin(txConn)
 		NoError(t, err)
 		for i := 1001; i <= 1005; i++ {
-			builder := NewQueryBuilder("user_logins").
+			builder := NewQueryBuilder("user_logins", driver.Adapter).
 				Eq("user_id", uint64(i)).
 				Eq("user_session_id", uint64(i))
 			var foundUserLogin UserLogin
@@ -173,7 +173,7 @@ func TestConfig(t *testing.T) {
 
 					t.Run("should retrieve each handled data", func(t *testing.T) {
 						key := fmt.Sprintf("key_%d", time.Now().UnixNano())
-						var resultFirst  int
+						var resultFirst int
 						var resultSecond int
 						expectFirst := 1
 						expectSecond := 2
@@ -213,7 +213,7 @@ func TestConfig(t *testing.T) {
 				t.Run("implicit cache control", func(t *testing.T) {
 					t.Run("should retrieve each handled data", func(t *testing.T) {
 						key := fmt.Sprintf("key_%d", time.Now().UnixNano())
-						var resultFirst  int
+						var resultFirst int
 						var resultSecond int
 						expectFirst := 1
 						expectSecond := 2

database/database.go

@@ -0,0 +1,124 @@
+package database
+
+import (
+	"database/sql"
+
+	"go.knocknote.io/rapidash/database/mysql"
+	"go.knocknote.io/rapidash/database/postgres"
+)
+
+type DBType int
+
+const (
+	None DBType = iota
+	MySQL
+	Postgres
+)
+
+const (
+	mysqlPlugin    = "mysql"
+	postgresPlugin = "postgres"
+)
+
+type Database interface {
+	database
+	Placeholder(int) string
+	Placeholders(int, int) string
+}
+
+type database interface {
+	TableDDL(*sql.DB, string) (string, error)
+	Quote(string) string
+	SupportLastInsertID() bool
+}
+
+type QueryHelper struct {
+	count   int
+	adapter *adapter
+}
+
+func (qh *QueryHelper) DBType() DBType {
+	return qh.adapter.DBType
+}
+
+func (qh *QueryHelper) Placeholder() string {
+	qh.count++
+	return qh.adapter.Placeholder(qh.count)
+}
+
+func (qh *QueryHelper) Placeholders(n int) string {
+	start := qh.count + 1
+	end := start + n - 1
+	qh.count += n
+	return qh.adapter.Placeholders(start, end)
+}
+
+func (qh *QueryHelper) Quote(str string) string {
+	return qh.adapter.Quote(str)
+}
+
+func (qh *QueryHelper) SupportLastInsertID() bool {
+	return qh.adapter.SupportLastInsertID()
+}
+
+func (qh *QueryHelper) ClearCount() {
+	qh.count = 0
+}
+
+type Adapter interface {
+	database
+	QueryHelper() *QueryHelper
+}
+
+type adapter struct {
+	DBType DBType
+	Database
+}
+
+func (d *adapter) QueryHelper() *QueryHelper {
+	return &QueryHelper{
+		count:   0,
+		adapter: d,
+	}
+}
+
+func NewAdapter() *adapter {
+	drivers := sql.Drivers()
+	if len(drivers) == 0 {
+		return nil
+	}
+	dbType := toDBType(drivers[0])
+	return &adapter{
+		DBType:   dbType,
+		Database: NewDatabase(dbType),
+	}
+}
+
+func NewAdapterWithDBType(dbType DBType) *adapter {
+	return &adapter{
+		DBType:   dbType,
+		Database: NewDatabase(dbType),
+	}
+}
+
+func NewDatabase(dbType DBType) Database {
+	switch dbType {
+	case MySQL:
+		return &mysql.MySQL{}
+	case Postgres:
+		return &postgres.Postgres{}
+	}
+	return nil
+}
+
+func toDBType(pluginName string) DBType {
+	switch pluginName {
+	case mysqlPlugin:
+		return MySQL
+	case postgresPlugin:
+		return Postgres
+	}
+	return None
+}
+
+var _ Adapter = (*adapter)(nil)

database/mysql/mysql.go

@@ -0,0 +1,46 @@
+package mysql
+
+import (
+	"database/sql"
+	"fmt"
+	"strings"
+
+	"golang.org/x/xerrors"
+)
+
+type MySQL struct{}
+
+func (ms *MySQL) TableDDL(conn *sql.DB, tableName string) (string, error) {
+	var (
+		tbl string
+		ddl string
+	)
+	if err := conn.QueryRow(fmt.Sprintf("SHOW CREATE TABLE `%s`", tableName)).Scan(&tbl, &ddl); err != nil {
+		return "", xerrors.Errorf("failed to execute 'SHOW CREATE TABLE `%s`': %w", tableName, err)
+	}
+	return ddl, nil
+}
+
+func (ms *MySQL) Placeholder(_ int) string {
+	return "?"
+}
+
+func (ms *MySQL) Placeholders(start, end int) string {
+	sb := &strings.Builder{}
+	sb.Grow((len(ms.Placeholder(0)) + 1) * (end - start + 1))
+	for i := start; i <= end; i++ {
+		sb.WriteString(ms.Placeholder(0))
+		if i < end {
+			sb.WriteString(",")
+		}
+	}
+	return sb.String()
+}
+
+func (ms *MySQL) Quote(s string) string {
+	return fmt.Sprintf("`%s`", s)
+}
+
+func (ms *MySQL) SupportLastInsertID() bool {
+	return true
+}