1、新增二板炸板情况统计；
2、更新项目结构；
This commit is contained in:
@@ -0,0 +1,231 @@
|
||||
package service
|
||||
|
||||
import (
|
||||
"sort"
|
||||
|
||||
"level_go/internal/model"
|
||||
)
|
||||
|
||||
type limitBoardHistoryRow struct {
|
||||
TsCode string `gorm:"column:ts_code"`
|
||||
TradeDate string `gorm:"column:trade_date"`
|
||||
}
|
||||
|
||||
func (s *LimitBoardService) GetLimitBoardOverview(tradeDate string) (model.LimitBoardResponse, error) {
|
||||
resp := model.LimitBoardResponse{TradeDate: tradeDate}
|
||||
|
||||
tradeDates, dateIndex, err := s.loadTradeCalendar(tradeDate)
|
||||
if err != nil {
|
||||
return resp, err
|
||||
}
|
||||
if dateIndex == -1 {
|
||||
return resp, nil
|
||||
}
|
||||
|
||||
resp.RecentMaxBoardStats, err = s.buildRecentMaxBoardStats(tradeDates, dateIndex)
|
||||
if err != nil {
|
||||
return resp, err
|
||||
}
|
||||
|
||||
limitStocks, err := s.loadCurrentLimitStocks(tradeDate)
|
||||
if err != nil {
|
||||
return resp, err
|
||||
}
|
||||
if len(limitStocks) == 0 {
|
||||
return resp, nil
|
||||
}
|
||||
|
||||
historyMap, err := s.loadLimitHistory(limitStocks, tradeDate)
|
||||
if err != nil {
|
||||
return resp, err
|
||||
}
|
||||
|
||||
groupMap, maxBoardCount := buildLimitBoardGroups(limitStocks, tradeDates, dateIndex, historyMap)
|
||||
resp.Groups = sortLimitBoardGroups(groupMap)
|
||||
resp.Summary = model.LimitBoardSummary{
|
||||
MaxBoardCount: maxBoardCount,
|
||||
LimitCount: len(limitStocks),
|
||||
GroupCount: len(resp.Groups),
|
||||
}
|
||||
|
||||
return resp, nil
|
||||
}
|
||||
|
||||
func (s *LimitBoardService) buildRecentMaxBoardStats(tradeDates []string, dateIndex int) ([]model.LimitBoardDailyMax, error) {
|
||||
startIndex := 0
|
||||
if dateIndex >= 29 {
|
||||
startIndex = dateIndex - 29
|
||||
}
|
||||
windowDates := tradeDates[startIndex : dateIndex+1]
|
||||
if len(windowDates) == 0 {
|
||||
return []model.LimitBoardDailyMax{}, nil
|
||||
}
|
||||
|
||||
var history []limitBoardHistoryRow
|
||||
if err := s.DB.Table(stockLimitCombinedTable).
|
||||
Select("ts_code, DATE_FORMAT(trade_date, '%Y-%m-%d') AS trade_date").
|
||||
Where("trade_date BETWEEN ? AND ? AND final_status = ?", windowDates[0], windowDates[len(windowDates)-1], 1).
|
||||
Order("ts_code ASC, trade_date ASC").
|
||||
Scan(&history).Error; err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
dateByCode := make(map[string]map[string]struct{})
|
||||
for _, item := range history {
|
||||
if _, ok := dateByCode[item.TsCode]; !ok {
|
||||
dateByCode[item.TsCode] = make(map[string]struct{})
|
||||
}
|
||||
dateByCode[item.TsCode][item.TradeDate] = struct{}{}
|
||||
}
|
||||
|
||||
maxBoardByDate := make(map[string]int, len(windowDates))
|
||||
for _, tradeDate := range windowDates {
|
||||
maxBoardByDate[tradeDate] = 0
|
||||
}
|
||||
|
||||
for _, limitDates := range dateByCode {
|
||||
for idx, tradeDate := range windowDates {
|
||||
if _, ok := limitDates[tradeDate]; !ok {
|
||||
continue
|
||||
}
|
||||
|
||||
boardCount := 0
|
||||
for j := idx; j >= 0; j-- {
|
||||
if _, ok := limitDates[windowDates[j]]; !ok {
|
||||
break
|
||||
}
|
||||
boardCount++
|
||||
}
|
||||
|
||||
if boardCount > maxBoardByDate[tradeDate] {
|
||||
maxBoardByDate[tradeDate] = boardCount
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
stats := make([]model.LimitBoardDailyMax, 0, len(windowDates))
|
||||
for _, tradeDate := range windowDates {
|
||||
stats = append(stats, model.LimitBoardDailyMax{
|
||||
TradeDate: tradeDate,
|
||||
MaxBoardCount: maxBoardByDate[tradeDate],
|
||||
})
|
||||
}
|
||||
|
||||
return stats, nil
|
||||
}
|
||||
|
||||
func (s *LimitBoardService) loadCurrentLimitStocks(tradeDate string) ([]model.LimitBoardStock, error) {
|
||||
var stocks []model.LimitBoardStock
|
||||
query := `SELECT
|
||||
l.symbol,
|
||||
l.ts_code,
|
||||
l.name,
|
||||
COALESCE(l.market, '') AS market,
|
||||
DATE_FORMAT(l.trade_date, '%Y-%m-%d') AS trade_date,
|
||||
COALESCE(l.calc_close, l.tushare_close, d.close, 0) AS close,
|
||||
COALESCE(l.calc_limit_price, 0) AS limit_price,
|
||||
COALESCE(l.tushare_pct_chg, d.pct_chg, 0) AS pct_chg,
|
||||
l.tushare_amount AS ak_amount,
|
||||
l.tushare_float_mv AS ak_float_market_value,
|
||||
l.tushare_total_mv AS ak_total_market_value,
|
||||
l.tushare_turnover_ratio AS ak_turnover_rate,
|
||||
l.tushare_fd_amount AS ak_seal_amount,
|
||||
l.tushare_first_time AS ak_first_limit_time,
|
||||
l.tushare_last_time AS ak_last_limit_time,
|
||||
l.tushare_open_times AS ak_break_limit_count,
|
||||
l.tushare_up_stat AS ak_limit_up_stat,
|
||||
l.tushare_limit_times AS ak_limit_up_days,
|
||||
l.tushare_industry AS ak_industry,
|
||||
l.calc_vol AS vol
|
||||
FROM stock_limit_combined_data l
|
||||
LEFT JOIN stock_daily d
|
||||
ON l.ts_code = d.ts_code AND d.trade_date = l.trade_date
|
||||
WHERE l.trade_date = ? AND l.final_status = 1
|
||||
ORDER BY d.pct_chg DESC, l.ts_code ASC`
|
||||
|
||||
err := s.DB.Raw(query, tradeDate).Scan(&stocks).Error
|
||||
return stocks, err
|
||||
}
|
||||
|
||||
func (s *LimitBoardService) loadLimitHistory(stocks []model.LimitBoardStock, tradeDate string) (map[string]map[string]struct{}, error) {
|
||||
codes := make([]string, 0, len(stocks))
|
||||
for _, stock := range stocks {
|
||||
codes = append(codes, stock.TsCode)
|
||||
}
|
||||
|
||||
var history []limitBoardHistoryRow
|
||||
if err := s.DB.Table(stockLimitCombinedTable).
|
||||
Select("ts_code, DATE_FORMAT(trade_date, '%Y-%m-%d') AS trade_date").
|
||||
Where("ts_code IN ? AND trade_date <= ? AND final_status = ?", codes, tradeDate, 1).
|
||||
Scan(&history).Error; err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
historyMap := make(map[string]map[string]struct{}, len(codes))
|
||||
for _, item := range history {
|
||||
if _, ok := historyMap[item.TsCode]; !ok {
|
||||
historyMap[item.TsCode] = make(map[string]struct{})
|
||||
}
|
||||
historyMap[item.TsCode][item.TradeDate] = struct{}{}
|
||||
}
|
||||
|
||||
return historyMap, nil
|
||||
}
|
||||
|
||||
func buildLimitBoardGroups(stocks []model.LimitBoardStock, tradeDates []string, dateIndex int, historyMap map[string]map[string]struct{}) (map[int][]model.LimitBoardStock, int) {
|
||||
groupMap := make(map[int][]model.LimitBoardStock)
|
||||
maxBoardCount := 0
|
||||
|
||||
for i := range stocks {
|
||||
boardCount := calcBoardCount(tradeDates, dateIndex, historyMap[stocks[i].TsCode])
|
||||
stocks[i].BoardCount = boardCount
|
||||
stocks[i].BoardLabel = boardLabel(boardCount)
|
||||
stocks[i].DisplayStatus = boardLabel(boardCount)
|
||||
|
||||
if boardCount > maxBoardCount {
|
||||
maxBoardCount = boardCount
|
||||
}
|
||||
groupMap[boardCount] = append(groupMap[boardCount], stocks[i])
|
||||
}
|
||||
|
||||
return groupMap, maxBoardCount
|
||||
}
|
||||
|
||||
func calcBoardCount(tradeDates []string, dateIndex int, limitDates map[string]struct{}) int {
|
||||
boardCount := 0
|
||||
for i := dateIndex; i >= 0; i-- {
|
||||
if _, ok := limitDates[tradeDates[i]]; !ok {
|
||||
break
|
||||
}
|
||||
boardCount++
|
||||
}
|
||||
return boardCount
|
||||
}
|
||||
|
||||
func sortLimitBoardGroups(groupMap map[int][]model.LimitBoardStock) []model.LimitBoardGroup {
|
||||
boardCounts := make([]int, 0, len(groupMap))
|
||||
for boardCount := range groupMap {
|
||||
boardCounts = append(boardCounts, boardCount)
|
||||
}
|
||||
sort.Sort(sort.Reverse(sort.IntSlice(boardCounts)))
|
||||
|
||||
groups := make([]model.LimitBoardGroup, 0, len(boardCounts))
|
||||
for _, boardCount := range boardCounts {
|
||||
stocks := groupMap[boardCount]
|
||||
sort.Slice(stocks, func(i, j int) bool {
|
||||
if stocks[i].PctChg == stocks[j].PctChg {
|
||||
return stocks[i].TsCode < stocks[j].TsCode
|
||||
}
|
||||
return stocks[i].PctChg > stocks[j].PctChg
|
||||
})
|
||||
|
||||
groups = append(groups, model.LimitBoardGroup{
|
||||
BoardCount: boardCount,
|
||||
Label: boardLabel(boardCount),
|
||||
Count: len(stocks),
|
||||
Stocks: stocks,
|
||||
})
|
||||
}
|
||||
|
||||
return groups
|
||||
}
|
||||
Reference in New Issue
Block a user