migrate from github

internal/repository/template_repo/update.go

@@ -0,0 +1,63 @@
+package template_repo
+
+import (
+	"github.com/XingfenD/yoresee_doc/internal/model"
+	"gorm.io/gorm"
+)
+
+type TemplateUpdateOperation struct {
+	repo         *TemplateRepository
+	template     *model.Template
+	updateFields map[string]bool
+	tx           *gorm.DB
+}
+
+func (r *TemplateRepository) Update(template *model.Template) *TemplateUpdateOperation {
+	return &TemplateUpdateOperation{
+		repo:         r,
+		template:     template,
+		updateFields: make(map[string]bool),
+	}
+}
+
+func (op *TemplateUpdateOperation) UpdateName() *TemplateUpdateOperation {
+	op.updateFields["name"] = true
+	return op
+}
+
+func (op *TemplateUpdateOperation) UpdateDescription() *TemplateUpdateOperation {
+	op.updateFields["description"] = true
+	return op
+}
+
+func (op *TemplateUpdateOperation) UpdateScope() *TemplateUpdateOperation {
+	op.updateFields["scope"] = true
+	return op
+}
+
+func (op *TemplateUpdateOperation) UpdateIsPublic() *TemplateUpdateOperation {
+	op.updateFields["is_public"] = true
+	return op
+}
+
+func (op *TemplateUpdateOperation) WithTx(tx *gorm.DB) *TemplateUpdateOperation {
+	op.tx = tx
+	return op
+}
+
+func (op *TemplateUpdateOperation) Exec() error {
+	if op.tx == nil {
+		op.tx = op.repo.db
+	}
+
+	query := op.tx.Model(op.template)
+	if len(op.updateFields) > 0 {
+		fields := make([]string, 0, len(op.updateFields))
+		for field := range op.updateFields {
+			fields = append(fields, field)
+		}
+		query = query.Select(fields)
+	}
+
+	return query.Updates(*op.template).Error
+}