sql db: add sqlite driver

golangid · May 18, 2023 · 7d5b660 · 7d5b660
1 parent 991634a
commit 7d5b660
Show file tree

Hide file tree

Showing 6 changed files with 9 additions and 7 deletions.
diff --git a/cmd/candi/cli_parser.go b/cmd/candi/cli_parser.go
@@ -210,7 +210,8 @@ stageSelectDependencies:
 	stageSelectSQLDriver:
 		cmdInput = readInput("Please select SQL database driver (choose one)\n" +
 			"1) Postgres\n" +
-			"2) MySQL")
+			"2) MySQL\n" +
+			"3) SQLite3")
 		srvConfig.SQLDriver, ok = sqlDrivers[cmdInput]
 		if !ok {
 			fmt.Printf(RedFormat, "Invalid option, try again")

diff --git a/cmd/candi/template_cmd.go b/cmd/candi/template_cmd.go
@@ -9,7 +9,8 @@ import (
 	"fmt"
 	"runtime/debug"{{if eq .SQLDriver "postgres"}}
 	_ "github.com/lib/pq"{{else if eq .SQLDriver "mysql"}}
-	_ "github.com/go-sql-driver/mysql"{{end}}
+	_ "github.com/go-sql-driver/mysql"{{else if eq .SQLDriver "sqlite3"}}
+	_ "github.com/mattn/go-sqlite3"{{end}}
 
 	"{{.LibraryName}}/codebase/app"
 	"{{.LibraryName}}/config"

diff --git a/cmd/candi/template_delivery_worker.go b/cmd/candi/template_delivery_worker.go
@@ -200,7 +200,7 @@ func (h *TaskQueueHandler) handleTask{{upper (camel .ModuleName)}}(eventContext
 	// h.uc.SomethingUsecase()
 
 	return &candishared.ErrorRetrier{
-		Delay:   10 * time.Second,
+		Delay:   2 * time.Second,
 		Message: "Error retry",
 	}
 }

diff --git a/cmd/candi/template_env.go b/cmd/candi/template_env.go
@@ -37,11 +37,11 @@ MONGODB_HOST_READ=mongodb://user:pass@localhost:27017/{{.ServiceName}}?authSourc
 
 SQL_DB_READ_DSN={{ if .SQLDeps }}{{.SQLDriver}}://` +
 	"{{if eq .SQLDriver \"postgres\"}}user:pass@localhost:5432/db_name?sslmode=disable&TimeZone=Asia/Jakarta{{else if eq .SQLDriver \"mysql\"}}" +
-	"root:pass@tcp(127.0.0.1:3306)/db_name{{end}}" +
+	"root:pass@tcp(127.0.0.1:3306)/db_name{{else if eq .SQLDriver \"sqlite3\"}}database.db{{end}}" +
 	`{{ end }}
 SQL_DB_WRITE_DSN={{ if .SQLDeps }}{{.SQLDriver}}://` +
 	"{{if eq .SQLDriver \"postgres\"}}user:pass@localhost:5432/db_name?sslmode=disable&TimeZone=Asia/Jakarta{{else if eq .SQLDriver \"mysql\"}}" +
-	"root:pass@tcp(127.0.0.1:3306)/db_name{{end}}" +
+	"root:pass@tcp(127.0.0.1:3306)/db_name{{else if eq .SQLDriver \"sqlite3\"}}database.db{{end}}" +
 	`{{ end }}
 {{if .ArangoDeps}}
 ARANGODB_HOST_WRITE=http://user:pass@localhost:8529/{{.ServiceName}}

diff --git a/cmd/candi/types.go b/cmd/candi/types.go
@@ -26,7 +26,7 @@ var (
 		"1": RedisDeps, "2": SqldbDeps, "3": MongodbDeps, "4": ArangodbDeps,
 	}
 	sqlDrivers = map[string]string{
-		"1": "postgres", "2": "mysql",
+		"1": "postgres", "2": "mysql", "3": "sqlite3",
 	}
 	optionYesNo = map[string]bool{"y": true, "n": false}
 	licenseMap  = map[string]string{

diff --git a/config/database/sql.go b/config/database/sql.go
@@ -60,7 +60,7 @@ func ConnectSQLDatabase(dsn string) *sql.DB {
 	driverName := sqlDriver
 
 	switch sqlDriver {
-	case "mysql":
+	case "mysql", "sqlite3":
 		dsn = conn
 
 	case "sqlserver":