mirror of
https://github.com/NoFxAiOS/nofx.git
synced 2026-07-13 07:46:54 +08:00
feat(config): 增加新闻源配置与数据库迁移支持
- 在config.json.example中添加新闻源相关配置示例,支持telegram频道订阅 - 扩展数据库表结构,新增traders表多字段支持信号源和杠杆参数 - 新增NewsConfig结构体及相关telegram新闻配置数据模型 - 在交易决策上下文结构Context中添加新闻数据news字段支持传递新闻 - 在交易决策构建用户提示信息时加入相关新闻内容 - 优化数据库操作代码,支持交易所和交易员配置的完整字段读取与更新 - 添加数据库exchanges表迁移逻辑,重建表结构和触发器以支持新字段 - 引入第三方库github.com/samber/lo用于集合操作 - 在go.mod添加新的依赖模块,并更新相关依赖版本
This commit is contained in:
73
news/news.go
Normal file
73
news/news.go
Normal 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)
|
||||
}
|
||||
145
news/provider/default/default.go
Normal file
145
news/provider/default/default.go
Normal 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
|
||||
}
|
||||
299
news/provider/telegram/telegram.go
Normal file
299
news/provider/telegram/telegram.go
Normal 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 ""
|
||||
}
|
||||
Reference in New Issue
Block a user