quant/collector
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

optz

Browse Source
This commit is contained in:
yanweidong
2026-04-07 14:46:49 +08:00
parent aaa8308a90
commit 35e52bc5ea
14 changed files with 234 additions and 620 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
collector/collector.go
View File

@@ -9,7 +9,7 @@ import (
"net/http"
"time"
"git.apinb.com/quant/collector/models"
"git.apinb.com/quant/collector/types"
)
// Collector 数据采集器
@@ -31,7 +31,7 @@ func NewCollector(url string) *Collector {
}
// FetchData 从HTTP接口获取数据
func (c *Collector) FetchData() (*models.Status, error) {
func (c *Collector) FetchData() (*types.Status, error) {
resp, err := c.httpClient.Get(c.url)
if err != nil {
return nil, fmt.Errorf("HTTP请求失败: %w", err)
@@ -47,7 +47,7 @@ func (c *Collector) FetchData() (*models.Status, error) {
return nil, fmt.Errorf("读取响应失败: %w", err)
}
var status models.Status
var status types.Status
if err := json.Unmarshal(body, &status); err != nil {
return nil, fmt.Errorf("JSON解析失败: %w", err)
}
@@ -56,7 +56,7 @@ func (c *Collector) FetchData() (*models.Status, error) {
}
// CalculateHash 计算数据的SHA256哈希值
func (c *Collector) CalculateHash(status *models.Status) (string, error) {
func (c *Collector) CalculateHash(status *types.Status) (string, error) {
// 将数据序列化为JSON
data, err := json.Marshal(status)
if err != nil {
@@ -87,7 +87,7 @@ func (c *Collector) GetLastHash() string {
}
// CollectAndCheck 采集数据并检查是否变化
func (c *Collector) CollectAndCheck() (*models.Status, string, bool, error) {
func (c *Collector) CollectAndCheck() (*types.Status, string, bool, error) {
// 获取数据
status, err := c.FetchData()
if err != nil {

4
collector/collector_test.go
View File

@@ -5,7 +5,7 @@ import (
"os"
"testing"
"git.apinb.com/quant/collector/models"
"git.apinb.com/quant/collector/types"
)
// TestCalculateHash 测试Hash计算功能
@@ -16,7 +16,7 @@ func TestCalculateHash(t *testing.T) {
t.Fatalf("读取样本文件失败: %v", err)
}
var status models.Status
var status types.Status
if err := json.Unmarshal(data, &status); err != nil {
t.Fatalf("JSON解析失败: %v", err)
}