market-data-service/adapter/tushare/adapter.go

package tushare

import (
	"context"
	"fmt"
	"strings"
	"time"

	"market-data-service/adapter"
)

// Adapter Tushare数据源适配器
type Adapter struct {
	client *Client
	config map[string]string
}

// NewAdapter 创建Tushare适配器
func NewAdapter() *Adapter {
	return &Adapter{}
}

// Connect 建立连接
func (a *Adapter) Connect(config map[string]string) error {
	token, ok := config["token"]
	if !ok || token == "" {
		return fmt.Errorf("tushare token is required")
	}

	a.client = NewClient(token)
	if baseURL, ok := config["base_url"]; ok && baseURL != "" {
		a.client.SetBaseURL(baseURL)
	}

	a.config = config
	return nil
}

// SubscribeTicks 订阅实时Tick（Tushare不支持实时推送，返回错误）
func (a *Adapter) SubscribeTicks(symbols []string, callback adapter.TickCallback) error {
	return fmt.Errorf("tushare does not support real-time tick subscription")
}

// FetchKLines 拉取历史K线
func (a *Adapter) FetchKLines(symbol, start, end, freq string) ([]adapter.KLineData, error) {
	// 判断是股票还是期货
	if strings.Contains(symbol, ".SH") || strings.Contains(symbol, ".SZ") || strings.Contains(symbol, ".BJ") {
		return a.fetchStockKLines(symbol, start, end, freq)
	}
	return a.fetchFuturesKLines(symbol, start, end, freq)
}

// fetchStockKLines 获取股票K线
func (a *Adapter) fetchStockKLines(symbol, start, end, freq string) ([]adapter.KLineData, error) {
	// 转换symbol格式: 000001.SZ -> 000001.SZ
	tsCode := symbol

	switch freq {
	case "1d", "", "day":
		return a.fetchStockDaily(tsCode, start, end)
	case "1m", "5m", "15m", "30m", "60m":
		return a.fetchStockMinute(tsCode, start, end, freq)
	default:
		return nil, fmt.Errorf("unsupported frequency: %s", freq)
	}
}

// fetchStockDaily 获取股票日线
func (a *Adapter) fetchStockDaily(tsCode, start, end string) ([]adapter.KLineData, error) {
	data, err := a.client.GetStockDaily(tsCode, start, end)
	if err != nil {
		return nil, err
	}

	result := make([]adapter.KLineData, len(data))
	for i, d := range data {
		tradeDate, _ := time.Parse("20060102", d.TradeDate)
		result[i] = adapter.KLineData{
			Symbol: d.TSCode,
			Time:   tradeDate.Unix(),
			Open:   d.Open,
			High:   d.High,
			Low:    d.Low,
			Close:  d.Close,
			Volume: int64(d.Volume * 100), // 手 -> 股
			Amount: d.Amount * 1000,       // 千元 -> 元
		}
	}
	return result, nil
}

// fetchStockMinute 获取股票分钟线
func (a *Adapter) fetchStockMinute(tsCode, start, end, freq string) ([]adapter.KLineData, error) {
	// 去掉'm'后缀
	freqNum := strings.TrimSuffix(freq, "m")
	
	data, err := a.client.GetStockMinute(tsCode, start, end, freqNum)
	if err != nil {
		return nil, err
	}

	result := make([]adapter.KLineData, len(data))
	for i, d := range data {
		// 解析时间格式: 2025-03-07 09:31:00
		tradeTime, _ := time.Parse("2006-01-02 15:04:05", d.TradeTime)
		result[i] = adapter.KLineData{
			Symbol: d.TSCode,
			Time:   tradeTime.Unix(),
			Open:   d.Open,
			High:   d.High,
			Low:    d.Low,
			Close:  d.Close,
			Volume: int64(d.Volume * 100),
			Amount: d.Amount * 1000,
		}
	}
	return result, nil
}

// fetchFuturesKLines 获取期货K线
func (a *Adapter) fetchFuturesKLines(symbol, start, end, freq string) ([]adapter.KLineData, error) {
	switch freq {
	case "1d", "", "day":
		return a.fetchFuturesDaily(symbol, start, end)
	case "1m", "5m", "15m", "30m", "60m":
		return a.fetchFuturesMinute(symbol, start, end, freq)
	default:
		return nil, fmt.Errorf("unsupported frequency: %s", freq)
	}
}

