feat(config): 增加新闻源配置与数据库迁移支持

- 在config.json.example中添加新闻源相关配置示例,支持telegram频道订阅
- 扩展数据库表结构,新增traders表多字段支持信号源和杠杆参数
- 新增NewsConfig结构体及相关telegram新闻配置数据模型
- 在交易决策上下文结构Context中添加新闻数据news字段支持传递新闻
- 在交易决策构建用户提示信息时加入相关新闻内容
- 优化数据库操作代码,支持交易所和交易员配置的完整字段读取与更新
- 添加数据库exchanges表迁移逻辑,重建表结构和触发器以支持新字段
- 引入第三方库github.com/samber/lo用于集合操作
- 在go.mod添加新的依赖模块,并更新相关依赖版本
This commit is contained in:
wwg
2025-11-01 23:10:02 +08:00
parent 17c927b04a
commit d2af549bac
12 changed files with 934 additions and 209 deletions

73
news/news.go Normal file
View File

@@ -0,0 +1,73 @@
package news
import (
"fmt"
"time"
)
const (
// ProviderTelegrame 电报
ProviderTelegram = "telegram"
)
// NewsItem 表示一条新闻条目的核心数据结构。
// 该结构体通常用于金融资讯、舆情分析等场景,存储新闻的元数据、内容及情感分析结果。
type NewsItem struct {
// Symbol 代表与该新闻相关的金融产品代码(如股票代码 "AAPL")。
// 该字段是新闻关联资产的关键标识符。
Symbol string `json:"symbol"`
// Headline 是新闻的标题。
// 此字段为必填项,应简洁概括新闻主要内容。
Headline string `json:"headline"`
// Source 指明新闻的发布来源(例如:"路透社"、"彭博社")。
// 该字段为可选字段,若为空则不会在 JSON 输出中体现。
Source string `json:"source,omitempty"`
// URL 是新闻原文的完整链接。
// 该字段为可选字段,若为空则不会在 JSON 输出中体现。
URL string `json:"url,omitempty"`
// PublishedAt 记录新闻准确的发布时间。
// 使用 time.Time 类型以确保时间数据的一致性和可序列化。
PublishedAt time.Time `json:"published_at"`
// Sentiment 表示对新闻内容进行情感分析得出的分值。
// 取值范围为 -100 到 100其中 -100 代表极度负面100 代表极度正面0 为中性。
// 该字段为可选字段,若未进行分析则不会在 JSON 输出中体现。
Sentiment int `json:"sentiment,omitempty"`
// Impact 评估该新闻可能对市场产生的影响程度。
// 取值范围为 0 到 100数值越大表示潜在影响越大。
// 该字段为可选字段,若未进行评估则不会在 JSON 输出中体现。
Impact int `json:"impact,omitempty"`
// Summary 是新闻内容的简要摘要或关键要点。
// 该字段为可选字段,若为空则不会在 JSON 输出中体现。
Summary string `json:"summary,omitempty"`
// Tags 是与新闻内容相关的关键词标签列表(例如:["科技", "财报", "Apple"])。
// 该字段为可选字段,若为空切片则不会在 JSON 输出中体现。
Tags []string `json:"tags,omitempty"`
}
func (ni NewsItem) String() string {
return fmt.Sprintf("代币符号: %s\n标题: %s\n来源: %s\n发布时间: %s\n情感分数: %d\n影响指数: %d\n摘要: %s\n标签: %v\n%s\n",
ni.Symbol,
ni.Headline,
ni.Source,
ni.PublishedAt.Format("2006-01-02 15:04:05"),
ni.Sentiment,
ni.Impact,
ni.Summary,
ni.Tags,
"-----------",
)
}
// Provider 新闻提供者接口(由调用方实现)
type Provider interface {
// FetchNews 按币种批量获取最新新闻返回值按symbol分组
FetchNews(symbols []string, limit int) (map[string][]NewsItem, error)
}

View File

