2020-12-02 07:56:04 +03:00
// Copyright 2020 The Gitea Authors. All rights reserved.
// Use of this source code is governed by a MIT-style
// license that can be found in the LICENSE file.
package doctor
import (
"context"
"code.gitea.io/gitea/models"
"code.gitea.io/gitea/models/migrations"
"code.gitea.io/gitea/modules/log"
2021-03-18 09:06:40 +03:00
"code.gitea.io/gitea/modules/setting"
2020-12-02 07:56:04 +03:00
)
func checkDBConsistency ( logger log . Logger , autofix bool ) error {
// make sure DB version is uptodate
if err := models . NewEngine ( context . Background ( ) , migrations . EnsureUpToDate ) ; err != nil {
logger . Critical ( "Model version on the database does not match the current Gitea version. Model consistency will not be checked until the database is upgraded" )
return err
}
// find labels without existing repo or org
count , err := models . CountOrphanedLabels ( )
if err != nil {
logger . Critical ( "Error: %v whilst counting orphaned labels" )
return err
}
if count > 0 {
if autofix {
if err = models . DeleteOrphanedLabels ( ) ; err != nil {
logger . Critical ( "Error: %v whilst deleting orphaned labels" )
return err
}
logger . Info ( "%d labels without existing repository/organisation deleted" , count )
} else {
logger . Warn ( "%d labels without existing repository/organisation" , count )
}
}
// find issues without existing repository
count , err = models . CountOrphanedIssues ( )
if err != nil {
logger . Critical ( "Error: %v whilst counting orphaned issues" )
return err
}
if count > 0 {
if autofix {
if err = models . DeleteOrphanedIssues ( ) ; err != nil {
logger . Critical ( "Error: %v whilst deleting orphaned issues" )
return err
}
logger . Info ( "%d issues without existing repository deleted" , count )
} else {
logger . Warn ( "%d issues without existing repository" , count )
}
}
// find pulls without existing issues
count , err = models . CountOrphanedObjects ( "pull_request" , "issue" , "pull_request.issue_id=issue.id" )
if err != nil {
logger . Critical ( "Error: %v whilst counting orphaned objects" )
return err
}
if count > 0 {
if autofix {
if err = models . DeleteOrphanedObjects ( "pull_request" , "issue" , "pull_request.issue_id=issue.id" ) ; err != nil {
logger . Critical ( "Error: %v whilst deleting orphaned objects" )
return err
}
logger . Info ( "%d pull requests without existing issue deleted" , count )
} else {
logger . Warn ( "%d pull requests without existing issue" , count )
}
}
// find tracked times without existing issues/pulls
count , err = models . CountOrphanedObjects ( "tracked_time" , "issue" , "tracked_time.issue_id=issue.id" )
if err != nil {
logger . Critical ( "Error: %v whilst counting orphaned objects" )
return err
}
if count > 0 {
if autofix {
if err = models . DeleteOrphanedObjects ( "tracked_time" , "issue" , "tracked_time.issue_id=issue.id" ) ; err != nil {
logger . Critical ( "Error: %v whilst deleting orphaned objects" )
return err
}
logger . Info ( "%d tracked times without existing issue deleted" , count )
} else {
logger . Warn ( "%d tracked times without existing issue" , count )
}
}
// find null archived repositories
count , err = models . CountNullArchivedRepository ( )
if err != nil {
logger . Critical ( "Error: %v whilst counting null archived repositories" )
return err
}
if count > 0 {
if autofix {
updatedCount , err := models . FixNullArchivedRepository ( )
if err != nil {
logger . Critical ( "Error: %v whilst fixing null archived repositories" )
return err
}
logger . Info ( "%d repositories with null is_archived updated" , updatedCount )
} else {
logger . Warn ( "%d repositories with null is_archived" , count )
}
}
2021-02-10 05:50:44 +03:00
// find label comments with empty labels
count , err = models . CountCommentTypeLabelWithEmptyLabel ( )
if err != nil {
logger . Critical ( "Error: %v whilst counting label comments with empty labels" )
return err
}
if count > 0 {
if autofix {
updatedCount , err := models . FixCommentTypeLabelWithEmptyLabel ( )
if err != nil {
logger . Critical ( "Error: %v whilst removing label comments with empty labels" )
return err
}
logger . Info ( "%d label comments with empty labels removed" , updatedCount )
} else {
logger . Warn ( "%d label comments with empty labels" , count )
}
}
2020-12-02 07:56:04 +03:00
// TODO: function to recalc all counters
2021-03-18 09:06:40 +03:00
if setting . Database . UsePostgreSQL {
count , err = models . CountBadSequences ( )
if err != nil {
logger . Critical ( "Error: %v whilst checking sequence values" )
}
if count > 0 {
if autofix {
err := models . FixBadSequences ( )
if err != nil {
logger . Critical ( "Error: %v whilst attempting to fix sequences" )
return err
}
logger . Info ( "%d sequences updated" , count )
} else {
logger . Warn ( "%d sequences with incorrect values" , count )
}
}
}
2020-12-02 07:56:04 +03:00
return nil
}
func init ( ) {
Register ( & Check {
Title : "Check consistency of database" ,
Name : "check-db-consistency" ,
IsDefault : false ,
Run : checkDBConsistency ,
Priority : 3 ,
} )
}