go-trustlog/api/persistence/repository.go

package persistence

import (
	"context"
	"database/sql"
	"fmt"
	"time"

	"go.yandata.net/iod/iod/go-trustlog/api/logger"
	"go.yandata.net/iod/iod/go-trustlog/api/model"
)

// OperationRepository 操作记录数据库仓储接口
type OperationRepository interface {
	// Save 保存操作记录到数据库
	Save(ctx context.Context, op *model.Operation, status TrustlogStatus) error
	// SaveTx 在事务中保存操作记录
	SaveTx(ctx context.Context, tx *sql.Tx, op *model.Operation, status TrustlogStatus) error
	// UpdateStatus 更新操作记录的存证状态
	UpdateStatus(ctx context.Context, opID string, status TrustlogStatus) error
	// UpdateStatusTx 在事务中更新操作记录的存证状态
	UpdateStatusTx(ctx context.Context, tx *sql.Tx, opID string, status TrustlogStatus) error
	// FindByID 根据 OpID 查询操作记录
	FindByID(ctx context.Context, opID string) (*model.Operation, TrustlogStatus, error)
	// FindUntrustlogged 查询未存证的操作记录（用于重试机制）
	FindUntrustlogged(ctx context.Context, limit int) ([]*model.Operation, error)
}

// CursorRepository 游标仓储接口（Key-Value 模式）
type CursorRepository interface {
	// GetCursor 获取游标值
	GetCursor(ctx context.Context, cursorKey string) (string, error)
	// UpdateCursor 更新游标值
	UpdateCursor(ctx context.Context, cursorKey string, cursorValue string) error
	// UpdateCursorTx 在事务中更新游标值
	UpdateCursorTx(ctx context.Context, tx *sql.Tx, cursorKey string, cursorValue string) error
	// InitCursor 初始化游标（如果不存在）
	InitCursor(ctx context.Context, cursorKey string, initialValue string) error
}

// RetryRepository 重试仓储接口
type RetryRepository interface {
	// AddRetry 添加重试记录
	AddRetry(ctx context.Context, opID string, errorMsg string, nextRetryAt time.Time) error
	// AddRetryTx 在事务中添加重试记录
	AddRetryTx(ctx context.Context, tx *sql.Tx, opID string, errorMsg string, nextRetryAt time.Time) error
	// IncrementRetry 增加重试次数
	IncrementRetry(ctx context.Context, opID string, errorMsg string, nextRetryAt time.Time) error
	// MarkAsDeadLetter 标记为死信
	MarkAsDeadLetter(ctx context.Context, opID string, errorMsg string) error
	// FindPendingRetries 查找待重试的记录
	FindPendingRetries(ctx context.Context, limit int) ([]RetryRecord, error)
	// DeleteRetry 删除重试记录（成功后清理）
	DeleteRetry(ctx context.Context, opID string) error
}

// RetryRecord 重试记录
type RetryRecord struct {
	OpID         string
	RetryCount   int
	RetryStatus  RetryStatus
	LastRetryAt  *time.Time
	NextRetryAt  *time.Time
	ErrorMessage string
	CreatedAt    time.Time
	UpdatedAt    time.Time
}

// operationRepository 操作记录仓储实现
type operationRepository struct {
	db     *sql.DB
	logger logger.Logger
}

// NewOperationRepository 创建操作记录仓储
func NewOperationRepository(db *sql.DB, log logger.Logger) OperationRepository {
	return &operationRepository{
		db:     db,
		logger: log,
	}
}

func (r *operationRepository) Save(ctx context.Context, op *model.Operation, status TrustlogStatus) error {
	return r.SaveTx(ctx, nil, op, status)
}

func (r *operationRepository) SaveTx(ctx context.Context, tx *sql.Tx, op *model.Operation, status TrustlogStatus) error {
	query := `
		INSERT INTO operation (
			op_id, op_actor, doid, producer_id, 
			request_body_hash, response_body_hash, 
			op_source, op_type, do_prefix, do_repository,
			client_ip, server_ip, trustlog_status, timestamp
		) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
	`

	var reqHash, respHash, clientIP, serverIP sql.NullString
	if op.RequestBodyHash != nil {
		reqHash = sql.NullString{String: *op.RequestBodyHash, Valid: true}
	}
	if op.ResponseBodyHash != nil {
		respHash = sql.NullString{String: *op.ResponseBodyHash, Valid: true}
	}
	if op.ClientIP != nil {
		clientIP = sql.NullString{String: *op.ClientIP, Valid: true}
	}
	if op.ServerIP != nil {
		serverIP = sql.NullString{String: *op.ServerIP, Valid: true}
	}

	args := []interface{}{
		op.OpID,
		op.OpActor,
		op.Doid,
		op.ProducerID,
		reqHash,
		respHash,
		string(op.OpSource),
		string(op.OpType),
		op.DoPrefix,
		op.DoRepository,
		clientIP,
		serverIP,
		string(status),
		op.Timestamp,
	}

	var err error
	if tx != nil {
		_, err = tx.ExecContext(ctx, query, args...)
	} else {
		_, err = r.db.ExecContext(ctx, query, args...)
	}

	if err != nil {
		r.logger.ErrorContext(ctx, "failed to save operation",
			"opID", op.OpID,
			"error", err,
		)
		return fmt.Errorf("failed to save operation: %w", err)
	}

	r.logger.DebugContext(ctx, "operation saved to database",
		"opID", op.OpID,
		"status", status,
	)
	return nil
}

