Skip to content

Commit 0771a79

Browse files
lunnywxiaoguang
andauthored
Use standalone function to update repository cols (#34811)
Extract `UpdateRepository` Follow up #34762 --------- Co-authored-by: wxiaoguang <wxiaoguang@gmail.com>
1 parent 750af1c commit 0771a79

File tree

4 files changed

+31
-49
lines changed

4 files changed

+31
-49
lines changed

models/repo/update.go

Lines changed: 6 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -40,21 +40,15 @@ func UpdateRepositoryUpdatedTime(ctx context.Context, repoID int64, updateTime t
4040
return err
4141
}
4242

43-
// UpdateRepositoryColsWithAutoTime updates repository's columns
44-
func UpdateRepositoryColsWithAutoTime(ctx context.Context, repo *Repository, cols ...string) error {
45-
if len(cols) == 0 {
46-
return nil
47-
}
48-
_, err := db.GetEngine(ctx).ID(repo.ID).Cols(cols...).Update(repo)
43+
// UpdateRepositoryColsWithAutoTime updates repository's columns and the timestamp fields automatically
44+
func UpdateRepositoryColsWithAutoTime(ctx context.Context, repo *Repository, colName string, moreColNames ...string) error {
45+
_, err := db.GetEngine(ctx).ID(repo.ID).Cols(append([]string{colName}, moreColNames...)...).Update(repo)
4946
return err
5047
}
5148

52-
// UpdateRepositoryColsNoAutoTime updates repository's columns and but applies time change automatically
53-
func UpdateRepositoryColsNoAutoTime(ctx context.Context, repo *Repository, cols ...string) error {
54-
if len(cols) == 0 {
55-
return nil
56-
}
57-
_, err := db.GetEngine(ctx).ID(repo.ID).Cols(cols...).NoAutoTime().Update(repo)
49+
// UpdateRepositoryColsNoAutoTime updates repository's columns, doesn't change timestamp field automatically
50+
func UpdateRepositoryColsNoAutoTime(ctx context.Context, repo *Repository, colName string, moreColNames ...string) error {
51+
_, err := db.GetEngine(ctx).ID(repo.ID).Cols(append([]string{colName}, moreColNames...)...).NoAutoTime().Update(repo)
5852
return err
5953
}
6054

modules/repository/branch.go

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -41,11 +41,12 @@ func SyncRepoBranchesWithRepo(ctx context.Context, repo *repo_model.Repository,
4141
if err != nil {
4242
return 0, fmt.Errorf("GetObjectFormat: %w", err)
4343
}
44-
_, err = db.GetEngine(ctx).ID(repo.ID).Update(&repo_model.Repository{ObjectFormatName: objFmt.Name()})
45-
if err != nil {
46-
return 0, fmt.Errorf("UpdateRepository: %w", err)
44+
if objFmt.Name() != repo.ObjectFormatName {
45+
repo.ObjectFormatName = objFmt.Name()
46+
if err = repo_model.UpdateRepositoryColsWithAutoTime(ctx, repo, "object_format_name"); err != nil {
47+
return 0, fmt.Errorf("UpdateRepositoryColsWithAutoTime: %w", err)
48+
}
4749
}
48-
repo.ObjectFormatName = objFmt.Name() // keep consistent with db
4950

5051
allBranches := container.Set[string]{}
5152
{

routers/private/hook_post_receive.go

Lines changed: 10 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -207,25 +207,19 @@ func HookPostReceive(ctx *gitea_context.PrivateContext) {
207207
return
208208
}
209209

210-
cols := make([]string, 0, 2)
211-
212-
if isPrivate.Has() {
210+
// FIXME: these options are not quite right, for example: changing visibility should do more works than just setting the is_private flag
211+
// These options should only be used for "push-to-create"
212+
if isPrivate.Has() && repo.IsPrivate != isPrivate.Value() {
213+
// TODO: it needs to do more work
213214
repo.IsPrivate = isPrivate.Value()
214-
cols = append(cols, "is_private")
215+
if err = repo_model.UpdateRepositoryColsNoAutoTime(ctx, repo, "is_private"); err != nil {
216+
ctx.JSON(http.StatusInternalServerError, private.HookPostReceiveResult{Err: "Failed to change visibility"})
217+
}
215218
}
216-
217-
if isTemplate.Has() {
219+
if isTemplate.Has() && repo.IsTemplate != isTemplate.Value() {
218220
repo.IsTemplate = isTemplate.Value()
219-
cols = append(cols, "is_template")
220-
}
221-
222-
if len(cols) > 0 {
223-
if err := repo_model.UpdateRepositoryColsNoAutoTime(ctx, repo, cols...); err != nil {
224-
log.Error("Failed to Update: %s/%s Error: %v", ownerName, repoName, err)
225-
ctx.JSON(http.StatusInternalServerError, private.HookPostReceiveResult{
226-
Err: fmt.Sprintf("Failed to Update: %s/%s Error: %v", ownerName, repoName, err),
227-
})
228-
return
221+
if err = repo_model.UpdateRepositoryColsNoAutoTime(ctx, repo, "is_template"); err != nil {
222+
ctx.JSON(http.StatusInternalServerError, private.HookPostReceiveResult{Err: "Failed to change template status"})
229223
}
230224
}
231225
}

routers/web/repo/setting/setting.go

Lines changed: 10 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -663,44 +663,37 @@ func handleSettingsPostAdvanced(ctx *context.Context) {
663663
func handleSettingsPostSigning(ctx *context.Context) {
664664
form := web.GetForm(ctx).(*forms.RepoSettingForm)
665665
repo := ctx.Repo.Repository
666-
changed := false
667666
trustModel := repo_model.ToTrustModel(form.TrustModel)
668667
if trustModel != repo.TrustModel {
669668
repo.TrustModel = trustModel
670-
changed = true
671-
}
672-
673-
if changed {
674-
if err := repo_service.UpdateRepository(ctx, repo, false); err != nil {
675-
ctx.ServerError("UpdateRepository", err)
669+
if err := repo_model.UpdateRepositoryColsNoAutoTime(ctx, repo, "trust_model"); err != nil {
670+
ctx.ServerError("UpdateRepositoryColsNoAutoTime", err)
676671
return
677672
}
673+
log.Trace("Repository signing settings updated: %s/%s", ctx.Repo.Owner.Name, repo.Name)
678674
}
679-
log.Trace("Repository signing settings updated: %s/%s", ctx.Repo.Owner.Name, repo.Name)
680675

681676
ctx.Flash.Success(ctx.Tr("repo.settings.update_settings_success"))
682677
ctx.Redirect(ctx.Repo.RepoLink + "/settings")
683678
}
684679

685680
func handleSettingsPostAdmin(ctx *context.Context) {
686-
form := web.GetForm(ctx).(*forms.RepoSettingForm)
687-
repo := ctx.Repo.Repository
688681
if !ctx.Doer.IsAdmin {
689682
ctx.HTTPError(http.StatusForbidden)
690683
return
691684
}
692685

686+
repo := ctx.Repo.Repository
687+
form := web.GetForm(ctx).(*forms.RepoSettingForm)
693688
if repo.IsFsckEnabled != form.EnableHealthCheck {
694689
repo.IsFsckEnabled = form.EnableHealthCheck
690+
if err := repo_model.UpdateRepositoryColsNoAutoTime(ctx, repo, "is_fsck_enabled"); err != nil {
691+
ctx.ServerError("UpdateRepositoryColsNoAutoTime", err)
692+
return
693+
}
694+
log.Trace("Repository admin settings updated: %s/%s", ctx.Repo.Owner.Name, repo.Name)
695695
}
696696

697-
if err := repo_service.UpdateRepository(ctx, repo, false); err != nil {
698-
ctx.ServerError("UpdateRepository", err)
699-
return
700-
}
701-
702-
log.Trace("Repository admin settings updated: %s/%s", ctx.Repo.Owner.Name, repo.Name)
703-
704697
ctx.Flash.Success(ctx.Tr("repo.settings.update_settings_success"))
705698
ctx.Redirect(ctx.Repo.RepoLink + "/settings")
706699
}

0 commit comments

Comments
 (0)