feat: rule and message_log store CRUD

2026-06-27 13:06:54 +08:00
parent 236105ea9c
commit a8e939da40
2 changed files with 228 additions and 0 deletions
@@ -0,0 +1,156 @@
+package store
+
+import (
+	"context"
+	"database/sql"
+	"encoding/json"
+	"fmt"
+
+	"aiaa-notification-service/internal/model"
+)
+
+func (s *Store) CreateRule(ctx context.Context, r *model.Rule, channelIDs []int) error {
+	tx, err := s.DB.BeginTxx(ctx, nil)
+	if err != nil {
+		return fmt.Errorf("begin tx: %w", err)
+	}
+	defer tx.Rollback()
+
+	query := `INSERT INTO rule (source_id, event, template_id, conditions, enabled) VALUES (?, ?, ?, ?, ?)`
+	condsJSON, _ := marshalJSON(r.Conditions)
+	result, err := tx.ExecContext(ctx, query, r.SourceID, r.Event, r.TemplateID, condsJSON, r.Enabled)
+	if err != nil {
+		return fmt.Errorf("create rule: %w", err)
+	}
+	id, _ := result.LastInsertId()
+	r.ID = int(id)
+
+	for _, chID := range channelIDs {
+		_, err := tx.ExecContext(ctx, `INSERT INTO rule_channel (rule_id, channel_id, enabled) VALUES (?, ?, 1)`, r.ID, chID)
+		if err != nil {
+			return fmt.Errorf("add rule_channel: %w", err)
+		}
+	}
+	return tx.Commit()
+}
+
+func (s *Store) GetRule(ctx context.Context, id int) (*model.Rule, error) {
+	var r model.Rule
+	var condsBytes []byte
+	row := s.DB.QueryRowContext(ctx, `SELECT id, source_id, event, template_id, conditions, enabled, created_at, updated_at FROM rule WHERE id = ?`, id)
+	if err := row.Scan(&r.ID, &r.SourceID, &r.Event, &r.TemplateID, &condsBytes, &r.Enabled, &r.CreatedAt, &r.UpdatedAt); err != nil {
+		return nil, fmt.Errorf("get rule %d: %w", id, err)
+	}
+	if len(condsBytes) > 0 && string(condsBytes) != "null" {
+		raw := json.RawMessage(condsBytes)
+		r.Conditions = &raw
+	}
+	return &r, nil
+}
+
+func (s *Store) GetRuleBySourceEvent(ctx context.Context, sourceID int, event string) (*model.Rule, error) {
+	var r model.Rule
+	var condsBytes []byte
+	query := `SELECT id, source_id, event, template_id, conditions, enabled, created_at, updated_at FROM rule WHERE source_id = ? AND event = ? AND enabled = 1`
+	row := s.DB.QueryRowContext(ctx, query, sourceID, event)
+	if err := row.Scan(&r.ID, &r.SourceID, &r.Event, &r.TemplateID, &condsBytes, &r.Enabled, &r.CreatedAt, &r.UpdatedAt); err != nil {
+		return nil, fmt.Errorf("get rule by source+event: %w", err)
+	}
+	if len(condsBytes) > 0 && string(condsBytes) != "null" {
+		raw := json.RawMessage(condsBytes)
+		r.Conditions = &raw
+	}
+	return &r, nil
+}
+
+func (s *Store) ListRules(ctx context.Context) ([]model.Rule, error) {
+	rows, err := s.DB.QueryContext(ctx, `SELECT id, source_id, event, template_id, conditions, enabled, created_at, updated_at FROM rule ORDER BY id`)
+	if err != nil {
+		return nil, fmt.Errorf("list rules: %w", err)
+	}
+	defer rows.Close()
+	return scanRules(rows)
+}
+
+func (s *Store) UpdateRule(ctx context.Context, id int, r *model.Rule, channelIDs []int) error {
+	tx, err := s.DB.BeginTxx(ctx, nil)
+	if err != nil {
+		return fmt.Errorf("begin tx: %w", err)
+	}
+	defer tx.Rollback()
+
+	condsJSON, _ := marshalJSON(r.Conditions)
+	_, err = tx.ExecContext(ctx, `UPDATE rule SET source_id=?, event=?, template_id=?, conditions=?, enabled=? WHERE id=?`,
+		r.SourceID, r.Event, r.TemplateID, condsJSON, r.Enabled, id)
+	if err != nil {
+		return fmt.Errorf("update rule: %w", err)
+	}
+
+	if channelIDs != nil {
+		_, _ = tx.ExecContext(ctx, `DELETE FROM rule_channel WHERE rule_id = ?`, id)
+		for _, chID := range channelIDs {
+			_, err := tx.ExecContext(ctx, `INSERT INTO rule_channel (rule_id, channel_id, enabled) VALUES (?, ?, 1)`, id, chID)
+			if err != nil {
+				return fmt.Errorf("add rule_channel: %w", err)
+			}
+		}
+	}
+	return tx.Commit()
+}
+
+func (s *Store) DeleteRule(ctx context.Context, id int) error {
+	_, err := s.DB.ExecContext(ctx, `DELETE FROM rule WHERE id = ?`, id)
+	return err
+}
+
+func (s *Store) SetRuleEnabled(ctx context.Context, id int, enabled bool) error {
+	v := 0
+	if enabled {
+		v = 1
+	}
+	_, err := s.DB.ExecContext(ctx, `UPDATE rule SET enabled = ? WHERE id = ?`, v, id)
+	return err
+}
+
+func (s *Store) GetRuleChannels(ctx context.Context, ruleID int) ([]model.RuleChannel, error) {
+	var rcs []model.RuleChannel
+	err := s.DB.SelectContext(ctx, &rcs, `SELECT id, rule_id, channel_id, enabled FROM rule_channel WHERE rule_id = ? AND enabled = 1`, ruleID)
+	if err != nil {
+		return nil, fmt.Errorf("get rule channels: %w", err)
+	}
+	return rcs, nil
+}
+
+func (s *Store) SetRuleChannelEnabled(ctx context.Context, ruleID, channelID int, enabled bool) error {
+	v := 0
+	if enabled {
+		v = 1
+	}
+	_, err := s.DB.ExecContext(ctx, `UPDATE rule_channel SET enabled = ? WHERE rule_id = ? AND channel_id = ?`, v, ruleID, channelID)
+	return err
+}
+
+// helpers
+func marshalJSON(v *json.RawMessage) (interface{}, error) {
+	if v == nil {
+		return nil, nil
+	}
+	return json.Marshal(v)
+}
+
+func scanRules(rows *sql.Rows) ([]model.Rule, error) {
+	var rules []model.Rule
+	for rows.Next() {
+		var r model.Rule
+		var condsBytes []byte
+		if err := rows.Scan(&r.ID, &r.SourceID, &r.Event, &r.TemplateID, &condsBytes, &r.Enabled, &r.CreatedAt, &r.UpdatedAt); err != nil {
+			return nil, err
+		}
+		if len(condsBytes) > 0 && string(condsBytes) != "null" {
+			raw := json.RawMessage(condsBytes)
+			r.Conditions = &raw
+		}
+		rules = append(rules, r)
+	}
+	return rules, rows.Err()
+}