dtm/dtmcli/message.go
2021-08-11 15:46:25 +08:00

40 lines
875 B
Go

package dtmcli
// Msg reliable msg type
type Msg struct {
TransBase
Steps []MsgStep `json:"steps"`
QueryPrepared string `json:"query_prepared"`
}
// MsgStep struct of one step msg
type MsgStep struct {
Action string `json:"action"`
Data string `json:"data"`
}
// NewMsg create new msg
func NewMsg(server string, gid string) *Msg {
return &Msg{TransBase: *NewTransBase(gid, "msg", server, "")}
}
// Add add a new step
func (s *Msg) Add(action string, postData interface{}) *Msg {
s.Steps = append(s.Steps, MsgStep{
Action: action,
Data: MustMarshalString(postData),
})
return s
}
// Prepare prepare the msg
func (s *Msg) Prepare(queryPrepared string) error {
s.QueryPrepared = OrString(queryPrepared, s.QueryPrepared)
return s.callDtm(s, "prepare")
}
// Submit submit the msg
func (s *Msg) Submit() error {
return s.callDtm(s, "submit")
}