2020-02-11 04:34:17 -05:00
// Copyright 2020 The Gitea Authors. All rights reserved.
2022-11-27 13:20:29 -05:00
// SPDX-License-Identifier: MIT
2020-02-11 04:34:17 -05:00
package stats
import (
2021-11-30 15:06:32 -05:00
"fmt"
2021-12-09 20:27:50 -05:00
repo_model "code.gitea.io/gitea/models/repo"
2020-02-11 04:34:17 -05:00
"code.gitea.io/gitea/modules/git"
2021-11-30 15:06:32 -05:00
"code.gitea.io/gitea/modules/graceful"
2020-12-17 09:00:47 -05:00
"code.gitea.io/gitea/modules/log"
2021-11-30 15:06:32 -05:00
"code.gitea.io/gitea/modules/process"
2020-02-11 04:34:17 -05:00
)
// DBIndexer implements Indexer interface to use database's like search
2022-01-20 12:46:10 -05:00
type DBIndexer struct { }
2020-02-11 04:34:17 -05:00
// Index repository status function
func ( db * DBIndexer ) Index ( id int64 ) error {
2021-11-30 15:06:32 -05:00
ctx , _ , finished := process . GetManager ( ) . AddContext ( graceful . GetManager ( ) . ShutdownContext ( ) , fmt . Sprintf ( "Stats.DB Index Repo[%d]" , id ) )
defer finished ( )
2022-12-02 21:48:26 -05:00
repo , err := repo_model . GetRepositoryByID ( ctx , id )
2020-02-11 04:34:17 -05:00
if err != nil {
return err
}
2020-02-12 05:12:27 -05:00
if repo . IsEmpty {
return nil
}
2022-05-20 10:08:52 -04:00
status , err := repo_model . GetIndexerStatus ( ctx , repo , repo_model . RepoIndexerTypeStats )
2020-02-11 04:34:17 -05:00
if err != nil {
return err
}
2022-03-29 15:13:41 -04:00
gitRepo , err := git . OpenRepository ( ctx , repo . RepoPath ( ) )
2020-02-11 04:34:17 -05:00
if err != nil {
2022-02-24 23:22:09 -05:00
if err . Error ( ) == "no such file or directory" {
return nil
}
2020-02-11 04:34:17 -05:00
return err
}
defer gitRepo . Close ( )
// Get latest commit for default branch
commitID , err := gitRepo . GetBranchCommitID ( repo . DefaultBranch )
if err != nil {
2021-09-20 15:46:51 -04:00
if git . IsErrBranchNotExist ( err ) || git . IsErrNotExist ( err ) {
log . Debug ( "Unable to get commit ID for default branch %s in %s ... skipping this repository" , repo . DefaultBranch , repo . RepoPath ( ) )
2021-04-21 21:19:21 -04:00
return nil
}
2021-09-20 15:46:51 -04:00
log . Error ( "Unable to get commit ID for default branch %s in %s. Error: %v" , repo . DefaultBranch , repo . RepoPath ( ) , err )
2020-02-11 04:34:17 -05:00
return err
}
// Do not recalculate stats if already calculated for this commit
if status . CommitSha == commitID {
return nil
}
// Calculate and save language statistics to database
stats , err := gitRepo . GetLanguageStats ( commitID )
if err != nil {
2021-09-20 15:46:51 -04:00
log . Error ( "Unable to get language stats for ID %s for default branch %s in %s. Error: %v" , commitID , repo . DefaultBranch , repo . RepoPath ( ) , err )
2020-02-11 04:34:17 -05:00
return err
}
2021-12-09 20:27:50 -05:00
err = repo_model . UpdateLanguageStats ( repo , commitID , stats )
2021-09-20 15:46:51 -04:00
if err != nil {
log . Error ( "Unable to update language stats for ID %s for default branch %s in %s. Error: %v" , commitID , repo . DefaultBranch , repo . RepoPath ( ) , err )
return err
}
log . Debug ( "DBIndexer completed language stats for ID %s for default branch %s in %s. stats count: %d" , commitID , repo . DefaultBranch , repo . RepoPath ( ) , len ( stats ) )
return nil
2020-02-11 04:34:17 -05:00
}
// Close dummy function
func ( db * DBIndexer ) Close ( ) {
}