新增分享码功能

internal/repository/share_code.go

@@ -0,0 +1,58 @@
+package repository
+
+import (
+	"context"
+	"database/sql"
+	"fmt"
+	"time"
+
+	"github.com/luxsin/app-api/internal/model"
+)
+
+type ShareCodeRepository struct {
+	db *sql.DB
+}
+
+func NewShareCodeRepository(db *sql.DB) *ShareCodeRepository {
+	return &ShareCodeRepository{db: db}
+}
+
+func (r *ShareCodeRepository) InsertLog(ctx context.Context, log model.ShareCodeLog) error {
+	const query = `INSERT INTO share_code_log (mac_addr, share_code, action, ip_addr, eq_data, expire_at)
+		VALUES (?, ?, ?, ?, ?, ?)`
+
+	var expireAt any
+	if log.ExpireAt != nil {
+		expireAt = *log.ExpireAt
+	}
+
+	_, err := r.db.ExecContext(ctx, query,
+		log.MacAddr, log.ShareCode, log.Action, log.IpAddr, log.EqData, expireAt,
+	)
+	if err != nil {
+		return fmt.Errorf("insert share_code_log: %w", err)
+	}
+	return nil
+}
+
+func (r *ShareCodeRepository) HasExportLog(ctx context.Context, shareCode string) (bool, error) {
+	const query = `SELECT 1 FROM share_code_log WHERE share_code = ? AND action = 'export' LIMIT 1`
+
+	var one int
+	err := r.db.QueryRowContext(ctx, query, shareCode).Scan(&one)
+	if err == sql.ErrNoRows {
+		return false, nil
+	}
+	if err != nil {
+		return false, fmt.Errorf("query share export log: %w", err)
+	}
+	return true, nil
+}
+
+func ParseShareExpireAt(expireAt time.Time) *time.Time {
+	if expireAt.IsZero() {
+		return nil
+	}
+	t := expireAt
+	return &t
+}