Fix branch commits ount in pull mirror

Author: lunny

models/git/branch_list.go

@@ -163,3 +163,21 @@ func FindCommitsCountOutdatedBranches(ctx context.Context, startID, limit int64)
 	}
 	return branches, nil
 }
+
+func FindRepoCommitsCountOutdatedBranches(ctx context.Context, repoID, startID, limit int64) (BranchList, error) {
+	var branches BranchList
+	if err := db.GetEngine(ctx).
+		Join("INNER", "repository", "branch.repo_id = repository.id").
+		Where("branch.repo_id = ?", repoID).
+		And("repository.is_empty = ?", false).
+		And("id > ?", startID).
+		And(
+			builder.Expr("commit_count_id IS NULL").
+				Or(builder.Expr("commit_id <> commit_count_id")),
+		).
+		Asc("id").
+		Limit(int(limit)).Find(&branches); err != nil {
+		return nil, err
+	}
+	return branches, nil
+}

services/mirror/mirror_pull.go

@@ -410,6 +410,8 @@ func runSync(ctx context.Context, m *repo_model.Mirror) ([]*mirrorSyncResult, bo
 		log.Trace("SyncMirrors [repo: %-v Wiki]: git remote update complete", m.Repo)
 	}
 
+	repo_service.SyncRepoBranchesCommitsCount(ctx, m.Repo)
+
 	m.UpdatedUnix = timeutil.TimeStampNow()
 	return parseRemoteUpdateOutput(output, m.GetRemoteName()), true
 }

services/repository/branch.go

@@ -737,6 +737,36 @@ func GetBranchDivergingInfo(ctx reqctx.RequestContext, baseRepo *repo_model.Repo
 	return info, nil
 }
 
+func SyncRepoBranchesCommitsCount(ctx context.Context, repo *repo_model.Repository) error {
+	startID := int64(0)
+	for {
+		select {
+		case <-ctx.Done():
+			return nil
+		default:
+		}
+
+		// search all branches commits count are not synced
+		branches, err := git_model.FindRepoCommitsCountOutdatedBranches(ctx, repo.ID, startID, 100)
+		if err != nil {
+			return err
+		}
+		if len(branches) == 0 {
+			return nil
+		}
+
+		for _, branch := range branches {
+			branch.Repo = repo
+			if branch.ID > startID {
+				startID = branch.ID
+			}
+			if err := syncBranchCommitsCount(ctx, branch); err != nil {
+				log.Error("syncBranchCommitsCount: %v", err)
+			}
+		}
+	}
+}
+
 func SyncBranchCommitsCount(ctx context.Context) error {
 	startID := int64(0)
 	for {