deving

internal/logic/strategy/rule/up_date.go

@@ -1,6 +1,8 @@
 package rule
 
 import (
+	"time"
+
 	"git.apinb.com/quant/gostock/internal/logic/types"
 )
 
@@ -16,6 +18,18 @@ func NewUpDate() *UpDate {
 	}
 }
 
-func (r *UpDate) Run(in string) *types.RuleResult {
-	return &types.RuleResult{Key: r.Key, Name: r.Name, Score: 1, Desc: "暂不计算"}
+func (r *UpDate) Run(lastdate string) *types.RuleResult {
+	// 计算上市不足半年的股票
+	if lastdate == "" {
+		return &types.RuleResult{Key: r.Key, Name: r.Name, Score: -1, Desc: "无上市时间"}
+	}
+	lastDate, err := time.Parse("20060102", lastdate)
+	if err != nil {
+		return &types.RuleResult{Key: r.Key, Name: r.Name, Score: -1, Desc: "上市时间格式错误"}
+	}
+
+	if time.Now().Sub(lastDate) < time.Hour*24*365/2 {
+		return &types.RuleResult{Key: r.Key, Name: r.Name, Score: -1, Desc: "上市时间不足6个月"}
+	}
+	return &types.RuleResult{Key: r.Key, Name: r.Name, Score: 1, Desc: "上市超过半年"}
 }