diff options
author | Unknwon <joe2010xtmf@163.com> | 2015-01-23 00:47:03 +0800 |
---|---|---|
committer | Unknwon <joe2010xtmf@163.com> | 2015-01-23 00:47:03 +0800 |
commit | c6b87aaf02b711d8b1ee7a40f57ba23f560e2a63 (patch) | |
tree | ed1feb5095ce25eeaf32367927bfe28a83228238 | |
parent | 077d1ae70be9ad6e4a2803e6f1fcb21efaca4fb1 (diff) | |
parent | 61608f13a04d6fb7b3f947dc98294640f3d243ac (diff) | |
download | gitea-c6b87aaf02b711d8b1ee7a40f57ba23f560e2a63.tar.gz gitea-c6b87aaf02b711d8b1ee7a40f57ba23f560e2a63.zip |
Merge branch 'dev' of github.com:gogits/gogs into dev
-rw-r--r-- | models/migrations/migrations.go | 53 | ||||
-rw-r--r-- | models/models.go | 6 |
2 files changed, 59 insertions, 0 deletions
diff --git a/models/migrations/migrations.go b/models/migrations/migrations.go new file mode 100644 index 0000000000..3586e5d0b6 --- /dev/null +++ b/models/migrations/migrations.go @@ -0,0 +1,53 @@ +package migrations + +import ( + "errors" + + "github.com/go-xorm/xorm" +) + +type migration func(*xorm.Engine) error + +// The version table. Should have only one row with id==1 +type Version struct { + Id int64 + Version int64 +} + +// This is a sequence of migrations. Add new migrations to the bottom of the list. +// If you want to "retire" a migration, replace it with "expiredMigration" +var migrations = []migration{} + +// Migrate database to current version +func Migrate(x *xorm.Engine) error { + if err := x.Sync(new(Version)); err != nil { + return err + } + + currentVersion := &Version{Id: 1} + has, err := x.Get(currentVersion) + if err != nil { + return err + } else if !has { + if _, err = x.InsertOne(currentVersion); err != nil { + return err + } + } + + v := currentVersion.Version + + for i, migration := range migrations[v:] { + if err = migration(x); err != nil { + return err + } + currentVersion.Version = v + int64(i) + 1 + if _, err = x.Id(1).Update(currentVersion); err != nil { + return err + } + } + return nil +} + +func expiredMigration(x *xorm.Engine) error { + return errors.New("You are migrating from a too old gogs version") +} diff --git a/models/models.go b/models/models.go index cf2124417a..55e7bf5822 100644 --- a/models/models.go +++ b/models/models.go @@ -15,6 +15,7 @@ import ( "github.com/go-xorm/xorm" _ "github.com/lib/pq" + "github.com/gogits/gogs/models/migrations" "github.com/gogits/gogs/modules/setting" ) @@ -131,6 +132,11 @@ func NewEngine() (err error) { if err = SetEngine(); err != nil { return err } + + if err = migrations.Migrate(x); err != nil { + return err + } + if err = x.StoreEngine("InnoDB").Sync2(tables...); err != nil { return fmt.Errorf("sync database struct error: %v\n", err) } |