This commit is contained in:
2026-04-20 21:29:22 +02:00
parent 53dd49612d
commit 89fc0119f3
25 changed files with 3744 additions and 134 deletions
+75
View File
@@ -0,0 +1,75 @@
package finnhub
import (
"encoding/json"
"fmt"
"net/http"
"time"
)
const baseURL = "https://finnhub.io/api/v1"
type Client struct {
apiKey string
http *http.Client
}
type NewsItem struct {
ID int `json:"id"`
Category string `json:"category"`
Datetime int64 `json:"datetime"`
Headline string `json:"headline"`
Related string `json:"related"`
Source string `json:"source"`
URL string `json:"url"`
Summary string `json:"summary"`
}
func New(apiKey string) *Client {
return &Client{
apiKey: apiKey,
http: &http.Client{Timeout: 10 * time.Second},
}
}
func (c *Client) CompanyNews(symbol, from, to string) ([]NewsItem, error) {
url := fmt.Sprintf("%s/company-news?symbol=%s&from=%s&to=%s&token=%s",
baseURL, symbol, from, to, c.apiKey)
return c.fetchNews(url)
}
func (c *Client) MarketNews() ([]NewsItem, error) {
url := fmt.Sprintf("%s/news?category=general&token=%s", baseURL, c.apiKey)
return c.fetchNews(url)
}
func (c *Client) Ping() error {
url := fmt.Sprintf("%s/news?category=general&minId=999999999&token=%s", baseURL, c.apiKey)
resp, err := c.http.Get(url)
if err != nil {
return err
}
defer resp.Body.Close()
if resp.StatusCode == 401 || resp.StatusCode == 403 {
return fmt.Errorf("invalid API key (HTTP %d)", resp.StatusCode)
}
return nil
}
func (c *Client) fetchNews(url string) ([]NewsItem, error) {
resp, err := c.http.Get(url)
if err != nil {
return nil, err
}
defer resp.Body.Close()
if resp.StatusCode != 200 {
return nil, fmt.Errorf("finnhub: HTTP %d", resp.StatusCode)
}
var items []NewsItem
if err := json.NewDecoder(resp.Body).Decode(&items); err != nil {
return nil, err
}
return items, nil
}
+131
View File
@@ -0,0 +1,131 @@
package finnhub
import (
"log"
"time"
"git.rouggy.com/rouggy/stockradar/internal/db"
)
type Poller struct {
db *db.DB
getKey func() (string, error)
ticker *time.Ticker
done chan struct{}
lastRun time.Time
}
func NewPoller(database *db.DB, getKey func() (string, error)) *Poller {
return &Poller{
db: database,
getKey: getKey,
done: make(chan struct{}),
}
}
func (p *Poller) Start() {
p.ticker = time.NewTicker(15 * time.Minute)
go func() {
// Run immediately on start
if err := p.Sync(); err != nil {
log.Printf("finnhub poller: initial sync: %v", err)
}
for {
select {
case <-p.ticker.C:
if err := p.Sync(); err != nil {
log.Printf("finnhub poller: sync: %v", err)
}
case <-p.done:
return
}
}
}()
}
func (p *Poller) Stop() {
if p.ticker != nil {
p.ticker.Stop()
}
close(p.done)
}
func (p *Poller) Sync() error {
apiKey, err := p.getKey()
if err != nil || apiKey == "" {
return nil // pas de clé configurée, on skip silencieusement
}
client := New(apiKey)
tickers, err := p.watchlistTickers()
if err != nil {
return err
}
now := time.Now()
from := now.AddDate(0, 0, -7).Format("2006-01-02")
to := now.Format("2006-01-02")
total := 0
for _, sym := range tickers {
items, err := client.CompanyNews(sym, from, to)
if err != nil {
log.Printf("finnhub: news %s: %v", sym, err)
continue
}
for _, item := range items {
if p.insertNews(sym, item) {
total++
}
}
time.Sleep(250 * time.Millisecond) // Finnhub free tier: 60 req/min
}
// News marché général (sans ticker spécifique)
market, err := client.MarketNews()
if err == nil {
for _, item := range market {
p.insertNews("", item)
}
}
p.lastRun = now
if total > 0 {
log.Printf("finnhub: sync done — %d nouvelles news", total)
}
return nil
}
func (p *Poller) LastRun() time.Time { return p.lastRun }
func (p *Poller) watchlistTickers() ([]string, error) {
rows, err := p.db.Query(`SELECT ticker FROM watchlist WHERE active=1`)
if err != nil {
return nil, err
}
defer rows.Close()
var tickers []string
for rows.Next() {
var t string
if err := rows.Scan(&t); err != nil {
return nil, err
}
tickers = append(tickers, t)
}
return tickers, nil
}
func (p *Poller) insertNews(ticker string, item NewsItem) bool {
published := time.Unix(item.Datetime, 0).UTC().Format(time.RFC3339)
res, err := p.db.Exec(`
INSERT OR IGNORE INTO news (finnhub_id, ticker, headline, source, url, published_at)
VALUES (?, ?, ?, ?, ?, ?)
`, item.ID, ticker, item.Headline, item.Source, item.URL, published)
if err != nil {
return false
}
n, _ := res.RowsAffected()
return n > 0
}