package dtmgrpc

import (
	context "context"

	"github.com/yedf/dtm/dtmcli"
)

// SagaGrpc struct of saga
type SagaGrpc struct {
	dtmcli.SagaData
	dtmcli.TransBase
}

// NewSaga create a saga
func NewSaga(server string, gid string) *SagaGrpc {
	return &SagaGrpc{
		SagaData: dtmcli.SagaData{TransData: dtmcli.TransData{
			Gid:       gid,
			TransType: "saga",
		}},
		TransBase: dtmcli.TransBase{
			Dtm: server,
		},
	}
}

// Add add a saga step
func (s *SagaGrpc) Add(action string, compensate string, busiData []byte) *SagaGrpc {
	dtmcli.Logf("saga %s Add %s %s %v", s.SagaData.Gid, action, compensate, string(busiData))
	step := dtmcli.SagaStep{
		Action:     action,
		Compensate: compensate,
		Data:       string(busiData),
	}
	s.Steps = append(s.Steps, step)
	return s
}

// Submit submit the saga trans
func (s *SagaGrpc) Submit() error {
	_, err := MustGetDtmClient(s.Dtm).Submit(context.Background(), &DtmRequest{
		Gid:       s.Gid,
		TransType: s.TransType,
		Data:      dtmcli.MustMarshalString(&s.Steps),
	})
	return err
}