summaryrefslogtreecommitdiffstats
path: root/models
diff options
context:
space:
mode:
Diffstat (limited to 'models')
-rw-r--r--models/repo_language_stats.go29
1 files changed, 29 insertions, 0 deletions
diff --git a/models/repo_language_stats.go b/models/repo_language_stats.go
index bf8f5f59d0..fd5bcbb8ec 100644
--- a/models/repo_language_stats.go
+++ b/models/repo_language_stats.go
@@ -144,3 +144,32 @@ func (repo *Repository) UpdateLanguageStats(commitID string, stats map[string]fl
return sess.Commit()
}
+
+// CopyLanguageStat Copy originalRepo language stat information to destRepo (use for forked repo)
+func CopyLanguageStat(originalRepo, destRepo *Repository) error {
+ sess := x.NewSession()
+ defer sess.Close()
+ if err := sess.Begin(); err != nil {
+ return err
+ }
+ RepoLang := make(LanguageStatList, 0, 6)
+ if err := sess.Where("`repo_id` = ?", originalRepo.ID).Desc("`percentage`").Find(&RepoLang); err != nil {
+ return err
+ }
+ if len(RepoLang) > 0 {
+ for i := range RepoLang {
+ RepoLang[i].ID = 0
+ RepoLang[i].RepoID = destRepo.ID
+ RepoLang[i].CreatedUnix = timeutil.TimeStampNow()
+ }
+ //update destRepo's indexer status
+ tmpCommitID := RepoLang[0].CommitID
+ if err := destRepo.updateIndexerStatus(sess, RepoIndexerTypeStats, tmpCommitID); err != nil {
+ return err
+ }
+ if _, err := sess.Insert(&RepoLang); err != nil {
+ return err
+ }
+ }
+ return sess.Commit()
+}