// fetchFuturesDaily 获取期货日线
func (a *Adapter) fetchFuturesDaily(tsCode, start, end string) ([]adapter.KLineData, error) {
	data, err := a.client.GetFuturesDaily(tsCode, start, end)
	if err != nil {
		return nil, err
	}

	result := make([]adapter.KLineData, len(data))
	for i, d := range data {
		tradeDate, _ := time.Parse("20060102", d.TradeDate)
		result[i] = adapter.KLineData{
			Symbol:       d.TSCode,
			Time:         tradeDate.Unix(),
			Open:         d.Open,
			High:         d.High,
			Low:          d.Low,
			Close:        d.Close,
			Volume:       int64(d.Volume),
			Amount:       d.Amount * 10000, // 万元 -> 元
			OpenInterest: int64(d.OpenInterest),
		}
	}
	return result, nil
}

// fetchFuturesMinute 获取期货分钟线
func (a *Adapter) fetchFuturesMinute(tsCode, start, end, freq string) ([]adapter.KLineData, error) {
	freqNum := strings.TrimSuffix(freq, "m")
	
	data, err := a.client.GetFuturesMinute(tsCode, start, end, freqNum)
	if err != nil {
		return nil, err
	}

	result := make([]adapter.KLineData, len(data))
	for i, d := range data {
		tradeTime, _ := time.Parse("2006-01-02 15:04:05", d.TradeTime)
		result[i] = adapter.KLineData{
			Symbol:       d.TSCode,
			Time:         tradeTime.Unix(),
			Open:         d.Open,
			High:         d.High,
			Low:          d.Low,
			Close:        d.Close,
			Volume:       int64(d.Volume),
			Amount:       d.Amount * 10000,
			OpenInterest: int64(d.OpenInterest),
		}
	}
	return result, nil
}

// FetchSymbols 获取标的列表
func (a *Adapter) FetchSymbols(assetType string) ([]adapter.SymbolInfo, error) {
	switch assetType {
	case "stock":
		return a.fetchStockSymbols()
	case "futures":
		return a.fetchFuturesSymbols()
	default:
		return nil, fmt.Errorf("unsupported asset type: %s", assetType)
	}
}

// fetchStockSymbols 获取股票列表
func (a *Adapter) fetchStockSymbols() ([]adapter.SymbolInfo, error) {
	data, err := a.client.GetStockBasic()
	if err != nil {
		return nil, err
	}

	result := make([]adapter.SymbolInfo, 0, len(data))
	for _, d := range data {
		// 只返回上市状态的
		if d.ListStatus != "L" {
			continue
		}
		result = append(result, adapter.SymbolInfo{
			SymbolID: d.TSCode,
			Name:     d.Name,
			Exchange: d.Exchange,
		})
	}
	return result, nil
}

// fetchFuturesSymbols 获取期货列表
func (a *Adapter) fetchFuturesSymbols() ([]adapter.SymbolInfo, error) {
	data, err := a.client.GetFuturesBasic("")
	if err != nil {
		return nil, err
	}

	result := make([]adapter.SymbolInfo, len(data))
	for i, d := range data {
		result[i] = adapter.SymbolInfo{
			SymbolID: d.TSCode,
			Name:     d.Name,
			Exchange: d.Exchange,
		}
	}
	return result, nil
}

// FetchTradingCalendar 获取交易日历
func (a *Adapter) FetchTradingCalendar(exchange, start, end string) ([]adapter.TradeCalData, error) {
	// Tushare交易所代码映射
	exchangeMap := map[string]string{
		"SH": "SSE",
		"SZ": "SZSE",
		"SHFE": "SHFE",
		"DCE":  "DCE",
		"CZCE": "CZCE",
		"CFFEX": "CFFEX",
		"INE":  "INE",
	}

	tsExchange, ok := exchangeMap[exchange]
	if !ok {
		tsExchange = "SSE"
	}

	data, err := a.client.GetTradeCal(tsExchange, start, end, -1)
	if err != nil {
		return nil, err
	}

	result := make([]adapter.TradeCalData, len(data))
	for i, d := range data {
		calDate, _ := time.Parse("20060102", d.CalDate)
		result[i] = adapter.TradeCalData{
			Date:        calDate,
			IsTradingDay: d.IsOpen == 1,
		}
	}
	return result, nil
}

// HealthCheck 健康检查
func (a *Adapter) HealthCheck() error {
	if a.client == nil {
		return fmt.Errorf("client not initialized")
	}
	// 尝试获取交易日历作为健康检查
	now := time.Now()
	start := now.AddDate(0, 0, -7).Format("20060102")
	end := now.Format("20060102")
	
	_, err := a.client.GetTradeCal("SSE", start, end, 1)
	return err
}

// Close 关闭连接
func (a *Adapter) Close() error {
	// Tushare是HTTP接口，无需关闭
	return nil
}

// GetClient 获取底层客户端（用于直接调用Tushare API）
func (a *Adapter) GetClient() *Client {
	return a.client
}