feat: 优化web

pkg/db/db.go

@@ -0,0 +1,27 @@
+package db
+
+import (
+	"giter.top/smart/pkg/config"
+	"gorm.io/gorm"
+)
+
+func NewDB(cfg *config.Config) *gorm.DB {
+	driver := cfg.Data.Database.Driver
+	var db *gorm.DB
+	var err error
+	switch driver {
+	case "mysql":
+		// db, err = NewMySQLDB(cfg)
+	case "postgres":
+		db, err = NewPgSQLDB(cfg)
+	case "sqlite":
+		// return NewSQLiteDB(cfg)
+	default:
+		panic("unsupported driver")
+	}
+
+	if err != nil {
+		panic(err)
+	}
+	return db
+}