.

2026-01-14 14:17:38 +08:00
commit f1e2ff6563
126 changed files with 13636 additions and 0 deletions
--- a/internal/storage/repository/sqlite_repo.go
+++ b/internal/storage/repository/sqlite_repo.go
@@ -0,0 +1,136 @@
+package repository
+
+import (
+	"fmt"
+	"gorm.io/gorm"
+	"ssq-desk/internal/database"
+	"ssq-desk/internal/storage/models"
+	"time"
+)
+
+// SQLiteSsqRepository SQLite 实现
+type SQLiteSsqRepository struct {
+	db *gorm.DB
+}
+
+// NewSQLiteSsqRepository 创建 SQLite 仓库
+func NewSQLiteSsqRepository() (SsqRepository, error) {
+	db := database.GetSQLite()
+	if db == nil {
+		return nil, fmt.Errorf("SQLite 数据库未初始化或初始化失败")
+	}
+
+	// 自动迁移表结构（数据库初始化时已迁移，这里再次确保）
+	err := db.AutoMigrate(&models.SsqHistory{})
+	if err != nil {
+		return nil, fmt.Errorf("SQLite 表迁移失败: %v", err)
+	}
+
+	return &SQLiteSsqRepository{db: db}, nil
+}
+
+// FindAll 查询所有历史数据
+func (r *SQLiteSsqRepository) FindAll() ([]models.SsqHistory, error) {
+	var histories []models.SsqHistory
+	err := r.db.Order("issue_number DESC").Find(&histories).Error
+	return histories, err
+}
+
+// FindByIssue 根据期号查询
+func (r *SQLiteSsqRepository) FindByIssue(issueNumber string) (*models.SsqHistory, error) {
+	var history models.SsqHistory
+	err := r.db.Where("issue_number = ?", issueNumber).First(&history).Error
+	if err == gorm.ErrRecordNotFound {
+		return nil, nil
+	}
+	return &history, err
+}
+
+// FindByRedBalls 根据红球查询（支持部分匹配）
+func (r *SQLiteSsqRepository) FindByRedBalls(redBalls []int) ([]models.SsqHistory, error) {
+	if len(redBalls) == 0 {
+		return r.FindAll()
+	}
+
+	var histories []models.SsqHistory
+	query := r.db
+
+	// 构建查询条件：红球在输入的红球列表中
+	for _, ball := range redBalls {
+		query = query.Where("red_ball_1 = ? OR red_ball_2 = ? OR red_ball_3 = ? OR red_ball_4 = ? OR red_ball_5 = ? OR red_ball_6 = ?",
+			ball, ball, ball, ball, ball, ball)
+	}
+
+	err := query.Order("issue_number DESC").Find(&histories).Error
+	return histories, err
+}
+
+// FindByRedAndBlue 根据红球和蓝球查询
+func (r *SQLiteSsqRepository) FindByRedAndBlue(redBalls []int, blueBall int, blueBallRange []int) ([]models.SsqHistory, error) {
+	var histories []models.SsqHistory
+	query := r.db
+
+	// 红球条件
+	if len(redBalls) > 0 {
+		for _, ball := range redBalls {
+			query = query.Where("red_ball_1 = ? OR red_ball_2 = ? OR red_ball_3 = ? OR red_ball_4 = ? OR red_ball_5 = ? OR red_ball_6 = ?",
+				ball, ball, ball, ball, ball, ball)
+		}
+	}
+
+	// 蓝球条件
+	if blueBall > 0 {
+		query = query.Where("blue_ball = ?", blueBall)
+	} else if len(blueBallRange) > 0 {
+		query = query.Where("blue_ball IN ?", blueBallRange)
+	}
+
+	err := query.Order("issue_number DESC").Find(&histories).Error
+	return histories, err
+}
+
+// Create 创建记录
+func (r *SQLiteSsqRepository) Create(history *models.SsqHistory) error {
+	now := time.Now()
+	if history.CreatedAt.IsZero() {
+		history.CreatedAt = now
+	}
+	if history.UpdatedAt.IsZero() {
+		history.UpdatedAt = now
+	}
+	return r.db.Create(history).Error
+}
+
+// BatchCreate 批量创建
+func (r *SQLiteSsqRepository) BatchCreate(histories []models.SsqHistory) error {
+	if len(histories) == 0 {
+		return nil
+	}
+	now := time.Now()
+	for i := range histories {
+		if histories[i].CreatedAt.IsZero() {
+			histories[i].CreatedAt = now
+		}
+		if histories[i].UpdatedAt.IsZero() {
+			histories[i].UpdatedAt = now
+		}
+	}
+	return r.db.CreateInBatches(histories, 100).Error
+}
+
+// GetLatestIssue 获取最新期号
+func (r *SQLiteSsqRepository) GetLatestIssue() (string, error) {
+	var history models.SsqHistory
+	err := r.db.Order("issue_number DESC").First(&history).Error
+	if err == gorm.ErrRecordNotFound {
+		return "", nil
+	}
+	return history.IssueNumber, err
+}
+
+// Count 统计总数
+func (r *SQLiteSsqRepository) Count() (int64, error) {
+	var count int64
+	err := r.db.Model(&models.SsqHistory{}).Count(&count).Error
+	return count, err
+}