add release plan instance code

Signed-off-by: Patrick Zhao <[email protected]>

Author: PetrusZ

pkg/microservice/aslan/core/common/repository/models/release_plan.go

@@ -51,6 +51,7 @@ type ReleasePlan struct {
 	ExecutingTime int64 `bson:"executing_time"      yaml:"executing_time"                  json:"executing_time"`
 	SuccessTime   int64 `bson:"success_time"        yaml:"success_time"                    json:"success_time"`
 
+	InstanceCode string        `bson:"instance_code"       yaml:"instance_code"                   json:"instance_code"`
 	HookSettings *HookSettings `bson:"hook_settings"       yaml:"hook_settings"                   json:"hook_settings"`
 
 	WaitForApproveExternalCheckTime int64  `bson:"wait_for_approve_external_check_time"       yaml:"wait_for_approve_external_check_time"                   json:"wait_for_approve_external_check_time"`

pkg/microservice/aslan/core/common/repository/models/workflow_v4.go

@@ -158,7 +158,7 @@ type Approval struct {
 type NativeApproval struct {
 	Timeout           int                   `bson:"timeout"                     yaml:"timeout"                    json:"timeout"`
 	ApproveUsers      []*User               `bson:"approve_users"               yaml:"approve_users"              json:"approve_users"`
-	FloatApproveUsers []*User               `bson:"-"                           yaml:"flat_approve_users"          json:"flat_approve_users"`
+	FloatApproveUsers []*User               `bson:"-"                           yaml:"flat_approve_users"         json:"flat_approve_users"`
 	NeededApprovers   int                   `bson:"needed_approvers"            yaml:"needed_approvers"           json:"needed_approvers"`
 	RejectOrApprove   config.ApprovalStatus `bson:"reject_or_approve"           yaml:"-"                          json:"reject_or_approve"`
 	// InstanceCode: native approval instance code, save for working after restart aslan

pkg/microservice/aslan/core/common/service/webhooknotify/types.go

@@ -156,6 +156,8 @@ type ReleasePlanHookBody struct {
 	UpdatedBy string `json:"updated_by"`
 	// 更新时间
 	UpdateTime int64 `json:"update_time"`
+	// 实例代码
+	InstanceCode string `json:"instance_code"`
 
 	// 发布任务列表
 	Jobs []*ReleasePlanHookJob `json:"jobs"`

pkg/microservice/aslan/core/release_plan/service/release_plan.go

@@ -18,6 +18,8 @@ package service
 
 import (
 	"context"
+	"crypto/sha256"
+	"encoding/hex"
 	"encoding/json"
 	"fmt"
 	"strconv"
@@ -31,6 +33,7 @@ import (
 	"github.com/koderover/zadig/v2/pkg/types"
 	"github.com/pkg/errors"
 	"github.com/samber/lo"
+	"go.mongodb.org/mongo-driver/bson"
 	"go.mongodb.org/mongo-driver/bson/primitive"
 	"go.mongodb.org/mongo-driver/mongo"
 
@@ -400,6 +403,14 @@ func UpdateReleasePlan(c *handler.Context, planID string, args *UpdateReleasePla
 	}
 	plan.HookSettings = hookSetting.ToHookSettings()
 
+	instanceCode, err := generateInstanceCode(plan)
+	if err != nil {
+		fmtErr := fmt.Errorf("failed generate instance code, err: %v", err)
+		log.Error(fmtErr)
+		return fmtErr
+	}
+	plan.InstanceCode = instanceCode
+
 	if err = mongodb.NewReleasePlanColl().UpdateByID(ctx, planID, plan); err != nil {
 		return errors.Wrap(err, "update plan")
 	}
@@ -881,6 +892,12 @@ func UpdateReleasePlanStatus(c *handler.Context, planID, targetStatus string, is
 		plan.WaitForExecuteExternalCheckTime = 0
 		plan.WaitForAllDoneExternalCheckTime = 0
 		plan.ExternalCheckFailedReason = ""
+		plan.InstanceCode, err = generateInstanceCode(plan)
+		if err != nil {
+			fmtErr := fmt.Errorf("failed generate instance code, err: %v", err)
+			log.Error(fmtErr)
+			return fmtErr
+		}
 	case config.ReleasePlanStatusExecuting:
 		if plan.Approval != nil && plan.Approval.Enabled == true && plan.Approval.Status != config.StatusPassed {
 			return errors.Errorf("approval status is %s, can not execute", plan.Approval.Status)
@@ -1293,13 +1310,14 @@ func UpdateReleasePlanHookSetting(c *handler.Context, req *models.ReleasePlanHoo
 
 type ReleasePlanCallBackBody struct {
 	ReleasePlanID string                                `json:"release_plan_id"`
+	InstanceCode  string                                `json:"instance_code"`
 	HookEvent     models.ReleasePlanHookEvent           `json:"hook_event"`
 	Result        setting.ReleasePlanCallBackResultType `json:"result"`
 	FailedReason  string                                `json:"failed_reason"`
 }
 
 func ReleasePlanHookCallback(c *handler.Context, callback *ReleasePlanCallBackBody) error {
-	log.Infof("release plan hook callback, id: %s, hook event: %s, result: %s, failed reason: %s", callback.ReleasePlanID, callback.HookEvent, callback.Result, callback.FailedReason)
+	log.Infof("release plan hook callback, id: %s, instance code: %s, hook event: %s, result: %s, failed reason: %s", callback.ReleasePlanID, callback.InstanceCode, callback.HookEvent, callback.Result, callback.FailedReason)
 
 	hookSetting, err := mongodb.NewSystemSettingColl().GetReleasePlanHookSetting()
 	if err != nil {
@@ -1335,6 +1353,12 @@ func ReleasePlanHookCallback(c *handler.Context, callback *ReleasePlanCallBackBo
 		return fmtErr
 	}
 
+	if releasePlan.InstanceCode != callback.InstanceCode {
+		fmtErr := fmt.Errorf("release plan instance code is not correct, name: %s, instance code: %s, expected: %s", releasePlan.Name, callback.InstanceCode, releasePlan.InstanceCode)
+		log.Error(fmtErr)
+		return fmtErr
+	}
+
 	if !hookEventStatusMap[releasePlan.Status] {
 		fmtErr := fmt.Errorf("release plan's status is not correct, status: %s", releasePlan.Status)
 		log.Error(fmtErr)
@@ -1512,6 +1536,7 @@ func convertReleasePlanToHookBody(plan *models.ReleasePlan, hookEvent commonmode
 		Name:                plan.Name,
 		Manager:             plan.Manager,
 		ManagerID:           plan.ManagerID,
+		InstanceCode:        plan.InstanceCode,
 		StartTime:           plan.StartTime,
 		EndTime:             plan.EndTime,
 		ScheduleExecuteTime: plan.ScheduleExecuteTime,
@@ -2056,3 +2081,37 @@ func waitForExternalCheck(plan *models.ReleasePlan, systemHookSetting *commonmod
 
 	return &nextStatus, shouldWait
 }
+
+func generateInstanceCode(plan *models.ReleasePlan) (string, error) {
+	newPlan := new(models.ReleasePlan)
+	err := util.DeepCopy(newPlan, plan)
+	if err != nil {
+		err := fmt.Errorf("failed deep copy plan, err: %v", err)
+		log.Error(err)
+		return "", err
+	}
+	newPlan.UpdateTime = 0
+	newPlan.UpdatedBy = ""
+	newPlan.InstanceCode = ""
+	newPlan.Status = ""
+	newPlan.PlanningTime = 0
+	newPlan.ApprovalTime = 0
+	newPlan.ExecutingTime = 0
+	newPlan.SuccessTime = 0
+	newPlan.WaitForApproveExternalCheckTime = 0
+	newPlan.WaitForExecuteExternalCheckTime = 0
+	newPlan.WaitForAllDoneExternalCheckTime = 0
+	newPlan.ExternalCheckFailedReason = ""
+
+	newPlanBytes, err := bson.Marshal(newPlan)
+	if err != nil {
+		err := fmt.Errorf("failed marshal plan, err: %v", err)
+		log.Error(err)
+		return "", err
+	}
+
+	log.Debugf("new plan bytes: %s", string(newPlanBytes))
+
+	sum := sha256.Sum256([]byte(newPlanBytes))
+	return hex.EncodeToString(sum[:])[:16], nil
+}