package examples

import (
	context "context"

	"github.com/gin-gonic/gin"
	"github.com/yedf/dtm/dtmcli"
	"github.com/yedf/dtm/dtmgrpc"
	emptypb "google.golang.org/protobuf/types/known/emptypb"
)

// XaGrpcClient XA client connection
var XaGrpcClient *dtmgrpc.XaGrpcClient = nil

func init() {
	setupFuncs["XaGrpcSetup"] = func(app *gin.Engine) {
		XaGrpcClient = dtmgrpc.NewXaGrpcClient(DtmGrpcServer, config.DB, BusiGrpc+"/examples.Busi/XaNotify")
	}
	addSample("grpc_xa", func() string {
		gid := dtmgrpc.MustGenGid(DtmGrpcServer)
		busiData := dtmcli.MustMarshal(&TransReq{Amount: 30})
		err := XaGrpcClient.XaGlobalTransaction(gid, func(xa *dtmgrpc.XaGrpc) error {
			_, err := xa.CallBranch(busiData, BusiGrpc+"/examples.Busi/TransOutXa")
			if err != nil {
				return err
			}
			_, err = xa.CallBranch(busiData, BusiGrpc+"/examples.Busi/TransInXa")
			return err
		})
		dtmcli.FatalIfError(err)
		return gid
	})
}

func (s *busiServer) XaNotify(ctx context.Context, in *dtmgrpc.BusiRequest) (*emptypb.Empty, error) {
	err := XaGrpcClient.HandleCallback(in.Info.Gid, in.Info.BranchID, in.Info.BranchType)
	return &emptypb.Empty{}, dtmgrpc.Result2Error(nil, err)
}