feat(database): add mysql support (#39)

This commit is contained in:
tardc
2021-12-23 01:34:53 +08:00
committed by GitHub
parent 8e96962791
commit 9d7a5b4598
3 changed files with 27 additions and 1 deletions
+10 -1
View File
@@ -1,6 +1,8 @@
package server
import (
"github.com/pkg/errors"
"strings"
"sync"
"github.com/gin-gonic/gin"
@@ -60,7 +62,14 @@ func initDatabase(dsn string) {
Level: log.LevelInfo,
})
err := database.InitDB(database.Sqlite, dsn)
var err error
if strings.Contains(dsn, "@tcp") {
err = database.InitDB(database.Mysql, dsn)
} else if strings.Contains(dsn, ".db") {
err = database.InitDB(database.Sqlite, dsn)
} else {
err = errors.New("unsupported database")
}
if err != nil {
log.Fatal(err.Error())
}