support for target platforms

ayu-devtron · ayu-devtron · commit b7bff2bff8a2 · 2025-01-15T17:41:00.000+05:30
diff --git a/internal/sql/repository/CiArtifactRepository.go b/internal/sql/repository/CiArtifactRepository.go
@@ -77,6 +77,7 @@ type CiArtifact struct {
 	IsArtifactUploaded    bool      `sql:"is_artifact_uploaded"` // Deprecated; Use pipelineConfig.CiWorkflow instead.
 	CredentialsSourceType string    `sql:"credentials_source_type"`
 	CredentialSourceValue string    `sql:"credentials_source_value"`
+	TargetPlatforms       string    `sql:"target_platforms"`
 	ComponentId           int       `sql:"component_id"`
 	DeployedTime          time.Time `sql:"-"`
 	Deployed              bool      `sql:"-"`
diff --git a/internal/sql/repository/pipelineConfig/CiWorkflowRepository.go b/internal/sql/repository/pipelineConfig/CiWorkflowRepository.go
@@ -120,6 +120,7 @@ type WorkflowWithArtifact struct {
 	TriggeredBy             int32                           `sql:"triggered_by"`
 	EmailId                 string                          `sql:"email_id"`
 	Image                   string                          `sql:"image"`
+	TargetPlatforms         string                          `sql:"target_platforms"`
 	CiArtifactLocation      string                          `sql:"ci_artifact_location"`
 	CiArtifactId            int                             `sql:"ci_artifact_id"`
 	BlobStorageEnabled      bool                            `sql:"blob_storage_enabled"`
@@ -193,6 +194,7 @@ func (impl *CiWorkflowRepositoryImpl) FindByPipelineId(pipelineId int, offset in
 		ColumnExpr("cia.id AS ci_artifact_id").
 		ColumnExpr("env.environment_name").
 		ColumnExpr("cia.image").
+		ColumnExpr("cia.target_platforms").
 		ColumnExpr("cia.is_artifact_uploaded AS old_is_artifact_uploaded").
 		ColumnExpr("wf.*").
 		ColumnExpr("u.email_id").
diff --git a/pkg/bean/app.go b/pkg/bean/app.go
@@ -827,6 +827,7 @@ type CiArtifactBean struct {
 	CiWorkflowId                  int                       `json:"-"`
 	RegistryType                  string                    `json:"registryType"`
 	RegistryName                  string                    `json:"registryName"`
+	TargetPlatforms               []bean.TargetPlatform     `json:"targetPlatforms"`
 	CiPipelineId                  int                       `json:"-"`
 	CredentialsSourceType         string                    `json:"-"`
 	CredentialsSourceValue        string                    `json:"-"`
diff --git a/pkg/eventProcessor/bean/workflowEventBean.go b/pkg/eventProcessor/bean/workflowEventBean.go
@@ -83,6 +83,7 @@ type CiCompleteEvent struct {
 	PluginRegistryArtifactDetails map[string][]string      `json:"PluginRegistryArtifactDetails"`
 	PluginArtifactStage           string                   `json:"pluginArtifactStage"`
 	IsScanEnabled                 bool                     `json:"isScanEnabled"`
+	TargetPlatforms               []string                 `json:"targetPlatforms"`
 	pluginImageDetails            *registry.ImageDetailsFromCR
 	PluginArtifacts               *PluginArtifacts `json:"pluginArtifacts"`
 }
diff --git a/pkg/eventProcessor/in/WorkflowEventProcessorService.go b/pkg/eventProcessor/in/WorkflowEventProcessorService.go
@@ -678,6 +678,7 @@ func (impl *WorkflowEventProcessorImpl) BuildCiArtifactRequest(event bean.CiComp
 		PluginRegistryArtifactDetails: pluginArtifacts,
 		PluginArtifactStage:           event.PluginArtifactStage,
 		IsScanEnabled:                 event.IsScanEnabled,
+		TargetPlatforms:               event.TargetPlatforms,
 	}
 	// if DataSource is empty, repository.WEBHOOK is considered as default
 	if request.DataSource == "" {
diff --git a/pkg/pipeline/AppArtifactManager.go b/pkg/pipeline/AppArtifactManager.go
@@ -20,6 +20,7 @@ import (
 	argoApplication "github.com/devtron-labs/devtron/client/argocdServer/bean"
 	"github.com/devtron-labs/devtron/pkg/build/artifacts/imageTagging"
 	"github.com/devtron-labs/devtron/pkg/build/pipeline"
+	"github.com/devtron-labs/devtron/pkg/pipeline/adapter"
 	pipelineBean "github.com/devtron-labs/devtron/pkg/pipeline/bean"
 	"sort"
 	"strings"
@@ -402,6 +403,7 @@ func (impl *AppArtifactManagerImpl) BuildRollbackArtifactsList(artifactListingFi
 		deployedCiArtifacts = append(deployedCiArtifacts, bean2.CiArtifactBean{
 			Id:                     ciArtifact.Id,
 			Image:                  ciArtifact.Image,
+			TargetPlatforms:        adapter.GetTargetPlatformObjectFromString(ciArtifact.TargetPlatforms),
 			MaterialInfo:           mInfo,
 			DeployedTime:           formatDate(ciArtifact.StartedOn, bean2.LayoutRFC3339),
 			WfrId:                  ciArtifact.CdWorkflowRunnerId,
@@ -656,6 +658,7 @@ func (impl *AppArtifactManagerImpl) BuildArtifactsList(listingFilterOpts *bean.A
 		currentRunningArtifactBean = &bean2.CiArtifactBean{
 			Id:                     currentRunningArtifact.Id,
 			Image:                  currentRunningArtifact.Image,
+			TargetPlatforms:        adapter.GetTargetPlatformObjectFromString(currentRunningArtifact.TargetPlatforms),
 			ImageDigest:            currentRunningArtifact.ImageDigest,
 			MaterialInfo:           mInfo,
 			ScanEnabled:            currentRunningArtifact.ScanEnabled,
diff --git a/pkg/pipeline/CdHandler.go b/pkg/pipeline/CdHandler.go
@@ -27,6 +27,7 @@ import (
 	bean3 "github.com/devtron-labs/devtron/pkg/cluster/bean"
 	repository3 "github.com/devtron-labs/devtron/pkg/cluster/environment/repository"
 	common2 "github.com/devtron-labs/devtron/pkg/deployment/common"
+	adapter2 "github.com/devtron-labs/devtron/pkg/pipeline/adapter"
 	util2 "github.com/devtron-labs/devtron/pkg/pipeline/util"
 	"os"
 	"path/filepath"
@@ -611,13 +612,16 @@ func (impl *CdHandlerImpl) FetchCdWorkflowDetails(appId int, environmentId int,
 		triggeredByUserEmailId = "anonymous"
 	}
 	ciArtifactId := workflow.CiArtifactId
+	targetPlatforms := []pipelineBean.TargetPlatform{}
 	if ciArtifactId > 0 {
 		ciArtifact, err := impl.ciArtifactRepository.Get(ciArtifactId)
 		if err != nil {
 			impl.Logger.Errorw("error fetching artifact data", "err", err)
 			return types.WorkflowResponse{}, err
 		}
 
+		targetPlatforms = adapter2.GetTargetPlatformObjectFromString(ciArtifact.TargetPlatforms)
+
 		// handling linked ci pipeline
 		if ciArtifact.ParentCiArtifact > 0 && ciArtifact.WorkflowId == nil {
 			ciArtifactId = ciArtifact.ParentCiArtifact
@@ -673,6 +677,7 @@ func (impl *CdHandlerImpl) FetchCdWorkflowDetails(appId int, environmentId int,
 		ArtifactId:           workflow.CiArtifactId,
 		IsArtifactUploaded:   workflow.IsArtifactUploaded,
 		CiPipelineId:         ciWf.CiPipelineId,
+		TargetPlatforms:      targetPlatforms,
 	}
 	return workflowResponse, nil
 
@@ -776,6 +781,7 @@ func (impl *CdHandlerImpl) converterWFR(wfr pipelineConfig.CdWorkflowRunner) pip
 		workflow.WorkflowType = string(wfr.WorkflowType)
 		workflow.CdWorkflowId = wfr.CdWorkflowId
 		workflow.Image = wfr.CdWorkflow.CiArtifact.Image
+		workflow.TargetPlatforms = adapter2.GetTargetPlatformObjectFromString(wfr.CdWorkflow.CiArtifact.TargetPlatforms)
 		workflow.PipelineId = wfr.CdWorkflow.PipelineId
 		workflow.CiArtifactId = wfr.CdWorkflow.CiArtifactId
 		isArtifactUploaded, isMigrationRequired := wfr.GetIsArtifactUploaded()
diff --git a/pkg/pipeline/CiHandler.go b/pkg/pipeline/CiHandler.go
@@ -31,6 +31,7 @@ import (
 	bean5 "github.com/devtron-labs/devtron/pkg/cluster/bean"
 	"github.com/devtron-labs/devtron/pkg/cluster/environment"
 	repository2 "github.com/devtron-labs/devtron/pkg/cluster/environment/repository"
+	adapter2 "github.com/devtron-labs/devtron/pkg/pipeline/adapter"
 	util3 "github.com/devtron-labs/devtron/pkg/pipeline/util"
 	"io/ioutil"
 	"net/http"
@@ -571,6 +572,7 @@ func (impl *CiHandlerImpl) GetBuildHistory(pipelineId int, appId int, offset int
 			EnvironmentName:     w.EnvironmentName,
 			ReferenceWorkflowId: w.RefCiWorkflowId,
 			PodName:             w.PodName,
+			TargetPlatforms:     adapter2.GetTargetPlatformObjectFromString(w.TargetPlatforms),
 		}
 
 		if w.Message == bean3.ImageTagUnavailableMessage {
@@ -1597,6 +1599,7 @@ func (impl *CiHandlerImpl) FetchMaterialInfoByArtifactId(ciArtifactId int, envId
 		Default:          deployDetail.Default,
 		ImageTaggingData: *imageTaggingData,
 		Image:            ciArtifact.Image,
+		TargetPlatforms:  adapter2.GetTargetPlatformObjectFromString(ciArtifact.TargetPlatforms),
 	}
 	return gitTriggerInfoResponse, nil
 }
diff --git a/pkg/pipeline/adapter/adapter.go b/pkg/pipeline/adapter/adapter.go
@@ -27,6 +27,7 @@ import (
 	"github.com/devtron-labs/devtron/pkg/pipeline/repository"
 	"github.com/devtron-labs/devtron/pkg/pipeline/types"
 	"github.com/devtron-labs/devtron/pkg/sql"
+	"strings"
 	"time"
 )
 
@@ -377,4 +378,17 @@ func GetStepVariableDto(variable *repository.PipelineStageStepVariable) (*pipeli
 		VariableStepIndexInPlugin: variable.VariableStepIndexInPlugin,
 	}
 	return variableDto, nil
-}
+}
+
+func GetTargetPlatformObjectFromString(targetPlatform string) []pipelineConfigBean.TargetPlatform {
+	targetPlatforms := GetTargetPlatformListFromString(targetPlatform)
+	var targetPlatformObject []pipelineConfigBean.TargetPlatform
+	for _, targetPlatform := range targetPlatforms {
+		targetPlatformObject = append(targetPlatformObject, pipelineConfigBean.TargetPlatform{Name: targetPlatform})
+	}
+	return targetPlatformObject
+}
+
+func GetTargetPlatformListFromString(targetPlatform string) []string {
+	return strings.Split(targetPlatform, ",")
+}
diff --git a/pkg/pipeline/bean/ArtifactManagerBean.go b/pkg/pipeline/bean/ArtifactManagerBean.go
@@ -0,0 +1,5 @@
+package bean
+
+type TargetPlatform struct {
+	Name string `json:"name"`
+}
diff --git a/pkg/pipeline/bean/CdHandlerBean.go b/pkg/pipeline/bean/CdHandlerBean.go
@@ -21,6 +21,7 @@ type CdWorkflowWithArtifact struct {
 	TriggeredBy           int32                                       `json:"triggered_by"`
 	EmailId               string                                      `json:"email_id"`
 	Image                 string                                      `json:"image"`
+	TargetPlatforms       []TargetPlatform                            `json:"target_platforms"`
 	MaterialInfo          string                                      `json:"material_info,omitempty"`
 	DataSource            string                                      `json:"data_source,omitempty"`
 	CiArtifactId          int                                         `json:"ci_artifact_id,omitempty"`
diff --git a/pkg/pipeline/types/CiCdConfig.go b/pkg/pipeline/types/CiCdConfig.go
@@ -557,6 +557,7 @@ type GitTriggerInfoResponse struct {
 	Default          bool                                        `json:"default,omitempty"`
 	ImageTaggingData ImageTaggingResponseDTO                     `json:"imageTaggingData"`
 	Image            string                                      `json:"image"`
+	TargetPlatforms  []bean.TargetPlatform                       `json:"targetPlatforms"`
 }
 
 type Trigger struct {
diff --git a/pkg/pipeline/types/Workflow.go b/pkg/pipeline/types/Workflow.go
@@ -660,6 +660,7 @@ type WorkflowResponse struct {
 	CustomTag            *bean3.CustomTagErrorResponse               `json:"customTag,omitempty"`
 	PipelineType         string                                      `json:"pipelineType"`
 	ReferenceWorkflowId  int                                         `json:"referenceWorkflowId"`
+	TargetPlatforms      []bean.TargetPlatform                       `json:"targetPlatforms"`
 }
 
 type ConfigMapSecretDto struct {
diff --git a/pkg/workflow/dag/WorkflowDagExecutor.go b/pkg/workflow/dag/WorkflowDagExecutor.go
@@ -869,6 +869,7 @@ func (impl *WorkflowDagExecutorImpl) HandleCiSuccessEvent(triggerContext trigger
 			IsArtifactUploaded: request.IsArtifactUploaded, // for backward compatibility
 			ScanEnabled:        buildArtifact.ScanEnabled,
 			Scanned:            false,
+			TargetPlatforms:    adaptor.GetTargetPlatformStringFromList(request.TargetPlatforms),
 			AuditLog:           sql.AuditLog{CreatedBy: request.UserId, UpdatedBy: request.UserId, CreatedOn: time.Now(), UpdatedOn: time.Now()},
 		}
 		if buildArtifact.ScanEnabled {
diff --git a/pkg/workflow/dag/adaptor/adaptor.go b/pkg/workflow/dag/adaptor/adaptor.go
@@ -4,6 +4,7 @@ import (
 	"github.com/devtron-labs/devtron/internal/sql/repository"
 	"github.com/devtron-labs/devtron/pkg/sql"
 	bean2 "github.com/devtron-labs/devtron/pkg/workflow/dag/bean"
+	"strings"
 	"time"
 )
 
@@ -18,6 +19,11 @@ func GetBuildArtifact(request *bean2.CiArtifactWebhookRequest, ciPipelineId int,
 		ScanEnabled:        request.IsScanEnabled,
 		IsArtifactUploaded: request.IsArtifactUploaded, // for backward compatibility
 		Scanned:            false,
+		TargetPlatforms:    GetTargetPlatformStringFromList(request.TargetPlatforms),
 		AuditLog:           sql.AuditLog{CreatedBy: request.UserId, UpdatedBy: request.UserId, CreatedOn: createdOn, UpdatedOn: updatedOn},
 	}
 }
+
+func GetTargetPlatformStringFromList(targetPlatforms []string) string {
+	return strings.Join(targetPlatforms, ",")
+}
diff --git a/pkg/workflow/dag/bean/bean.go b/pkg/workflow/dag/bean/bean.go
@@ -35,6 +35,7 @@ type CiArtifactWebhookRequest struct {
 	PluginRegistryArtifactDetails map[string][]string            `json:"PluginRegistryArtifactDetails"` //map of registry and array of images generated by Copy container image plugin
 	PluginArtifactStage           string                         `json:"pluginArtifactStage"`           // at which stage of CI artifact was generated by plugin ("pre_ci/post_ci")
 	IsScanEnabled                 bool                           `json:"isScanEnabled"`
+	TargetPlatforms               []string                       `json:"targetPlatforms"`
 }
 
 const (
diff --git a/scripts/sql/31602700_artifact_target_platform.down.sql b/scripts/sql/31602700_artifact_target_platform.down.sql
@@ -0,0 +1 @@
+ALTER TABLE public.ci_artifact DROP COLUMN IF EXISTS target_platforms;
diff --git a/scripts/sql/31602700_artifact_target_platform.up.sql b/scripts/sql/31602700_artifact_target_platform.up.sql
@@ -0,0 +1 @@
+ALTER TABLE public.ci_artifact ADD COLUMN IF NOT EXISTS target_platforms varchar(1000) NULL;

Original file line number	Diff line number	Diff line change
`@@ -83,6 +83,7 @@ type CiCompleteEvent struct {`
`83`	`83`	PluginRegistryArtifactDetails map[string][]string `json:"PluginRegistryArtifactDetails"`
`84`	`84`	PluginArtifactStage string `json:"pluginArtifactStage"`
`85`	`85`	IsScanEnabled bool `json:"isScanEnabled"`
	`86`	+ TargetPlatforms []string `json:"targetPlatforms"`
`86`	`87`	`pluginImageDetails *registry.ImageDetailsFromCR`
`87`	`88`	PluginArtifacts *PluginArtifacts `json:"pluginArtifacts"`
`88`	`89`	`}`
Original file line number	Diff line number	Diff line change
`@@ -678,6 +678,7 @@ func (impl *WorkflowEventProcessorImpl) BuildCiArtifactRequest(event bean.CiComp`
`678`	`678`	`PluginRegistryArtifactDetails: pluginArtifacts,`
`679`	`679`	`PluginArtifactStage: event.PluginArtifactStage,`
`680`	`680`	`IsScanEnabled: event.IsScanEnabled,`
	`681`	`+ TargetPlatforms: event.TargetPlatforms,`
`681`	`682`	`}`
`682`	`683`	`// if DataSource is empty, repository.WEBHOOK is considered as default`
`683`	`684`	`if request.DataSource == "" {`