ryan
c74bab3033
Motivation: 当数据库查询无结果时，Go 的 `var slice []Type` 声明会使 JSON 序列化输出 `null`，而非 `[]`。这在 API 响应中造成不一致，也增加了客户端的空值处理负担。改为 `make([]Type, 0)` 后空结果统一输出为 `[]`，提升 API 规范性。 Changes: * 统一 store 层所有 List 方法使用 `make([]Type, 0)` 初始化切片，覆盖 Sources、Channels、Templates、Rules、MessageLogs 及 RuleChannels 查询 * 新增完整的 cURL 示例文档，覆盖全部 API 端点（通知、Source、Template、Channel、Rule、消息记录） * 新增 Postman Collection JSON 文件，可直接导入 HTTPie 使用，包含环境变量配置
91 lines
3.0 KiB
Go
91 lines
3.0 KiB
Go
package store
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"fmt"
|
|
|
|
"aiaa-notification-service/internal/model"
|
|
)
|
|
|
|
func (s *Store) CreateChannel(ctx context.Context, ch *model.Channel) error {
|
|
configJSON, err := json.Marshal(ch.Config)
|
|
if err != nil {
|
|
return fmt.Errorf("marshal channel config: %w", err)
|
|
}
|
|
query := `INSERT INTO notification_channel (name, type, config, status) VALUES (?, ?, ?, ?)`
|
|
result, err := s.DB.ExecContext(ctx, query, ch.Name, ch.Type, configJSON, ch.Status)
|
|
if err != nil {
|
|
return fmt.Errorf("create channel: %w", err)
|
|
}
|
|
id, _ := result.LastInsertId()
|
|
ch.ID = int(id)
|
|
return nil
|
|
}
|
|
|
|
func (s *Store) GetChannel(ctx context.Context, id int) (*model.Channel, error) {
|
|
var ch model.Channel
|
|
var configBytes []byte
|
|
row := s.DB.QueryRowContext(ctx, `SELECT id, name, type, config, status, created_at, updated_at FROM notification_channel WHERE id = ?`, id)
|
|
if err := row.Scan(&ch.ID, &ch.Name, &ch.Type, &configBytes, &ch.Status, &ch.CreatedAt, &ch.UpdatedAt); err != nil {
|
|
return nil, fmt.Errorf("get channel %d: %w", id, err)
|
|
}
|
|
raw := json.RawMessage(configBytes)
|
|
ch.Config = &raw
|
|
return &ch, nil
|
|
}
|
|
|
|
func (s *Store) GetChannelByName(ctx context.Context, name string) (*model.Channel, error) {
|
|
var ch model.Channel
|
|
var configBytes []byte
|
|
row := s.DB.QueryRowContext(ctx, `SELECT id, name, type, config, status, created_at, updated_at FROM notification_channel WHERE name = ?`, name)
|
|
if err := row.Scan(&ch.ID, &ch.Name, &ch.Type, &configBytes, &ch.Status, &ch.CreatedAt, &ch.UpdatedAt); err != nil {
|
|
return nil, fmt.Errorf("get channel by name %s: %w", name, err)
|
|
}
|
|
raw := json.RawMessage(configBytes)
|
|
ch.Config = &raw
|
|
return &ch, nil
|
|
}
|
|
|
|
func (s *Store) ListChannels(ctx context.Context) ([]model.Channel, error) {
|
|
rows, err := s.DB.QueryContext(ctx, `SELECT id, name, type, config, status, created_at, updated_at FROM notification_channel ORDER BY id`)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("list channels: %w", err)
|
|
}
|
|
defer rows.Close()
|
|
|
|
channels := make([]model.Channel, 0)
|
|
for rows.Next() {
|
|
var ch model.Channel
|
|
var configBytes []byte
|
|
if err := rows.Scan(&ch.ID, &ch.Name, &ch.Type, &configBytes, &ch.Status, &ch.CreatedAt, &ch.UpdatedAt); err != nil {
|
|
return nil, fmt.Errorf("scan channel: %w", err)
|
|
}
|
|
raw := json.RawMessage(configBytes)
|
|
ch.Config = &raw
|
|
channels = append(channels, ch)
|
|
}
|
|
return channels, rows.Err()
|
|
}
|
|
|
|
func (s *Store) UpdateChannel(ctx context.Context, id int, ch *model.Channel) error {
|
|
configJSON, err := json.Marshal(ch.Config)
|
|
if err != nil {
|
|
return fmt.Errorf("marshal channel config: %w", err)
|
|
}
|
|
query := `UPDATE notification_channel SET name=?, type=?, config=?, status=? WHERE id=?`
|
|
_, err = s.DB.ExecContext(ctx, query, ch.Name, ch.Type, configJSON, ch.Status, id)
|
|
if err != nil {
|
|
return fmt.Errorf("update channel %d: %w", id, err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (s *Store) DeleteChannel(ctx context.Context, id int) error {
|
|
_, err := s.DB.ExecContext(ctx, `DELETE FROM notification_channel WHERE id = ?`, id)
|
|
if err != nil {
|
|
return fmt.Errorf("delete channel %d: %w", id, err)
|
|
}
|
|
return nil
|
|
}
|