initial

usmarket/watch.go

@@ -0,0 +1,102 @@
+package usmarket
+
+import (
+	"log"
+	"sync"
+	"time"
+)
+
+var (
+	UsMarket *MarketIndex
+)
+
+// lock
+type MarketIndex struct {
+	sync.RWMutex
+	LastIndex map[string]*StockDataIndex
+	Stats     int // -1:未开市，0：开市，数据不正常，1: 开市，数据正常
+}
+
+func NewMarket() {
+	UsMarket = &MarketIndex{
+		LastIndex: make(map[string]*StockDataIndex),
+		Stats:     -1,
+	}
+}
+
+func (m *MarketIndex) Set(stats int, index map[string]*StockDataIndex) {
+	m.Lock()
+	defer m.Unlock()
+
+	m.Stats = stats
+	m.LastIndex = index
+}
+
+func (m *MarketIndex) Get() (int, map[string]*StockDataIndex) {
+	m.Lock()
+	defer m.Unlock()
+
+	return m.Stats, m.LastIndex
+}
+
+func CheckUsMarketOpen() int {
+	var err error
+	if !IsUSMarketOpen() {
+		UsMarket.Set(-1, nil)
+		return -1
+	}
+
+	newIndex, err := Baidu_UsIndex()
+	if err != nil {
+		log.Println("Baidu_UsIndex Error:", err)
+		UsMarket.Set(-1, nil)
+		return -1
+	}
+
+	_, LastStockIndex := UsMarket.Get()
+	if len(LastStockIndex) == 0 {
+		UsMarket.Set(0, newIndex)
+		return 0
+	}
+
+	// IXIC 纳斯达克,DJI 道琼斯,SPX 标普500
+	if LastStockIndex["IXIC"].LastPrice == newIndex["IXIC"].LastPrice && LastStockIndex["DJI"].LastPrice == newIndex["DJI"].LastPrice && LastStockIndex["SPX"].LastPrice == newIndex["SPX"].LastPrice {
+		// log.Println("ERROR #103, IXIC,DJI,SPX LastPrice no change")
+		UsMarket.Set(-1, LastStockIndex)
+		return 0
+	}
+
+	if newIndex["IXIC"].Increase == "0.00" && newIndex["DJI"].Increase == "0.00" && newIndex["SPX"].Increase == "0.00" {
+		// log.Println("ERROR #104, IXIC,DJI,SPX Increase is 0")
+		UsMarket.Set(-1, LastStockIndex)
+		return 0
+	}
+
+	// jsonBytes, _ := json.Marshal(newIndex)
+	// log.Println("USMarket:", string(jsonBytes))
+	UsMarket.Set(1, newIndex)
+	return 1
+}
+
+// 判断美股是否开市
+func IsUSMarketOpen() bool {
+	// 获取当前时间
+	now := time.Now()
+
+	// 设置美国东部时间
+	loc, _ := time.LoadLocation("America/New_York")
+
+	// 将当前时间转换为美国东部时间
+	nowInNY := now.In(loc)
+
+	// 判断是否为周末
+	if nowInNY.Weekday() == time.Saturday || nowInNY.Weekday() == time.Sunday {
+		return false
+	}
+
+	// 判断是否在交易时间内
+	openTime := time.Date(nowInNY.Year(), nowInNY.Month(), nowInNY.Day(), 9, 30, 0, 0, loc)
+	closeTime := time.Date(nowInNY.Year(), nowInNY.Month(), nowInNY.Day(), 16, 0, 0, 0, loc)
+
+	return nowInNY.After(openTime) && nowInNY.Before(closeTime)
+}