package flowgraph

import "github.com/zilliztech/milvus-distributed/internal/msgstream"

type Msg interface {
	TimeTick() Timestamp
	DownStreamNodeIdx() int
}

type MsgStreamMsg struct {
	tsMessages   []msgstream.TsMsg
	timestampMin Timestamp
	timestampMax Timestamp
}

func (msMsg *MsgStreamMsg) TimeTick() Timestamp {
	return msMsg.timestampMax
}

func (msMsg *MsgStreamMsg) DownStreamNodeIdx() int {
	return 0
}

func (msMsg *MsgStreamMsg) TsMessages() []msgstream.TsMsg {
	return msMsg.tsMessages
}

func (msMsg *MsgStreamMsg) TimestampMin() Timestamp {
	return msMsg.timestampMin
}

func (msMsg *MsgStreamMsg) TimestampMax() Timestamp {
	return msMsg.timestampMax
}