migrate from github

internal/repository/membership_repo/update.go

@@ -0,0 +1,56 @@
+package membership_repo
+
+import (
+	"github.com/XingfenD/yoresee_doc/internal/model"
+	"gorm.io/gorm"
+)
+
+type UpdateUserGroupOperation struct {
+	repo  *MembershipRepository
+	group *model.UserGroupMeta
+	tx    *gorm.DB
+}
+
+func (r *MembershipRepository) UpdateUserGroup(group *model.UserGroupMeta) *UpdateUserGroupOperation {
+	return &UpdateUserGroupOperation{
+		repo:  r,
+		group: group,
+	}
+}
+
+func (op *UpdateUserGroupOperation) WithTx(tx *gorm.DB) *UpdateUserGroupOperation {
+	op.tx = tx
+	return op
+}
+
+func (op *UpdateUserGroupOperation) Exec() error {
+	if op.tx != nil {
+		return op.tx.Save(op.group).Error
+	}
+	return op.repo.db.Save(op.group).Error
+}
+
+type UpdateOrgNodeOperation struct {
+	repo    *MembershipRepository
+	orgNode *model.OrgNodeMeta
+	tx      *gorm.DB
+}
+
+func (r *MembershipRepository) UpdateOrgNode(orgNode *model.OrgNodeMeta) *UpdateOrgNodeOperation {
+	return &UpdateOrgNodeOperation{
+		repo:    r,
+		orgNode: orgNode,
+	}
+}
+
+func (op *UpdateOrgNodeOperation) WithTx(tx *gorm.DB) *UpdateOrgNodeOperation {
+	op.tx = tx
+	return op
+}
+
+func (op *UpdateOrgNodeOperation) Exec() error {
+	if op.tx != nil {
+		return op.tx.Save(op.orgNode).Error
+	}
+	return op.repo.db.Save(op.orgNode).Error
+}