func (r *operationRepository) UpdateStatus(ctx context.Context, opID string, status TrustlogStatus) error {
	return r.UpdateStatusTx(ctx, nil, opID, status)
}

func (r *operationRepository) UpdateStatusTx(ctx context.Context, tx *sql.Tx, opID string, status TrustlogStatus) error {
	query := `UPDATE operation SET trustlog_status = ? WHERE op_id = ?`

	var err error
	if tx != nil {
		_, err = tx.ExecContext(ctx, query, string(status), opID)
	} else {
		_, err = r.db.ExecContext(ctx, query, string(status), opID)
	}

	if err != nil {
		r.logger.ErrorContext(ctx, "failed to update operation status",
			"opID", opID,
			"status", status,
			"error", err,
		)
		return fmt.Errorf("failed to update operation status: %w", err)
	}

	r.logger.DebugContext(ctx, "operation status updated",
		"opID", opID,
		"status", status,
	)
	return nil
}

func (r *operationRepository) FindByID(ctx context.Context, opID string) (*model.Operation, TrustlogStatus, error) {
	query := `
		SELECT 
			op_id, op_actor, doid, producer_id,
			request_body_hash, response_body_hash,
			op_source, op_type, do_prefix, do_repository,
			client_ip, server_ip, trustlog_status, timestamp
		FROM operation
		WHERE op_id = ?
	`

	var op model.Operation
	var statusStr string
	var reqHash, respHash, clientIP, serverIP sql.NullString

	err := r.db.QueryRowContext(ctx, query, opID).Scan(
		&op.OpID,
		&op.OpActor,
		&op.Doid,
		&op.ProducerID,
		&reqHash,
		&respHash,
		&op.OpSource,
		&op.OpType,
		&op.DoPrefix,
		&op.DoRepository,
		&clientIP,
		&serverIP,
		&statusStr,
		&op.Timestamp,
	)

	if err == sql.ErrNoRows {
		return nil, "", fmt.Errorf("operation not found: %s", opID)
	}
	if err != nil {
		r.logger.ErrorContext(ctx, "failed to find operation",
			"opID", opID,
			"error", err,
		)
		return nil, "", fmt.Errorf("failed to find operation: %w", err)
	}

	if reqHash.Valid {
		op.RequestBodyHash = &reqHash.String
	}
	if respHash.Valid {
		op.ResponseBodyHash = &respHash.String
	}
	if clientIP.Valid {
		op.ClientIP = &clientIP.String
	}
	if serverIP.Valid {
		op.ServerIP = &serverIP.String
	}

	return &op, TrustlogStatus(statusStr), nil
}

func (r *operationRepository) FindUntrustlogged(ctx context.Context, limit int) ([]*model.Operation, error) {
	query := `
		SELECT 
			op_id, op_actor, doid, producer_id,
			request_body_hash, response_body_hash,
			op_source, op_type, do_prefix, do_repository,
			client_ip, server_ip, timestamp
		FROM operation
		WHERE trustlog_status = ?
		ORDER BY timestamp ASC
		LIMIT ?
	`

	rows, err := r.db.QueryContext(ctx, query, string(StatusNotTrustlogged), limit)
	if err != nil {
		r.logger.ErrorContext(ctx, "failed to find untrustlogged operations",
			"error", err,
		)
		return nil, fmt.Errorf("failed to find untrustlogged operations: %w", err)
	}
	defer rows.Close()

	var operations []*model.Operation
	for rows.Next() {
		var op model.Operation
		var reqHash, respHash, clientIP, serverIP sql.NullString

		err := rows.Scan(
			&op.OpID,
			&op.OpActor,
			&op.Doid,
			&op.ProducerID,
			&reqHash,
			&respHash,
			&op.OpSource,
			&op.OpType,
			&op.DoPrefix,
			&op.DoRepository,
			&clientIP,
			&serverIP,
			&op.Timestamp,
		)
		if err != nil {
			r.logger.ErrorContext(ctx, "failed to scan operation row",
				"error", err,
			)
			return nil, fmt.Errorf("failed to scan operation row: %w", err)
		}

		if reqHash.Valid {
			op.RequestBodyHash = &reqHash.String
		}
		if respHash.Valid {
			op.ResponseBodyHash = &respHash.String
		}
		if clientIP.Valid {
			op.ClientIP = &clientIP.String
		}
		if serverIP.Valid {
			op.ServerIP = &serverIP.String
		}

		operations = append(operations, &op)
	}

	if err := rows.Err(); err != nil {
		return nil, fmt.Errorf("error iterating operation rows: %w", err)
	}

	return operations, nil
}

