登录功能完成

pkg/dao/dao.go

@@ -1 +1,76 @@
 package dao
+
+import (
+	"fmt"
+	"github.com/sirupsen/logrus"
+	"gorm.io/driver/postgres"
+	"gorm.io/gorm"
+	"gorm.io/gorm/logger"
+	"mc-client-updater-server/pkg/conf"
+	"mc-client-updater-server/pkg/dao/entity"
+	"mc-client-updater-server/pkg/log"
+	"sync"
+	"time"
+)
+
+type LogrusWriter struct {
+	log *logrus.Logger
+}
+
+func (w *LogrusWriter) Printf(format string, v ...interface{}) {
+	logStr := fmt.Sprintf(format, v...)
+	w.log.WithField("method", "GORM").Warn(logStr)
+}
+
+func NewLogger() *LogrusWriter {
+	return &LogrusWriter{log: log.Logger}
+}
+
+var (
+	once sync.Once
+	db   *gorm.DB
+)
+
+func DB() *gorm.DB {
+	once.Do(NewDBConn)
+	return db
+}
+
+func NewDBConn() {
+	var err error
+	dsn := fmt.Sprintf(
+		"host=%s user=%s password=%s dbname=%s port=%d sslmode=disable TimeZone=Asia/Shanghai",
+		conf.Conf.Database.Host,
+		conf.Conf.Database.User,
+		conf.Conf.Database.Password,
+		conf.Conf.Database.DB,
+		conf.Conf.Database.Port,
+	)
+
+	// >= 1s SQL慢查询
+	slowLogger := logger.New(NewLogger(), logger.Config{
+		SlowThreshold: time.Second * 1,
+		LogLevel:      logger.Warn,
+	})
+	db, err = gorm.Open(postgres.Open(dsn), &gorm.Config{
+		Logger: slowLogger,
+	})
+	if err != nil {
+		log.Logger.Fatal("创建数据库连接失败：", err)
+	}
+
+	migrate()
+}
+
+func migrate() {
+	var err error
+	err = db.AutoMigrate(&entity.User{})
+	err = db.AutoMigrate(&entity.Instance{})
+	err = db.AutoMigrate(&entity.Update{})
+	err = db.AutoMigrate(&entity.Grant{})
+	err = db.AutoMigrate(&entity.Token{})
+
+	if err != nil {
+		log.Logger.Fatal("关联数据表失败：", err)
+	}
+}

pkg/dao/entity/grant.go

@@ -0,0 +1,13 @@
+package entity
+
+import (
+	"gorm.io/gorm"
+)
+
+// Grant Grant是授权给实例的，给予实例访问权限
+type Grant struct {
+	gorm.Model `json:"model"`
+	Token      string `gorm:"unique;not null" json:"token,omitempty"`
+	TTL        int    `gorm:"not null;default:0" json:"ttl,omitempty"`
+	GrantTo    uint   `gorm:"not null;default:0;comment:instances(id) 授权给实例，0表示无指定（所有）" json:"grant_to,omitempty"`
+}

pkg/dao/entity/instance.go

@@ -0,0 +1,11 @@
+package entity
+
+import (
+	"gorm.io/gorm"
+)
+
+type Instance struct {
+	gorm.Model `json:"model"`
+	Name       string `gorm:"unique;not null" json:"name,omitempty"`
+	UpdateURL  string `gorm:"column:update_url;not null;default:'';comment:更新URL，未指定使用默认" json:"update_url,omitempty"`
+}

pkg/dao/entity/token.go

@@ -0,0 +1,11 @@
+package entity
+
+import "gorm.io/gorm"
+
+// Token Token是授权给用户的，给予用户登录权限
+type Token struct {
+	gorm.Model `json:"model"`
+	Token      string `gorm:"unique;not null" json:"token,omitempty"`
+	GrantTo    string `gorm:"index;not null;default:''" json:"grant_to,omitempty"`
+	TTL        int    `gorm:"not null;default:0" json:"ttl,omitempty"`
+}

pkg/dao/entity/update.go

@@ -0,0 +1,10 @@
+package entity
+
+import "gorm.io/gorm"
+
+type Update struct {
+	gorm.Model `json:"model"`
+	HashID     string `gorm:"index;not null" json:"hash_id,omitempty"`
+	Comment    string `gorm:"not null;default:'';comment:更新内容或注释" json:"comment,omitempty"`
+	Changes    string `gorm:"not null;comment:更改的文件列表，逗号分隔，引用files(hash_id)" json:"changes,omitempty"`
+}

pkg/dao/entity/user.go

@@ -3,9 +3,8 @@ package entity
 import "gorm.io/gorm"
 
 type User struct {
-	gorm.Model
-	Username string `gorm:"unique;not null"`
-	Password string `gorm:"not null"`
-	Instance string `gorm:"not null;default:''"`
-	Role     string `gorm:"not null;default:0"`
+	gorm.Model `json:"model"`
+	Username   string `gorm:"unique;not null" json:"username,omitempty"`
+	Password   string `gorm:"not null" json:"password,omitempty"`
+	Roles      string `gorm:"not null;default:''" json:"roles,omitempty"`
 }