gitee.com/quant1x/engine@v1.8.4/config/range_session.go

package config

import (
	"gitee.com/quant1x/gox/api"
	"gitee.com/quant1x/pkg/yaml"
	"slices"
	"strings"
)

// TradingSession 交易时段
type TradingSession struct {
	sessions []TimeRange
}

func (this TradingSession) String() string {
	builder := strings.Builder{}
	builder.WriteByte('[')
	var arr []string
	for _, timeRange := range this.sessions {
		arr = append(arr, timeRange.String())
	}
	builder.WriteString(strings.Join(arr, ","))
	builder.WriteByte(']')
	return builder.String()
}

func (this TradingSession) v2String() string {
	builder := strings.Builder{}
	//builder.WriteByte('[')
	var arr []string
	for _, timeRange := range this.sessions {
		arr = append(arr, timeRange.String())
	}
	builder.WriteString(strings.Join(arr, ","))
	//builder.WriteByte(']')
	return builder.String()
}

func (this *TradingSession) Parse(text string) error {
	var sessions []TimeRange
	text = strings.TrimSpace(text)
	arr := arrayRegexp.Split(text, -1)
	for _, v := range arr {
		var tr TimeRange
		err := tr.Parse(v)
		if err != nil {
			return err
		}
		sessions = append(sessions, tr)
	}
	slices.SortFunc(sessions, func(a, b TimeRange) int {
		if a.begin < b.begin {
			return -1
		} else if a.begin > b.begin {
			return 1
		} else if a.end < b.end {
			return -1
		} else if a.end == b.end {
			return 0
		} else {
			return 1
		}
	})
	if len(sessions) == 0 {
		return ErrTimeFormat
	}
	this.sessions = sessions
	return nil
}

func (this TradingSession) MarshalText() (text []byte, err error) {
	str := this.String()
	return api.String2Bytes(str), nil
}

// UnmarshalYAML YAML自定义解析
func (this *TradingSession) UnmarshalYAML(node *yaml.Node) error {
	var value string
	if len(node.Content) == 0 {
		value = node.Value
	} else if len(node.Content) == 2 {
		value = node.Content[1].Value
	} else {
		return ErrRangeFormat
	}

	return this.Parse(value)
}

// UnmarshalText 设置默认值调用
func (this *TradingSession) UnmarshalText(text []byte) error {
	return this.Parse(api.Bytes2String(text))
}

// Size 获取时段总数
func (this *TradingSession) Size() int {
	return len(this.sessions)
}

// Index 判断timestamp是第几个交易时段
func (this *TradingSession) Index(timestamp ...string) int {
	var tm string
	if len(timestamp) > 0 {
		tm = strings.TrimSpace(timestamp[0])
	} else {
		tm = getTradingTimestamp()
	}
	for i, timeRange := range this.sessions {
		if timeRange.IsTrading(tm) {
			return i
		}
	}
	return -1
}

// IsTrading 是否交易时段
func (this *TradingSession) IsTrading(timestamp ...string) bool {
	index := this.Index(timestamp...)
	if index < 0 {
		return false
	}
	return true
}

// IsTodayLastSession 当前时段是否今天最后一个交易时段
//
//	备选函数名 IsTodayFinalSession
func (this *TradingSession) IsTodayLastSession(timestamp ...string) bool {
	n := this.Size()
	index := this.Index(timestamp...)
	if index+1 < n {
		return false
	}
	return true
}

// CanStopLoss 当前时段是否可以进行止损操作
//
//	如果是3个时段, 止损操作在第2时段, 如果是4个时段, 止损在第3个
//	如果是2个时段, 则是第2个时段, 也就是最后一个时段
func (this *TradingSession) CanStopLoss(timestamp ...string) bool {
	n := this.Size()
	index := this.Index(timestamp...)
	// 1个时段, 立即止损
	c1 := n == 1
	// 2个时段, 在第二个时间止损
	c2 := n == 2 && index == 1
	// 3个以上时段, 在倒数第2个时段止损
	c3 := n >= 3 && index+2 == n
	if c1 || c2 || c3 {
		return true
	}
	return false
}

// CanTakeProfit 当前时段是否可以止盈
func (this *TradingSession) CanTakeProfit(timestamp ...string) bool {
	_ = timestamp
	return true
}