// cursorRepository 游标仓储实现
type cursorRepository struct {
	db     *sql.DB
	logger logger.Logger
}

// NewCursorRepository 创建游标仓储
func NewCursorRepository(db *sql.DB, log logger.Logger) CursorRepository {
	return &cursorRepository{
		db:     db,
		logger: log,
	}
}

// GetCursor 获取游标值（Key-Value 模式）
func (r *cursorRepository) GetCursor(ctx context.Context, cursorKey string) (string, error) {
	query := `SELECT cursor_value FROM trustlog_cursor WHERE cursor_key = ?`

	var cursorValue string
	err := r.db.QueryRowContext(ctx, query, cursorKey).Scan(&cursorValue)
	if err == sql.ErrNoRows {
		r.logger.DebugContext(ctx, "cursor not found",
			"cursorKey", cursorKey,
		)
		return "", nil
	}
	if err != nil {
		r.logger.ErrorContext(ctx, "failed to get cursor",
			"cursorKey", cursorKey,
			"error", err,
		)
		return "", fmt.Errorf("failed to get cursor: %w", err)
	}

	return cursorValue, nil
}

// UpdateCursor 更新游标值
func (r *cursorRepository) UpdateCursor(ctx context.Context, cursorKey string, cursorValue string) error {
	return r.UpdateCursorTx(ctx, nil, cursorKey, cursorValue)
}

// UpdateCursorTx 在事务中更新游标值（使用 UPSERT）
func (r *cursorRepository) UpdateCursorTx(ctx context.Context, tx *sql.Tx, cursorKey string, cursorValue string) error {
	// 使用 UPSERT 语法（适配不同数据库）
	query := `
		INSERT INTO trustlog_cursor (cursor_key, cursor_value, last_updated_at)
		VALUES (?, ?, ?)
		ON CONFLICT (cursor_key) DO UPDATE SET
			cursor_value = excluded.cursor_value,
			last_updated_at = excluded.last_updated_at
	`

	var err error
	now := time.Now()
	if tx != nil {
		_, err = tx.ExecContext(ctx, query, cursorKey, cursorValue, now)
	} else {
		_, err = r.db.ExecContext(ctx, query, cursorKey, cursorValue, now)
	}

	if err != nil {
		r.logger.ErrorContext(ctx, "failed to update cursor",
			"cursorKey", cursorKey,
			"error", err,
		)
		return fmt.Errorf("failed to update cursor: %w", err)
	}

	r.logger.DebugContext(ctx, "cursor updated",
		"cursorKey", cursorKey,
		"cursorValue", cursorValue,
	)
	return nil
}

// InitCursor 初始化游标（如果不存在）
func (r *cursorRepository) InitCursor(ctx context.Context, cursorKey string, initialValue string) error {
	query := `
		INSERT INTO trustlog_cursor (cursor_key, cursor_value, last_updated_at)
		VALUES (?, ?, ?)
		ON CONFLICT (cursor_key) DO NOTHING
	`

	_, err := r.db.ExecContext(ctx, query, cursorKey, initialValue, time.Now())
	if err != nil {
		r.logger.ErrorContext(ctx, "failed to init cursor",
			"cursorKey", cursorKey,
			"error", err,
		)
		return fmt.Errorf("failed to init cursor: %w", err)
	}

	r.logger.DebugContext(ctx, "cursor initialized",
		"cursorKey", cursorKey,
		"initialValue", initialValue,
	)
	return nil
}

// retryRepository 重试仓储实现
type retryRepository struct {
	db     *sql.DB
	logger logger.Logger
}

// NewRetryRepository 创建重试仓储
func NewRetryRepository(db *sql.DB, log logger.Logger) RetryRepository {
	return &retryRepository{
		db:     db,
		logger: log,
	}
}

func (r *retryRepository) AddRetry(ctx context.Context, opID string, errorMsg string, nextRetryAt time.Time) error {
	return r.AddRetryTx(ctx, nil, opID, errorMsg, nextRetryAt)
}