@@ -0,0 +1,145 @@
package news
import (
"fmt"
"nofx/news"
"time"
)
// DefaultNewsProvider 默认新闻提供者实现(参考示例)
// 展示如何实现 NewsProvider 接口
type DefaultNewsProvider struct {
// API配置根据实际数据源调整
APIKey string
BaseURL string
Timeout time.Duration
CacheTime time.Duration
// 可选:缓存机制
cache map[string]cacheEntry
}
type cacheEntry struct {
news []news.NewsItem
timestamp time.Time
}
// NewDefaultNewsProvider 创建默认新闻提供者
func NewDefaultNewsProvider(apiKey, baseURL string) *DefaultNewsProvider {
return &DefaultNewsProvider{
APIKey: apiKey,
BaseURL: baseURL,
Timeout: 30 * time.Second,
CacheTime: 5 * time.Minute, // 5分钟缓存
cache: make(map[string]cacheEntry),
}
}
// FetchNews 实现 NewsProvider 接口
// 按币种批量获取最新新闻返回值按symbol分组
func (p *DefaultNewsProvider) FetchNews(symbols []string, limit int) (map[string][]news.NewsItem, error) {
if len(symbols) == 0 {
return make(map[string][]news.NewsItem), nil
}
result := make(map[string][]news.NewsItem)
for _, symbol := range symbols {
// 1. 检查缓存
if cached, ok := p.getFromCache(symbol); ok {
result[symbol] = cached
continue
}
// 2. 从数据源获取(具体实现留空,由你后续补充)
news, err := p.fetchFromSource(symbol, limit)
if err != nil {
// 单个币种失败不影响其他币种
continue
}
// 3. 更新缓存
p.updateCache(symbol, news)
result[symbol] = news
}
return result, nil
}
// getFromCache 从缓存获取新闻
func (p *DefaultNewsProvider) getFromCache(symbol string) ([]news.NewsItem, bool) {
entry, exists := p.cache[symbol]
if !exists {
return nil, false
}
// 检查缓存是否过期
if time.Since(entry.timestamp) > p.CacheTime {
delete(p.cache, symbol)
return nil, false
}
return entry.news, true
}
// updateCache 更新缓存
func (p *DefaultNewsProvider) updateCache(symbol string, news []news.NewsItem) {
p.cache[symbol] = cacheEntry{
news: news,
timestamp: time.Now(),
}
}
// fetchFromSource 从数据源获取新闻(具体实现由你补充)
func (p *DefaultNewsProvider) fetchFromSource(symbol string, limit int) ([]news.NewsItem, error) {
// TODO: 实现具体的新闻获取逻辑
// 可选的数据源:
// 1. CryptoPanic API: https://cryptopanic.com/developers/api/
// 2. CoinGecko Events API
// 3. 交易所公告币安、OKX等
// 4. Twitter/X API
// 5. Reddit API
// 6. 自定义爬虫
// 示例返回结构实际需要调用API
news := []news.NewsItem{
{
Symbol: symbol,
Headline: fmt.Sprintf("%s 相关新闻标题", symbol),
Source: "CryptoPanic", // 或其他来源
URL: "https://example.com/news/123",
PublishedAt: time.Now().Add(-1 * time.Hour),
Sentiment: 0, // -100 到 100
Impact: 50, // 0 到 100
Summary: "新闻摘要内容...",
Tags: []string{"breaking", "market"},
},
}
return news, nil
}
// 可选:实现其他辅助方法
// FilterByImpact 按影响力过滤新闻
func (p *DefaultNewsProvider) FilterByImpact(items []news.NewsItem, minImpact int) []news.NewsItem {
filtered := make([]news.NewsItem, 0)
for _, item := range items {
if item.Impact >= minImpact {
filtered = append(filtered, item)
}
}
return filtered
}
// FilterByTime 按时间过滤新闻只保留最近N小时的
func (p *DefaultNewsProvider) FilterByTime(items []news.NewsItem, hours int) []news.NewsItem {
cutoff := time.Now().Add(-time.Duration(hours) * time.Hour)
filtered := make([]news.NewsItem, 0)
for _, item := range items {
if item.PublishedAt.After(cutoff) {
filtered = append(filtered, item)
}
}
return filtered
}

View File

