// reset all migration // run all migration's down function func Reset() error { sm := sortMap(migrationMap) i := 0 for j := len(sm) - 1; j >= 0; j-- { v := sm[j] if isRollBack(v.name) { beego.Info("skip the", v.name) time.Sleep(1 * time.Second) continue } beego.Info("start reset:", v.name) v.m.Reset() v.m.Down() err := v.m.Exec(v.name, "down") if err != nil { beego.Error("execute error:", err) time.Sleep(2 * time.Second) return err } i++ beego.Info("end reset:", v.name) } beego.Info("total success reset:", i, " migration") time.Sleep(2 * time.Second) return nil }
func (c *ApiController) Get() { url := c.Input().Get("url") resp := utils.GetUrl(url, nil) html := resp.String() beego.Info(fmt.Sprint("Get from url[%s] return html length: [%d]", url, len(html))) var parser utils.ZingParser err, album := parser.ToAlbum(url, html) if err != nil { beego.Error(err.Error()) return } staticDir := beego.AppPath + "/" + beego.StaticDir["/file"] album.Folder = staticDir + "/" + album.Title // if err := os.Mkdir(album.Folder, 0777); err != nil { // beego.Error(err.Error()) // return // } beego.Info("Album folder: " + album.Folder + ".zip") zfile, err := os.Create(album.Folder + ".zip") if err != nil { beego.Error(err.Error()) return } w := zip.NewWriter(zfile) // w := zip.NewWriter(c.Ctx.ResponseWriter) // album.Song = album.Song[0:1] for i, song := range album.Song { beego.Info("Downloading: " + song.Source) song.Filename = song.Title + " - " + song.Performer + "." + song.Type // song.Filename = album.Folder + "/" + song.Filename resp := grequests.Get(song.Source, nil) beego.Info("Download status:", resp.StatusCode) if resp.Error != nil { beego.Error(resp.Error.Error()) } else { beego.Info(fmt.Sprintf("Downloaded %s. Done: %d/%d.", song.Source, i+1, len(album.Song))) wr, err := w.Create(song.Filename) if err == nil { _, err = wr.Write(resp.Bytes()) } if err != nil { beego.Error(err.Error()) } } } // c.Ctx.Output.Header("Content-Description", "File Transfer") // c.Ctx.Output.Header("Content-Type", "application/octet-stream") // c.Ctx.Output.Header("Content-Transfer-Encoding", "binary") // c.Ctx.Output.Header("Expires", "0") // c.Ctx.Output.Header("Cache-Control", "must-revalidate") // c.Ctx.Output.Header("Pragma", "public") // c.Ctx.Output.Header("Content-Disposition", "attachment; filename="+album.Title+".zip") if err := w.Close(); err != nil { beego.Error(err.Error()) return } c.Data["json"] = `{"success":"true"}` c.ServeJson() }
//rollback the migration by the name func Rollback(name string) error { if v, ok := migrationMap[name]; ok { beego.Info("start rollback") v.Reset() v.Down() err := v.Exec(name, "down") if err != nil { beego.Error("execute error:", err) time.Sleep(2 * time.Second) return err } beego.Info("end rollback") time.Sleep(2 * time.Second) return nil } else { beego.Error("not exist the migrationMap name:" + name) time.Sleep(2 * time.Second) return errors.New("not exist the migrationMap name:" + name) } }
// execute the sql already add in the sql func (m *Migration) Exec(name, status string) error { o := orm.NewOrm() for _, s := range m.sqls { beego.Info("exec sql:", s) r := o.Raw(s) _, err := r.Exec() if err != nil { return err } } return m.addOrUpdateRecord(name, status) }
// upgrate the migration from lasttime func Upgrade(lasttime int64) error { sm := sortMap(migrationMap) i := 0 for _, v := range sm { if v.created > lasttime { beego.Info("start upgrade", v.name) v.m.Reset() v.m.Up() err := v.m.Exec(v.name, "up") if err != nil { beego.Error("execute error:", err) time.Sleep(2 * time.Second) return err } beego.Info("end upgrade:", v.name) i++ } } beego.Info("total success upgrade:", i, " migration") time.Sleep(2 * time.Second) return nil }
func isRollBack(name string) bool { o := orm.NewOrm() var maps []orm.Params num, err := o.Raw("select * from migrations where `name` = ? order by id_migration desc", name).Values(&maps) if err != nil { beego.Info("get name has error", err) return false } if num <= 0 { return false } if maps[0]["status"] == "rollback" { return true } return false }