103 lines
2.3 KiB
Go
103 lines
2.3 KiB
Go
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)
|
||
}
|