go_gameshield/internal/repository/log.go

package repository

import (
	"context"
	"fmt"
	"time"

	"github.com/go-nunu/nunu-layout-advanced/internal/model"
)

type LogRepository interface {
	GetLog(ctx context.Context, id int64) (*model.Log, error)
	GetLogsByTimeRange(ctx context.Context, start, end *time.Time) ([]*model.Log, error)
	AddLog(ctx context.Context, log *model.Log) error
	EditLog(ctx context.Context, log *model.Log) error
}

func NewLogRepository(
	repository *Repository,
) LogRepository {
	return &logRepository{
		Repository: repository,
	}
}

type logRepository struct {
	*Repository
}

func (r *logRepository) GetLog(ctx context.Context, id int64) (*model.Log, error) {
	var log model.Log

	
	// 获取可能的表名(查询最近3个月)
	existingTables := r.Manager.GetTableNamesWithExistenceCheck(r.DBWithName(ctx, "admin"), "log", nil, nil)
	
	// 在各个分表中查找
	for _, tableName := range existingTables {
		err := r.DBWithName(ctx, "admin").Table(tableName).Where("id = ?", id).First(&log).Error
		if err == nil {
			log.SetTableName(tableName)
			return &log, nil
		}
	}
	
	return nil, fmt.Errorf("未找到ID为 %d 的日志记录", id)
}

func (r *logRepository) GetLogsByTimeRange(ctx context.Context, start, end *time.Time) ([]*model.Log, error) {
	var logs []*model.Log
	

	
	// 检查存在的表
	existingTables := r.Manager.GetTableNamesWithExistenceCheck(r.DBWithName(ctx, "admin"), "log", start, end)
	
	if len(existingTables) == 0 {
		return logs, nil // 没有分表，返回空结果
	}
	
	// 联合查询所有分表
	for _, tableName := range existingTables {
		var tableLogs []*model.Log
		query := r.DBWithName(ctx, "admin").Table(tableName)
		
		// 添加时间范围过滤
		if start != nil {
			query = query.Where("created_at >= ?", *start)
		}
		if end != nil {
			query = query.Where("created_at <= ?", *end)
		}
		
		err := query.Find(&tableLogs).Error
		if err != nil {
			return nil, err
		}
		
		// 设置表名
		for _, log := range tableLogs {
			log.SetTableName(tableName)
		}
		
		logs = append(logs, tableLogs...)
	}
	
	return logs, nil
}

func (r *logRepository) AddLog(ctx context.Context, log *model.Log) error {
	// 设置创建时间
	if log.CreatedAt.IsZero() {
		log.CreatedAt = time.Now()
	}

	
	// 获取最优的写入表(自动根据表名获取阈值)
	tableName, err := r.Manager.GetOptimalWriteTable(ctx, r.DBWithName(ctx, "admin"), log)
	if err != nil {
		return fmt.Errorf("获取写入表失败: %v", err)
	}
	
	log.SetTableName(tableName)
	
	// 确保表存在
	err = r.Manager.EnsureTableExists(ctx, r.DBWithName(ctx, "admin"), tableName, &model.Log{})
	if err != nil {
		return err
	}
	
	// 写入数据
	return r.DBWithName(ctx, "admin").Table(tableName).Create(log).Error
}

func (r *logRepository) EditLog(ctx context.Context, log *model.Log) error {
	// 如果已经指定了表名，直接更新该表
	if log.TableName() != "log" {
		return r.DBWithName(ctx, "admin").Table(log.TableName()).Updates(log).Error
	}

	
	// 确定表名
	tableName := r.Manager.GetWriteTableName(log)
	log.SetTableName(tableName)
	
	// 确保表存在
	err := r.Manager.EnsureTableExists(ctx, r.DBWithName(ctx, "admin"), tableName, &model.Log{})
	if err != nil {
		return err
	}
	
	return r.DBWithName(ctx, "admin").Table(tableName).Updates(log).Error
}