@@ -0,0 +1,299 @@
package telegram
import (
"fmt"
"io"
"net/http"
"net/url"
"nofx/news"
"regexp"
"strings"
"time"
"github.com/PuerkitoBio/goquery"
"github.com/samber/lo"
)
// Message 表示 Telegram 消息结构
type Message struct {
MessageID string `json:"messageId"`
Title string `json:"title"`
Content string `json:"content"`
PubDate string `json:"pubDate"`
Image string `json:"image"`
Tags []string `json:"tags"`
}
// ToNews 转新闻结构
func (m Message) ToNews() news.NewsItem {
return news.NewsItem{
Symbol: "",
Headline: m.Title,
Summary: m.Content,
PublishedAt: time.Now(),
}
}
// Channel 表示频道配置
type Channel struct {
ID string `json:"id"`
Name string `json:"name"`
}
// Searcher Telegram 搜索服务
type Searcher struct {
client *http.Client
baseURL string
channels []Channel // telegram频道
keywords []string // 关键词列表,暂时忽略
}
// NewSearcher 创建搜索服务实例
func NewSearcher(baseURL string, proxyURL string, channels []Channel) (*Searcher, error) {
client := &http.Client{
Timeout: 30 * time.Second,
}
// 配置代理(如果提供)
if proxyURL != "" {
proxy, err := url.Parse(proxyURL)
if err != nil {
return nil, fmt.Errorf("invalid proxy URL: %w", err)
}
client.Transport = &http.Transport{
Proxy: http.ProxyURL(proxy),
}
}
return &Searcher{
client: client,
baseURL: baseURL,
channels: channels,
keywords: []string{""},
}, nil
}
// FetchNews 按币种批量获取最新新闻返回值按symbol分组
//
// TODO: 此处未全部实现,当前不分币种,将所有消息返回
func (s *Searcher) FetchNews(symbols []string, limit int) (map[string][]news.NewsItem, error) {
newsItem := make(map[string][]news.NewsItem)
for _, keyword := range s.keywords {
mapMessages := s.SearchAllChannels(s.channels, keyword)
for symbol, mes := range mapMessages {
newsItem[symbol] = append(newsItem[symbol], lo.Map(mes, func(item Message, _ int) news.NewsItem { return item.ToNews() })...)
}
}
return newsItem, nil
}
// SearchChannel 搜索单个频道
func (s *Searcher) SearchChannel(channelID string, keyword string) ([]Message, string, error) {
// 构造搜索 URL
searchURL := fmt.Sprintf("%s/%s", s.baseURL, channelID)
if keyword != "" {
searchURL = fmt.Sprintf("%s?q=%s", searchURL, url.QueryEscape(keyword))
}
// 创建 HTTP 请求
req, err := http.NewRequest("GET", searchURL, nil)
if err != nil {
return nil, "", err
}
// 设置请求头,模拟浏览器
req.Header.Set("User-Agent", "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7)")
req.Header.Set("Accept", "text/html,application/xhtml+xml,application/xml;q=0.9")
req.Header.Set("Accept-Language", "zh-CN,zh;q=0.9,en;q=0.8")
// 发送请求
resp, err := s.client.Do(req)
if err != nil {
return nil, "", fmt.Errorf("request failed: %w", err)
}
defer resp.Body.Close()
if resp.StatusCode != http.StatusOK {
return nil, "", fmt.Errorf("unexpected status code: %d", resp.StatusCode)
}
// 解析 HTML
return s.parseHTML(resp.Body)
}
// parseHTML 解析 HTML 内容
func (s *Searcher) parseHTML(body io.Reader) ([]Message, string, error) {
doc, err := goquery.NewDocumentFromReader(body)
if err != nil {
return nil, "", fmt.Errorf("failed to parse HTML: %w", err)
}
var messages []Message
var channelLogo string
// 提取频道 logo
doc.Find(".tgme_header_link img").Each(func(i int, s *goquery.Selection) {
if src, exists := s.Attr("src"); exists {
channelLogo = src
}
})
// 遍历消息
doc.Find(".tgme_widget_message_wrap").Each(func(i int, sel *goquery.Selection) {
message := s.extractMessage(sel)
messages = append(messages, message)
})
return messages, channelLogo, nil
}
// extractMessage 从 HTML 元素中提取消息信息
func (s *Searcher) extractMessage(sel *goquery.Selection) Message {
msg := Message{}
// 提取消息 ID
if dataPost, exists := sel.Find(".tgme_widget_message").Attr("data-post"); exists {
// data-post 格式: "channelId/messageId"
if len(dataPost) > 0 {
parts := splitLast(dataPost, "/")
if len(parts) == 2 {
msg.MessageID = parts[1]
}
}
}
// 提取消息文本
messageText := sel.Find(".js-message_text")
if messageText.Length() > 0 {
html, _ := messageText.Html()
// 提取标题(第一行)
if html != "" {
lines := splitFirst(html, "<br/>")
if len(lines) > 0 {
msg.Title = stripHTML(lines[0])
}
// 提取内容(去除标题后的文本)
if len(lines) > 1 {
msg.Content = stripHTML(lines[1])
}
}
}
// 提取发布时间
if datetime, exists := sel.Find("time").Attr("datetime"); exists {
msg.PubDate = datetime
}
// 提取图片
if style, exists := sel.Find(".tgme_widget_message_photo_wrap").Attr("style"); exists {
msg.Image = extractImageURL(style)
}
// 提取标签
sel.Find(".tgme_widget_message_text a").Each(func(i int, s *goquery.Selection) {
text := s.Text()
if len(text) > 0 && text[0] == '#' {
msg.Tags = append(msg.Tags, text)
}
})
return msg
}
// SearchAllChannels 并行搜索多个频道
func (s *Searcher) SearchAllChannels(channels []Channel, keyword string) map[string][]Message {
type result struct {
channelID string
messages []Message
logo string
err error
}
resultChan := make(chan result, len(channels))
// 并行搜索
for _, channel := range channels {
go func(ch Channel) {
messages, logo, err := s.SearchChannel(ch.ID, keyword)
resultChan <- result{
channelID: ch.ID,
messages: messages,
logo: logo,
err: err,
}
}(channel)
}
// 收集结果
results := make(map[string][]Message)
for i := 0; i < len(channels); i++ {
res := <-resultChan
if res.err == nil {
results[res.channelID] = res.messages
}
}
return results
}
// splitFirst 在第一个分隔符处分割字符串,返回最多两个部分
func splitFirst(s, sep string) []string {
// 使用 SplitN 限制分割次数为 2
// 当 n=2 时,返回的切片最多包含两个元素
result := strings.SplitN(s, sep, 2)
// 如果字符串中不包含分隔符SplitN 会返回包含原字符串的切片
// 这符合预期行为,直接返回即可
return result
}
// splitLast 在最后一个分隔符处分割字符串,返回最多两个部分
func splitLast(s, sep string) []string {
// 查找最后一个分隔符的位置
index := strings.LastIndex(s, sep)
if index < 0 {
// 如果没有找到分隔符,返回包含原字符串的切片
return []string{s}
}
// 根据最后一个分隔符的位置分割字符串
part1 := s[:index]
part2 := s[index+len(sep):]
return []string{part1, part2}
}
// stripHTML 移除字符串中的所有 HTML 标签,只保留纯文本
func stripHTML(s string) string {
// 将HTML标签全转换成小写确保匹配大小写不敏感的标签
re := regexp.MustCompile(`\<[\S\s]+?\>`)
s = re.ReplaceAllStringFunc(s, strings.ToLower)
// 去除 <style> 标签及其内容
re = regexp.MustCompile(`\<style[\S\s]+?\</style\>`)
s = re.ReplaceAllString(s, "")
// 去除 <script> 标签及其内容
re = regexp.MustCompile(`\<script[\S\s]+?\</script\>`)
s = re.ReplaceAllString(s, "")
// 去除所有尖括号内的 HTML 代码,并换成换行符
re = regexp.MustCompile(`\<[\S\s]+?\>`)
s = re.ReplaceAllString(s, "\n")
// 去除连续的换行符和空白字符
re = regexp.MustCompile(`\s{2,}`)
s = re.ReplaceAllString(s, "\n")
// 去除首尾的空白字符
return strings.TrimSpace(s)
}
func extractImageURL(style string) string {
// 从 style 属性中提取图片 URL
// 格式: background-image:url('...')
return ""
}