Add rollback

Author: htdcx

DBPacth/v1.0.1-v1.0.2.sql

@@ -0,0 +1,2 @@
+ALTER TABLE `package` 
+ADD COLUMN `deployment_version_id` INT NULL AFTER `deployment_id`;

code-push.sql

@@ -109,6 +109,7 @@ DROP TABLE IF EXISTS `package`;
 CREATE TABLE `package` (
   `id` int NOT NULL AUTO_INCREMENT,
   `deployment_id` int DEFAULT NULL,
+  `deployment_version_id` int DEFAULT NULL,
   `size` bigint DEFAULT NULL,
   `hash` varchar(256) DEFAULT NULL,
   `description` TEXT DEFAULT NULL,

main.go

@@ -39,6 +39,7 @@ func main() {
 		authApi.POST("/lsDeployment", request.App{}.LsDeployment)
 		authApi.GET("/lsApp", request.App{}.LsApp)
 		authApi.POST("/uploadBundle", request.App{}.UploadBundle)
+		authApi.POST("/rollback", request.App{}.Rollback)
 	}
 
 	g.Run(configs.Port)

model/deploymentVersion.go

@@ -31,3 +31,7 @@ func (DeploymentVersion) GetNewVersionByKeyDeploymentId(deploymentId int) *Deplo
 	}
 	return deploymentVersion
 }
+
+func (DeploymentVersion) UpdateCurrentPackage(id int, pid *int) {
+	userDb.Raw("update deployment_version set current_package=? where id=?", pid, id).Scan(&DeploymentVersion{})
+}

model/package.go

@@ -1,16 +1,17 @@
 package model
 
 type Package struct {
-	Id           *int    `gorm:"primarykey;autoIncrement;size:32"`
-	DeploymentId *int    `json:"deploymentId"`
-	Size         *int64  `json:"size"`
-	Hash         *string `json:"hash"`
-	Download     *string `json:"download"`
-	Active       *int    `json:"active"`
-	Failed       *int    `json:"failed"`
-	Installed    *int    `json:"installed"`
-	CreateTime   *int64  `json:"create_time"`
-	Description  *string `json:"description"`
+	Id                  *int    `gorm:"primarykey;autoIncrement;size:32"`
+	DeploymentId        *int    `json:"deploymentId"`
+	DeploymentVersionId *int    `json:"deploymentVersionId"`
+	Size                *int64  `json:"size"`
+	Hash                *string `json:"hash"`
+	Download            *string `json:"download"`
+	Active              *int    `json:"active"`
+	Failed              *int    `json:"failed"`
+	Installed           *int    `json:"installed"`
+	CreateTime          *int64  `json:"create_time"`
+	Description         *string `json:"description"`
 }
 
 func (Package) TableName() string {
@@ -28,3 +29,12 @@ func (Package) AddFailed(pid int) {
 func (Package) AddInstalled(pid int) {
 	userDb.Raw("update package set installed=installed+1 where id=?", pid).Scan(&Package{})
 }
+
+func (Package) GetRollbackPack(deploymentId int, lastPakcId int, deploymentVersionId int) *Package {
+	var lastPackage *Package
+	err := userDb.Where("deployment_id=?", deploymentId).Where("id<?", lastPakcId).Where("deployment_version_id", deploymentVersionId).Order("id desc").First(&lastPackage).Error
+	if err != nil {
+		return nil
+	}
+	return lastPackage
+}

request/app.go

@@ -111,15 +111,16 @@ func (App) CreateBundle(ctx *gin.Context) {
 		// uuid, _ := uuid.NewUUID()
 		// hash := uuid.String()
 		newPackage := model.Package{
-			DeploymentId: deployment.Id,
-			Size:         createBundleReq.Size,
-			Hash:         createBundleReq.Hash,
-			Download:     createBundleReq.DownloadUrl,
-			Description:  createBundleReq.Description,
-			Active:       utils.CreateInt(0),
-			Installed:    utils.CreateInt(0),
-			Failed:       utils.CreateInt(0),
-			CreateTime:   utils.GetTimeNow(),
+			DeploymentId:        deployment.Id,
+			DeploymentVersionId: deploymentVersion.Id,
+			Size:                createBundleReq.Size,
+			Hash:                createBundleReq.Hash,
+			Download:            createBundleReq.DownloadUrl,
+			Description:         createBundleReq.Description,
+			Active:              utils.CreateInt(0),
+			Installed:           utils.CreateInt(0),
+			Failed:              utils.CreateInt(0),
+			CreateTime:          utils.GetTimeNow(),
 		}
 		model.Create[model.Package](&newPackage)
 		deploymentVersion.CurrentPackage = newPackage.Id
@@ -423,3 +424,69 @@ func (App) DelDeployment(ctx *gin.Context) {
 		log.Panic(err.Error())
 	}
 }
+
+type rollbackReq struct {
+	AppName    *string `json:"appName" binding:"required"`
+	Deployment *string `json:"deployment" binding:"required"`
+	Version    *string `json:"version"`
+}
+
+func (App) Rollback(ctx *gin.Context) {
+	rollbackReq := rollbackReq{}
+	if err := ctx.ShouldBindBodyWith(&rollbackReq, binding.JSON); err == nil {
+		uid := ctx.MustGet(constants.GIN_USER_ID).(int)
+
+		app := model.App{}.GetAppByUidAndAppName(uid, *rollbackReq.AppName)
+		if app == nil {
+			log.Panic("App not found")
+		}
+		deployment := model.Deployment{}.GetByAppidAndName(*app.Id, *rollbackReq.Deployment)
+		if deployment == nil {
+			log.Panic("Deployment " + *rollbackReq.Deployment + " not found")
+		}
+
+		var deploymentVersion *model.DeploymentVersion
+		if deployment.VersionId != nil {
+			deploymentVersion = model.DeploymentVersion{}.GetByKeyDeploymentIdAndVersion(*deployment.Id, *rollbackReq.Version)
+		}
+		if deploymentVersion == nil {
+			log.Panic("Version not found")
+		}
+		if deploymentVersion.CurrentPackage == nil {
+			log.Panic("Package is last")
+		}
+		newPackage := model.Package{}.GetRollbackPack(*deployment.Id, *deploymentVersion.CurrentPackage, *deploymentVersion.Id)
+
+		userDb, _ := db.GetUserDB()
+		err := userDb.Transaction(func(tx *gorm.DB) error {
+			if newPackage == nil {
+				model.DeploymentVersion{}.UpdateCurrentPackage(*deploymentVersion.Id, nil)
+			} else {
+				model.DeploymentVersion{}.UpdateCurrentPackage(*deploymentVersion.Id, newPackage.Id)
+			}
+			return nil
+		})
+		if err != nil {
+			panic("RollbackError:" + err.Error())
+		}
+
+		if newPackage != nil {
+			ctx.JSON(http.StatusOK, gin.H{
+				"Success":    true,
+				"Version":    *deploymentVersion.AppVersion,
+				"PackId":     *newPackage.Id,
+				"Size":       *newPackage.Size,
+				"Hash":       *newPackage.Hash,
+				"CreateTime": *newPackage.CreateTime,
+			})
+		} else {
+			ctx.JSON(http.StatusOK, gin.H{
+				"Success": true,
+				"Version": *deploymentVersion.AppVersion,
+			})
+		}
+		redis.DelRedisObj(constants.REDIS_UPDATE_INFO + *deployment.Key + "*")
+	} else {
+		log.Panic(err.Error())
+	}
+}