mirror of
https://github.com/hanzoai/forge.git
synced 2026-10-30 15:00:39 +00:00
zeekayandHanzo Dev
aa87442fdd
Source files keep their SPDX-License-Identifier and Hanzo copyright line; the Gitea, Gogs and LFS Test Server copyrights they used to repeat now live once in NOTICE, alongside LICENSE. The header lint accepts any copyright line above the SPDX line, so it no longer demands an upstream one. Co-authored-by: Hanzo Dev <[email protected]>
111 lines
3.0 KiB
Go
111 lines
3.0 KiB
Go
// Copyright 2026 Hanzo AI, Inc. All rights reserved.
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
package v1_14
|
|
|
|
import (
|
|
"errors"
|
|
"strconv"
|
|
|
|
"github.com/hanzoai/git/models/db"
|
|
"github.com/hanzoai/git/modules/log"
|
|
"github.com/hanzoai/git/modules/setting"
|
|
)
|
|
|
|
func UpdateCodeCommentReplies(x db.EngineMigration) error {
|
|
type Comment struct {
|
|
ID int64 `xorm:"pk autoincr"`
|
|
CommitSHA string `xorm:"VARCHAR(40)"`
|
|
Patch string `xorm:"TEXT patch"`
|
|
Invalidated bool
|
|
|
|
// Not extracted but used in the below query
|
|
Type int `xorm:"INDEX"`
|
|
Line int64 // - previous line / + proposed line
|
|
TreePath string
|
|
ReviewID int64 `xorm:"index"`
|
|
}
|
|
|
|
if err := x.Sync(new(Comment)); err != nil {
|
|
return err
|
|
}
|
|
|
|
sqlSelect := `SELECT comment.id as id, first.commit_sha as commit_sha, first.patch as patch, first.invalidated as invalidated`
|
|
sqlTail := ` FROM comment INNER JOIN (
|
|
SELECT C.id, C.review_id, C.line, C.tree_path, C.patch, C.commit_sha, C.invalidated
|
|
FROM comment AS C
|
|
WHERE C.type = 21
|
|
AND C.created_unix =
|
|
(SELECT MIN(comment.created_unix)
|
|
FROM comment
|
|
WHERE comment.review_id = C.review_id
|
|
AND comment.type = 21
|
|
AND comment.line = C.line
|
|
AND comment.tree_path = C.tree_path)
|
|
) AS first
|
|
ON comment.review_id = first.review_id
|
|
AND comment.tree_path = first.tree_path AND comment.line = first.line
|
|
WHERE comment.type = 21
|
|
AND comment.id != first.id
|
|
AND comment.commit_sha != first.commit_sha`
|
|
|
|
var (
|
|
sqlCmd string
|
|
start = 0
|
|
batchSize = 100
|
|
sess = x.NewSession()
|
|
)
|
|
defer sess.Close()
|
|
for {
|
|
if err := sess.Begin(); err != nil {
|
|
return err
|
|
}
|
|
|
|
if setting.Database.Type.IsMSSQL() {
|
|
if _, err := sess.Exec(sqlSelect + " INTO #temp_comments" + sqlTail); err != nil {
|
|
log.Error("unable to create temporary table")
|
|
return err
|
|
}
|
|
}
|
|
|
|
comments := make([]*Comment, 0, batchSize)
|
|
|
|
switch {
|
|
case setting.Database.Type.IsMySQL():
|
|
sqlCmd = sqlSelect + sqlTail + " LIMIT " + strconv.Itoa(batchSize) + ", " + strconv.Itoa(start)
|
|
case setting.Database.Type.IsPostgreSQL():
|
|
fallthrough
|
|
case setting.Database.Type.IsSQLite3():
|
|
sqlCmd = sqlSelect + sqlTail + " LIMIT " + strconv.Itoa(batchSize) + " OFFSET " + strconv.Itoa(start)
|
|
case setting.Database.Type.IsMSSQL():
|
|
sqlCmd = "SELECT TOP " + strconv.Itoa(batchSize) + " * FROM #temp_comments WHERE " +
|
|
"(id NOT IN ( SELECT TOP " + strconv.Itoa(start) + " id FROM #temp_comments ORDER BY id )) ORDER BY id"
|
|
default:
|
|
return errors.New("Unsupported database type")
|
|
}
|
|
|
|
if err := sess.SQL(sqlCmd).Find(&comments); err != nil {
|
|
log.Error("failed to select: %v", err)
|
|
return err
|
|
}
|
|
|
|
for _, comment := range comments {
|
|
if _, err := sess.Table("comment").ID(comment.ID).Cols("commit_sha", "patch", "invalidated").Update(comment); err != nil {
|
|
log.Error("failed to update comment[%d]: %v %v", comment.ID, comment, err)
|
|
return err
|
|
}
|
|
}
|
|
|
|
start += len(comments)
|
|
|
|
if err := sess.Commit(); err != nil {
|
|
return err
|
|
}
|
|
if len(comments) < batchSize {
|
|
break
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|