in server/datasource/mongo/kv/kv_dao.go [71:134]
func txnCreate(ctx context.Context, kv *model.KVDoc) (*model.KVDoc, error) {
taskSession, err := dmongo.GetClient().GetDB().Client().StartSession()
if err != nil {
return nil, err
}
if err = taskSession.StartTransaction(); err != nil {
return nil, err
}
defer taskSession.EndSession(ctx)
if err = mongo.WithSession(ctx, taskSession, func(sessionContext mongo.SessionContext) error {
collection := dmongo.GetClient().GetDB().Collection(mmodel.CollectionKV)
_, err = collection.InsertOne(sessionContext, kv)
if err != nil {
openlog.Error("create error", openlog.WithTags(openlog.Tags{
"err": err.Error(),
"kv": kv,
}))
errAbort := taskSession.AbortTransaction(sessionContext)
if errAbort != nil {
openlog.Error("fail to abort transaction", openlog.WithTags(openlog.Tags{
"err": errAbort.Error(),
"kv": kv,
}))
}
return err
}
task, err := sync.NewTask(kv.Domain, kv.Project, sync.CreateAction, datasource.ConfigResource, kv)
if err != nil {
openlog.Error("fail to create task" + err.Error())
errAbort := taskSession.AbortTransaction(sessionContext)
if errAbort != nil {
openlog.Error("fail to abort transaction", openlog.WithTags(openlog.Tags{
"err": errAbort.Error(),
"kv": kv,
}))
}
return err
}
collection = dmongo.GetClient().GetDB().Collection(mmodel.CollectionTask)
_, err = collection.InsertOne(sessionContext, task)
if err != nil {
openlog.Error("create task error", openlog.WithTags(openlog.Tags{
"err": err.Error(),
"task": task,
}))
errAbort := taskSession.AbortTransaction(sessionContext)
if errAbort != nil {
openlog.Error("abort transaction", openlog.WithTags(openlog.Tags{
"err": errAbort.Error(),
"task": task,
}))
}
return err
}
if err = taskSession.CommitTransaction(sessionContext); err != nil {
return err
}
return nil
}); err != nil {
openlog.Error(err.Error())
return nil, err
}
return kv, nil
}