52 lines
1.1 KiB
Go
52 lines
1.1 KiB
Go
package dtmpb
|
|
|
|
import "github.com/yedf/dtm/dtmcli"
|
|
|
|
// MsgGrpc reliable msg type
|
|
type MsgGrpc struct {
|
|
MsgDataGrpc
|
|
dtmcli.TransBase
|
|
}
|
|
|
|
// MsgDataGrpc msg data
|
|
type MsgDataGrpc struct {
|
|
dtmcli.TransData
|
|
Steps []MsgStepGrpc `json:"steps"`
|
|
QueryPrepared string `json:"query_prepared"`
|
|
}
|
|
|
|
// MsgStepGrpc struct of one step msg
|
|
type MsgStepGrpc struct {
|
|
Action string `json:"action"`
|
|
Data string `json:"data"`
|
|
}
|
|
|
|
// NewMsgGrpc create new msg
|
|
func NewMsgGrpc(server string, gid string) *MsgGrpc {
|
|
return &MsgGrpc{
|
|
MsgDataGrpc: MsgDataGrpc{TransData: dtmcli.TransData{
|
|
Gid: gid,
|
|
TransType: "msg",
|
|
}},
|
|
TransBase: dtmcli.TransBase{
|
|
Dtm: server,
|
|
},
|
|
}
|
|
}
|
|
|
|
// Add add a new step
|
|
func (s *MsgGrpc) Add(action string, postData interface{}) *MsgGrpc {
|
|
dtmcli.Logf("msg %s Add %s %v", s.MsgDataGrpc.Gid, action, postData)
|
|
step := MsgStepGrpc{
|
|
Action: action,
|
|
Data: dtmcli.MustMarshalString(postData),
|
|
}
|
|
s.Steps = append(s.Steps, step)
|
|
return s
|
|
}
|
|
|
|
// Submit submit the msg
|
|
func (s *MsgGrpc) Submit() error {
|
|
return s.CallDtm(&s.MsgDataGrpc, "submit")
|
|
}
|