feat 存档

db/model/user.go

@@ -0,0 +1,14 @@
+package model
+
+type List struct {
+	ID        int32  `json:"id" gorm:"column:id;primaryKey"`      // 主键
+	Name      string `json:"name" gorm:"column:name;uniqueIndex"` // 小说名称
+	Status    int32  `json:"status" gorm:"column:status"`         // 状态 0 未知 1 仿写中 2 仿写完成
+	Imitation string `json:"imitation" gorm:"column:imitation"`   // 仿文
+	AI        string `json:"ai" gorm:"column:ai"`                 // AI对话记录
+	Token     string `json:"token" gorm:"column:token"`           // Token消耗
+}
+
+func (*List) TableName() string {
+	return "list"
+}

db/sqlite/sqlite.go

@@ -0,0 +1,51 @@
+package sqlite
+
+import (
+	"fiction/db/model"
+	"fmt"
+	"gorm.io/driver/sqlite"
+	"gorm.io/gorm"
+	"gorm.io/gorm/logger"
+	"log"
+	"time"
+)
+
+var DB *gorm.DB
+
+func Init(dsn string) error {
+	fmt.Println("init SQLite")
+	db, err := gorm.Open(sqlite.Open(dsn), &gorm.Config{
+		TranslateError: true,
+		Logger: logger.New(
+			log.New(log.Writer(), "\r\n", log.LstdFlags),
+			logger.Config{
+				SlowThreshold: 10 * time.Millisecond,
+				Colorful:      true,
+				LogLevel:      logger.Info,
+			},
+		),
+	})
+	if err != nil {
+		return err
+	}
+	DB = db
+	if err = initDB(); err != nil {
+		return err
+	}
+	return nil
+}
+
+func initDB() error {
+	tableArr := []interface{}{
+		&model.List{},
+	}
+	for _, table := range tableArr {
+		if err := DB.AutoMigrate(table); err != nil {
+			return err
+		}
+	}
+	return nil
+}
+
+func Stop() {
+}