fix: 移除早间简报功能 + 修复多处"人机感"用语
- 删除 briefing_handler.go / briefing_store.go 及所有相关路由与配置 - 移除 Gateway Config 中 ToolEngineURL / BriefingTime 字段 - 移除 DevTools 中 gateway 的 TOOL_ENGINE_URL 环境变量 - webhook 错误提示从"AI 服务异常/暂不可用"改为昔涟自然口吻 - markdown 导出中昔涟头像从 🤖 改为 💫 - 后台思考提示词"系统会误解析"改为"我会误解析" Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>
This commit is contained in:
@@ -1,331 +0,0 @@
|
||||
package store
|
||||
|
||||
import (
|
||||
"database/sql"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"github.com/yourname/cyrene-ai/pkg/logger"
|
||||
"time"
|
||||
)
|
||||
|
||||
// Briefing 每日简报模型
|
||||
type Briefing struct {
|
||||
ID string `json:"id"`
|
||||
UserID string `json:"user_id"`
|
||||
Date string `json:"date"` // YYYY-MM-DD
|
||||
Weather *WeatherData `json:"weather"`
|
||||
News []NewsItem `json:"news"`
|
||||
Reminders []BriefReminder `json:"reminders"`
|
||||
Summary string `json:"summary"`
|
||||
SummarySource string `json:"summary_source"` // "ai" | "fallback"
|
||||
Status string `json:"status"` // pending, generated, delivered
|
||||
GeneratedAt *time.Time `json:"generated_at,omitempty"`
|
||||
DeliveredAt *time.Time `json:"delivered_at,omitempty"`
|
||||
CreatedAt time.Time `json:"created_at"`
|
||||
}
|
||||
|
||||
// WeatherData 天气数据
|
||||
type WeatherData struct {
|
||||
Location string `json:"location"`
|
||||
Temp float64 `json:"temp"`
|
||||
Condition string `json:"condition"`
|
||||
Icon string `json:"icon"`
|
||||
}
|
||||
|
||||
// NewsItem 新闻条目
|
||||
type NewsItem struct {
|
||||
Title string `json:"title"`
|
||||
URL string `json:"url"`
|
||||
Source string `json:"source"`
|
||||
Summary string `json:"summary"`
|
||||
}
|
||||
|
||||
// BriefReminder 简报中的提醒摘要
|
||||
type BriefReminder struct {
|
||||
ID string `json:"id"`
|
||||
Title string `json:"title"`
|
||||
RemindAt string `json:"remind_at"`
|
||||
}
|
||||
|
||||
// BriefingStore 每日简报持久化存储
|
||||
type BriefingStore struct {
|
||||
db *sql.DB
|
||||
}
|
||||
|
||||
// NewBriefingStore 使用已有数据库连接初始化简报存储并自动建表
|
||||
func NewBriefingStore(db *sql.DB) (*BriefingStore, error) {
|
||||
store := &BriefingStore{db: db}
|
||||
|
||||
if err := store.migrate(); err != nil {
|
||||
return nil, fmt.Errorf("简报表迁移失败: %w", err)
|
||||
}
|
||||
|
||||
logger.Println("[BriefingStore] 简报持久化存储已初始化")
|
||||
return store, nil
|
||||
}
|
||||
|
||||
// migrate 自动创建简报表结构
|
||||
func (s *BriefingStore) migrate() error {
|
||||
queries := []string{
|
||||
`CREATE TABLE IF NOT EXISTS daily_briefings (
|
||||
id VARCHAR(36) PRIMARY KEY,
|
||||
user_id VARCHAR(255) NOT NULL,
|
||||
date DATE NOT NULL,
|
||||
weather JSONB DEFAULT '{}',
|
||||
news JSONB DEFAULT '[]',
|
||||
reminders JSONB DEFAULT '[]',
|
||||
summary TEXT DEFAULT '',
|
||||
summary_source VARCHAR(20) DEFAULT 'ai',
|
||||
status VARCHAR(20) DEFAULT 'pending',
|
||||
generated_at TIMESTAMPTZ,
|
||||
delivered_at TIMESTAMPTZ,
|
||||
created_at TIMESTAMPTZ DEFAULT NOW(),
|
||||
UNIQUE(user_id, date)
|
||||
)`,
|
||||
`ALTER TABLE daily_briefings ADD COLUMN IF NOT EXISTS summary_source VARCHAR(20) DEFAULT 'ai'`,
|
||||
`CREATE INDEX IF NOT EXISTS idx_briefings_user_id ON daily_briefings(user_id)`,
|
||||
`CREATE INDEX IF NOT EXISTS idx_briefings_date ON daily_briefings(date)`,
|
||||
`CREATE INDEX IF NOT EXISTS idx_briefings_user_date ON daily_briefings(user_id, date)`,
|
||||
}
|
||||
|
||||
for _, q := range queries {
|
||||
if _, err := s.db.Exec(q); err != nil {
|
||||
return fmt.Errorf("迁移SQL执行失败: %w\nSQL: %s", err, q)
|
||||
}
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// CreateOrUpdateBriefing upsert 简报
|
||||
func (s *BriefingStore) CreateOrUpdateBriefing(b *Briefing) error {
|
||||
weatherJSON, err := json.Marshal(b.Weather)
|
||||
if err != nil {
|
||||
return fmt.Errorf("序列化天气数据失败: %w", err)
|
||||
}
|
||||
newsJSON, err := json.Marshal(b.News)
|
||||
if err != nil {
|
||||
return fmt.Errorf("序列化新闻数据失败: %w", err)
|
||||
}
|
||||
remindersJSON, err := json.Marshal(b.Reminders)
|
||||
if err != nil {
|
||||
return fmt.Errorf("序列化提醒数据失败: %w", err)
|
||||
}
|
||||
|
||||
if b.SummarySource == "" {
|
||||
b.SummarySource = "ai"
|
||||
}
|
||||
|
||||
_, err = s.db.Exec(
|
||||
`INSERT INTO daily_briefings (id, user_id, date, weather, news, reminders, summary, summary_source, status, generated_at, delivered_at)
|
||||
VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11)
|
||||
ON CONFLICT (user_id, date) DO UPDATE SET
|
||||
weather = EXCLUDED.weather,
|
||||
news = EXCLUDED.news,
|
||||
reminders = EXCLUDED.reminders,
|
||||
summary = EXCLUDED.summary,
|
||||
summary_source = EXCLUDED.summary_source,
|
||||
status = EXCLUDED.status,
|
||||
generated_at = EXCLUDED.generated_at,
|
||||
delivered_at = EXCLUDED.delivered_at`,
|
||||
b.ID, b.UserID, b.Date, string(weatherJSON), string(newsJSON), string(remindersJSON),
|
||||
b.Summary, b.SummarySource, b.Status, b.GeneratedAt, b.DeliveredAt,
|
||||
)
|
||||
if err != nil {
|
||||
return fmt.Errorf("upsert 简报失败: %w", err)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// GetBriefingByDate 获取指定日期简报
|
||||
func (s *BriefingStore) GetBriefingByDate(userID, date string) (*Briefing, error) {
|
||||
row := s.db.QueryRow(
|
||||
`SELECT id, user_id, date::TEXT, weather, news, reminders, summary, COALESCE(summary_source, 'ai'), status, generated_at, delivered_at, created_at
|
||||
FROM daily_briefings WHERE user_id = $1 AND date = $2::DATE`,
|
||||
userID, date,
|
||||
)
|
||||
|
||||
b, err := s.scanBriefing(row)
|
||||
if err != nil {
|
||||
if err == sql.ErrNoRows {
|
||||
return nil, nil
|
||||
}
|
||||
return nil, fmt.Errorf("查询简报失败: %w", err)
|
||||
}
|
||||
return b, nil
|
||||
}
|
||||
|
||||
// GetLatestBriefings 获取最近简报列表
|
||||
func (s *BriefingStore) GetLatestBriefings(userID string, limit int) ([]Briefing, error) {
|
||||
if limit <= 0 {
|
||||
limit = 7
|
||||
}
|
||||
|
||||
rows, err := s.db.Query(
|
||||
`SELECT id, user_id, date::TEXT, weather, news, reminders, summary, COALESCE(summary_source, 'ai'), status, generated_at, delivered_at, created_at
|
||||
FROM daily_briefings WHERE user_id = $1
|
||||
ORDER BY date DESC LIMIT $2`,
|
||||
userID, limit,
|
||||
)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("查询简报列表失败: %w", err)
|
||||
}
|
||||
defer rows.Close()
|
||||
|
||||
var briefings []Briefing
|
||||
for rows.Next() {
|
||||
var (
|
||||
id, uid, date, summary, summarySource, status string
|
||||
weatherRaw, newsRaw, remindersRaw []byte
|
||||
generatedAt, deliveredAt, createdAt sql.NullTime
|
||||
)
|
||||
if err := rows.Scan(&id, &uid, &date, &weatherRaw, &newsRaw, &remindersRaw,
|
||||
&summary, &summarySource, &status, &generatedAt, &deliveredAt, &createdAt); err != nil {
|
||||
return nil, fmt.Errorf("扫描简报行失败: %w", err)
|
||||
}
|
||||
|
||||
b := Briefing{
|
||||
ID: id,
|
||||
UserID: uid,
|
||||
Date: date,
|
||||
Summary: summary,
|
||||
SummarySource: summarySource,
|
||||
Status: status,
|
||||
}
|
||||
|
||||
if weatherRaw != nil {
|
||||
var w WeatherData
|
||||
if err := json.Unmarshal(weatherRaw, &w); err == nil {
|
||||
b.Weather = &w
|
||||
}
|
||||
}
|
||||
if newsRaw != nil {
|
||||
json.Unmarshal(newsRaw, &b.News)
|
||||
}
|
||||
if remindersRaw != nil {
|
||||
json.Unmarshal(remindersRaw, &b.Reminders)
|
||||
}
|
||||
if generatedAt.Valid {
|
||||
b.GeneratedAt = &generatedAt.Time
|
||||
}
|
||||
if deliveredAt.Valid {
|
||||
b.DeliveredAt = &deliveredAt.Time
|
||||
}
|
||||
b.CreatedAt = createdAt.Time
|
||||
|
||||
// 确保切片不为 nil
|
||||
if b.News == nil {
|
||||
b.News = []NewsItem{}
|
||||
}
|
||||
if b.Reminders == nil {
|
||||
b.Reminders = []BriefReminder{}
|
||||
}
|
||||
if b.Weather == nil {
|
||||
b.Weather = &WeatherData{}
|
||||
}
|
||||
|
||||
briefings = append(briefings, b)
|
||||
}
|
||||
|
||||
if briefings == nil {
|
||||
briefings = []Briefing{}
|
||||
}
|
||||
return briefings, rows.Err()
|
||||
}
|
||||
|
||||
// GetUsersWithBriefings 获取拥有简报的所有用户 ID 列表(用于调度器)
|
||||
func (s *BriefingStore) GetUsersWithBriefings() ([]string, error) {
|
||||
rows, err := s.db.Query(`SELECT DISTINCT user_id FROM daily_briefings`)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("查询简报用户列表失败: %w", err)
|
||||
}
|
||||
defer rows.Close()
|
||||
|
||||
var userIDs []string
|
||||
for rows.Next() {
|
||||
var uid string
|
||||
if err := rows.Scan(&uid); err != nil {
|
||||
return nil, fmt.Errorf("扫描用户ID失败: %w", err)
|
||||
}
|
||||
userIDs = append(userIDs, uid)
|
||||
}
|
||||
if userIDs == nil {
|
||||
userIDs = []string{}
|
||||
}
|
||||
return userIDs, rows.Err()
|
||||
}
|
||||
|
||||
// GetAllUsers 获取所有用户 ID(从 reminders 表获取,作为降级方案)
|
||||
func (s *BriefingStore) GetAllUsers() ([]string, error) {
|
||||
rows, err := s.db.Query(`SELECT DISTINCT user_id FROM reminders`)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("查询用户列表失败: %w", err)
|
||||
}
|
||||
defer rows.Close()
|
||||
|
||||
var userIDs []string
|
||||
for rows.Next() {
|
||||
var uid string
|
||||
if err := rows.Scan(&uid); err != nil {
|
||||
return nil, fmt.Errorf("扫描用户ID失败: %w", err)
|
||||
}
|
||||
userIDs = append(userIDs, uid)
|
||||
}
|
||||
if userIDs == nil {
|
||||
userIDs = []string{}
|
||||
}
|
||||
return userIDs, rows.Err()
|
||||
}
|
||||
|
||||
// scanBriefing 扫描单行简报
|
||||
func (s *BriefingStore) scanBriefing(row *sql.Row) (*Briefing, error) {
|
||||
var (
|
||||
id, uid, date, summary, summarySource, status string
|
||||
weatherRaw, newsRaw, remindersRaw []byte
|
||||
generatedAt, deliveredAt, createdAt sql.NullTime
|
||||
)
|
||||
|
||||
if err := row.Scan(&id, &uid, &date, &weatherRaw, &newsRaw, &remindersRaw,
|
||||
&summary, &summarySource, &status, &generatedAt, &deliveredAt, &createdAt); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
b := &Briefing{
|
||||
ID: id,
|
||||
UserID: uid,
|
||||
Date: date,
|
||||
Summary: summary,
|
||||
SummarySource: summarySource,
|
||||
Status: status,
|
||||
}
|
||||
|
||||
if weatherRaw != nil {
|
||||
var w WeatherData
|
||||
if err := json.Unmarshal(weatherRaw, &w); err == nil {
|
||||
b.Weather = &w
|
||||
}
|
||||
}
|
||||
if b.Weather == nil {
|
||||
b.Weather = &WeatherData{}
|
||||
}
|
||||
if newsRaw != nil {
|
||||
json.Unmarshal(newsRaw, &b.News)
|
||||
}
|
||||
if b.News == nil {
|
||||
b.News = []NewsItem{}
|
||||
}
|
||||
if remindersRaw != nil {
|
||||
json.Unmarshal(remindersRaw, &b.Reminders)
|
||||
}
|
||||
if b.Reminders == nil {
|
||||
b.Reminders = []BriefReminder{}
|
||||
}
|
||||
if generatedAt.Valid {
|
||||
b.GeneratedAt = &generatedAt.Time
|
||||
}
|
||||
if deliveredAt.Valid {
|
||||
b.DeliveredAt = &deliveredAt.Time
|
||||
}
|
||||
b.CreatedAt = createdAt.Time
|
||||
|
||||
return b, nil
|
||||
}
|
||||
Reference in New Issue
Block a user