func (r *retryRepository) AddRetryTx(ctx context.Context, tx *sql.Tx, opID string, errorMsg string, nextRetryAt time.Time) error {
	query := `
		INSERT INTO trustlog_retry (op_id, retry_count, retry_status, error_message, next_retry_at, updated_at)
		VALUES (?, 0, ?, ?, ?, ?)
	`

	var err error
	if tx != nil {
		_, err = tx.ExecContext(ctx, query, opID, string(RetryStatusPending), errorMsg, nextRetryAt, time.Now())
	} else {
		_, err = r.db.ExecContext(ctx, query, opID, string(RetryStatusPending), errorMsg, nextRetryAt, time.Now())
	}

	if err != nil {
		r.logger.ErrorContext(ctx, "failed to add retry record",
			"opID", opID,
			"error", err,
		)
		return fmt.Errorf("failed to add retry record: %w", err)
	}

	r.logger.DebugContext(ctx, "retry record added",
		"opID", opID,
		"nextRetryAt", nextRetryAt,
	)
	return nil
}

func (r *retryRepository) IncrementRetry(ctx context.Context, opID string, errorMsg string, nextRetryAt time.Time) error {
	query := `
		UPDATE trustlog_retry 
		SET retry_count = retry_count + 1, 
		    retry_status = ?,
		    last_retry_at = ?,
		    next_retry_at = ?,
		    error_message = ?,
		    updated_at = ?
		WHERE op_id = ?
	`

	_, err := r.db.ExecContext(ctx, query,
		string(RetryStatusRetrying),
		time.Now(),
		nextRetryAt,
		errorMsg,
		time.Now(),
		opID,
	)

	if err != nil {
		r.logger.ErrorContext(ctx, "failed to increment retry",
			"opID", opID,
			"error", err,
		)
		return fmt.Errorf("failed to increment retry: %w", err)
	}

	r.logger.DebugContext(ctx, "retry incremented",
		"opID", opID,
		"nextRetryAt", nextRetryAt,
	)
	return nil
}

func (r *retryRepository) MarkAsDeadLetter(ctx context.Context, opID string, errorMsg string) error {
	query := `
		UPDATE trustlog_retry 
		SET retry_status = ?,
		    error_message = ?,
		    updated_at = ?
		WHERE op_id = ?
	`

	_, err := r.db.ExecContext(ctx, query,
		string(RetryStatusDeadLetter),
		errorMsg,
		time.Now(),
		opID,
	)

	if err != nil {
		r.logger.ErrorContext(ctx, "failed to mark as dead letter",
			"opID", opID,
			"error", err,
		)
		return fmt.Errorf("failed to mark as dead letter: %w", err)
	}

	r.logger.WarnContext(ctx, "operation marked as dead letter",
		"opID", opID,
		"error", errorMsg,
	)
	return nil
}

func (r *retryRepository) FindPendingRetries(ctx context.Context, limit int) ([]RetryRecord, error) {
	query := `
		SELECT 
			op_id, retry_count, retry_status,
			last_retry_at, next_retry_at, error_message,
			created_at, updated_at
		FROM trustlog_retry
		WHERE retry_status IN (?, ?) AND next_retry_at <= ?
		ORDER BY next_retry_at ASC
		LIMIT ?
	`

	rows, err := r.db.QueryContext(ctx, query,
		string(RetryStatusPending),
		string(RetryStatusRetrying),
		time.Now(),
		limit,
	)
	if err != nil {
		r.logger.ErrorContext(ctx, "failed to find pending retries",
			"error", err,
		)
		return nil, fmt.Errorf("failed to find pending retries: %w", err)
	}
	defer rows.Close()

	var records []RetryRecord
	for rows.Next() {
		var record RetryRecord
		var lastRetry, nextRetry sql.NullTime

		err := rows.Scan(
			&record.OpID,
			&record.RetryCount,
			&record.RetryStatus,
			&lastRetry,
			&nextRetry,
			&record.ErrorMessage,
			&record.CreatedAt,
			&record.UpdatedAt,
		)
		if err != nil {
			r.logger.ErrorContext(ctx, "failed to scan retry record",
				"error", err,
			)
			return nil, fmt.Errorf("failed to scan retry record: %w", err)
		}

		if lastRetry.Valid {
			record.LastRetryAt = &lastRetry.Time
		}
		if nextRetry.Valid {
			record.NextRetryAt = &nextRetry.Time
		}

		records = append(records, record)
	}

	if err := rows.Err(); err != nil {
		return nil, fmt.Errorf("error iterating retry records: %w", err)
	}

	return records, nil
}

func (r *retryRepository) DeleteRetry(ctx context.Context, opID string) error {
	query := `DELETE FROM trustlog_retry WHERE op_id = ?`

	_, err := r.db.ExecContext(ctx, query, opID)
	if err != nil {
		r.logger.ErrorContext(ctx, "failed to delete retry record",
			"opID", opID,
			"error", err,
		)
		return fmt.Errorf("failed to delete retry record: %w", err)
	}

	r.logger.DebugContext(ctx, "retry record deleted",
		"opID", opID,
	)
	return nil
}