新增按交易所同步股票功能，支持 SSE/SZSE/BSE

backend/internal/handlers/data_sync.go

@@ -9,8 +9,37 @@ import (
 
 	"github.com/gin-gonic/gin"
 	"gorm.io/gorm"
+	"gorm.io/gorm/clause"
 )
 
+var validExchanges = map[string]string{
+	"SSE":  "上交所",
+	"SZSE": "深交所",
+	"BSE":  "北交所",
+}
+
+func toModelStock(s datasource.StockBasic) models.Stock {
+	return models.Stock{
+		TsCode:     s.TsCode,
+		Symbol:     s.Symbol,
+		Name:       s.Name,
+		Area:       s.Area,
+		Industry:   s.Industry,
+		Fullname:   s.Fullname,
+		Enname:     s.Enname,
+		Cnspell:    s.Cnspell,
+		Market:     s.Market,
+		Exchange:   s.Exchange,
+		CurrType:   s.CurrType,
+		ListStatus: s.ListStatus,
+		ListDate:   s.ListDate,
+		DelistDate: s.DelistDate,
+		IsHs:       s.IsHs,
+		ActName:    s.ActName,
+		ActEntType: s.ActEntType,
+	}
+}
+
 // DataSyncHandler 处理数据同步相关接口。
 type DataSyncHandler struct {
 	DB     *gorm.DB
@@ -40,16 +69,7 @@ func (h *DataSyncHandler) InitStocks(c *gin.Context) {
 
 	records := make([]models.Stock, 0, len(stocks))
 	for _, s := range stocks {
-		records = append(records, models.Stock{
-			TsCode:     s.TsCode,
-			Symbol:     s.Symbol,
-			Name:       s.Name,
-			Area:       s.Area,
-			Industry:   s.Industry,
-			Market:     s.Market,
-			Exchange:   s.Exchange,
-			ListStatus: s.ListStatus,
-		})
+		records = append(records, toModelStock(s))
 	}
 
 	if err := h.DB.Exec("TRUNCATE TABLE stocks RESTART IDENTITY").Error; err != nil {
@@ -72,3 +92,46 @@ func (h *DataSyncHandler) InitStocks(c *gin.Context) {
 		},
 	})
 }
+
+type syncStocksByExchangeResponse struct {
+	Exchange string `json:"exchange"`
+	Count    int    `json:"count"`
+	Message  string `json:"message"`
+}
+
+// SyncStocksByExchange 按交易所从 Tushare 同步股票基础信息。
+func (h *DataSyncHandler) SyncStocksByExchange(c *gin.Context) {
+	exchange := strings.ToUpper(strings.TrimSpace(c.Param("exchange")))
+	if _, ok := validExchanges[exchange]; !ok {
+		c.JSON(http.StatusBadRequest, gin.H{"success": false, "error": "无效的交易所代码，支持 SSE/SZSE/BSE"})
+		return
+	}
+
+	stocks, err := h.client.ListStocksByExchange(exchange)
+	if err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{"success": false, "error": err.Error()})
+		return
+	}
+
+	records := make([]models.Stock, 0, len(stocks))
+	for _, s := range stocks {
+		records = append(records, toModelStock(s))
+	}
+
+	if err := h.DB.Clauses(clause.OnConflict{
+		Columns:   []clause.Column{{Name: "ts_code"}},
+		UpdateAll: true,
+	}).CreateInBatches(records, 500).Error; err != nil {
+		c.JSON(http.StatusInternalServerError, gin.H{"success": false, "error": "写入股票列表失败: " + err.Error()})
+		return
+	}
+
+	c.JSON(http.StatusOK, gin.H{
+		"success": true,
+		"data": syncStocksByExchangeResponse{
+			Exchange: exchange,
+			Count:    len(records),
+			Message:  validExchanges[exchange] + "股票同步完成",
+		},
+	})
+}