128 lines
		
	
	
		
			3.7 KiB
		
	
	
	
		
			Go
		
	
	
	
			
		
		
	
	
			128 lines
		
	
	
		
			3.7 KiB
		
	
	
	
		
			Go
		
	
	
	
| // 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"
 | |
| )
 | |
| 
 | |
| 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)
 | |
| 		}
 | |
| 	}
 | |
| 
 | |
| 	// TODO: function to recalc all counters
 | |
| 
 | |
| 	return nil
 | |
| }
 | |
| 
 | |
| func init() {
 | |
| 	Register(&Check{
 | |
| 		Title:     "Check consistency of database",
 | |
| 		Name:      "check-db-consistency",
 | |
| 		IsDefault: false,
 | |
| 		Run:       checkDBConsistency,
 | |
| 		Priority:  3,
 | |
| 	})
 | |
| }
 |