summaryrefslogtreecommitdiffstats
path: root/models/migrate.go
diff options
context:
space:
mode:
authorLunny Xiao <xiaolunwen@gmail.com>2019-05-07 09:12:51 +0800
committerGitHub <noreply@github.com>2019-05-07 09:12:51 +0800
commit08069dc4656fa53ee5dd25189e15012cb4f8acb2 (patch)
tree2e08cb239fef3221e55da75f106dfcb0140e08a1 /models/migrate.go
parent1c7c739eb9ea1d2ffdaed3c776c84d42858c0851 (diff)
downloadgitea-08069dc4656fa53ee5dd25189e15012cb4f8acb2.tar.gz
gitea-08069dc4656fa53ee5dd25189e15012cb4f8acb2.zip
Improve migrations to support migrating milestones/labels/issues/comments/pullrequests (#6290)
* add migrations * fix package dependency * fix lints * implements migrations except pull requests * add releases * migrating releases * fix bug * fix lint * fix migrate releases * fix tests * add rollback * pull request migtations * fix import * fix go module vendor * add tests for upload to gitea * more migrate options * fix swagger-check * fix misspell * add options on migration UI * fix log error * improve UI options on migrating * add support for username password when migrating from github * fix tests * remove comments and fix migrate limitation * improve error handles * migrate API will also support migrate milestones/labels/issues/pulls/releases * fix tests and remove unused codes * add DownloaderFactory and docs about how to create a new Downloader * fix misspell * fix migration docs * Add hints about migrate options on migration page * fix tests
Diffstat (limited to 'models/migrate.go')
-rw-r--r--models/migrate.go145
1 files changed, 145 insertions, 0 deletions
diff --git a/models/migrate.go b/models/migrate.go
new file mode 100644
index 0000000000..54f9a3e24c
--- /dev/null
+++ b/models/migrate.go
@@ -0,0 +1,145 @@
+// Copyright 2019 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 models
+
+import "github.com/go-xorm/xorm"
+
+// InsertIssue insert one issue to database
+func InsertIssue(issue *Issue, labelIDs []int64) error {
+ sess := x.NewSession()
+ if err := sess.Begin(); err != nil {
+ return err
+ }
+
+ if err := insertIssue(sess, issue, labelIDs); err != nil {
+ return err
+ }
+ return sess.Commit()
+}
+
+func insertIssue(sess *xorm.Session, issue *Issue, labelIDs []int64) error {
+ if issue.MilestoneID > 0 {
+ sess.Incr("num_issues")
+ if issue.IsClosed {
+ sess.Incr("num_closed_issues")
+ }
+ if _, err := sess.ID(issue.MilestoneID).NoAutoTime().Update(new(Milestone)); err != nil {
+ return err
+ }
+ }
+ if _, err := sess.NoAutoTime().Insert(issue); err != nil {
+ return err
+ }
+ var issueLabels = make([]IssueLabel, 0, len(labelIDs))
+ for _, labelID := range labelIDs {
+ issueLabels = append(issueLabels, IssueLabel{
+ IssueID: issue.ID,
+ LabelID: labelID,
+ })
+ }
+ if _, err := sess.Insert(issueLabels); err != nil {
+ return err
+ }
+ if !issue.IsPull {
+ sess.ID(issue.RepoID).Incr("num_issues")
+ if issue.IsClosed {
+ sess.Incr("num_closed_issues")
+ }
+ } else {
+ sess.ID(issue.RepoID).Incr("num_pulls")
+ if issue.IsClosed {
+ sess.Incr("num_closed_pulls")
+ }
+ }
+ if _, err := sess.NoAutoTime().Update(issue.Repo); err != nil {
+ return err
+ }
+
+ sess.Incr("num_issues")
+ if issue.IsClosed {
+ sess.Incr("num_closed_issues")
+ }
+ if _, err := sess.In("id", labelIDs).Update(new(Label)); err != nil {
+ return err
+ }
+
+ if issue.MilestoneID > 0 {
+ if _, err := sess.ID(issue.MilestoneID).SetExpr("completeness", "num_closed_issues * 100 / num_issues").Update(new(Milestone)); err != nil {
+ return err
+ }
+ }
+
+ return nil
+}
+
+// InsertComment inserted a comment
+func InsertComment(comment *Comment) error {
+ sess := x.NewSession()
+ defer sess.Close()
+ if err := sess.Begin(); err != nil {
+ return err
+ }
+ if _, err := sess.NoAutoTime().Insert(comment); err != nil {
+ return err
+ }
+ if _, err := sess.ID(comment.IssueID).Incr("num_comments").Update(new(Issue)); err != nil {
+ return err
+ }
+ return sess.Commit()
+}
+
+// InsertPullRequest inserted a pull request
+func InsertPullRequest(pr *PullRequest, labelIDs []int64) error {
+ sess := x.NewSession()
+ defer sess.Close()
+ if err := sess.Begin(); err != nil {
+ return err
+ }
+ if err := insertIssue(sess, pr.Issue, labelIDs); err != nil {
+ return err
+ }
+ pr.IssueID = pr.Issue.ID
+ if _, err := sess.NoAutoTime().Insert(pr); err != nil {
+ return err
+ }
+ return sess.Commit()
+}
+
+// MigrateRelease migrates release
+func MigrateRelease(rel *Release) error {
+ sess := x.NewSession()
+ if err := sess.Begin(); err != nil {
+ return err
+ }
+
+ var oriRel = Release{
+ RepoID: rel.RepoID,
+ TagName: rel.TagName,
+ }
+ exist, err := sess.Get(&oriRel)
+ if err != nil {
+ return err
+ }
+ if !exist {
+ if _, err := sess.NoAutoTime().Insert(rel); err != nil {
+ return err
+ }
+ } else {
+ rel.ID = oriRel.ID
+ if _, err := sess.ID(rel.ID).Cols("target, title, note, is_tag, num_commits").Update(rel); err != nil {
+ return err
+ }
+ }
+
+ for i := 0; i < len(rel.Attachments); i++ {
+ rel.Attachments[i].ReleaseID = rel.ID
+ }
+
+ if _, err := sess.NoAutoTime().Insert(rel.Attachments); err != nil {
+ return err
+ }
+
+ return sess.Commit()
+}