Adding functionality to assign one API to multiple resources

Author: Vandita2020

pkg/config/operation.go

@@ -40,7 +40,7 @@ type OperationConfig struct {
 	OutputWrapperFieldPath string `json:"output_wrapper_field_path,omitempty"`
 	// Override for resource name in case of heuristic failure
 	// An example of this is correcting stutter when the resource logic doesn't properly determine the resource name
-	ResourceName string `json:"resource_name"`
+	ResourceName StringArray `json:"resource_name"`
 	// Override for operation type in case of heuristic failure
 	// An example of this is `Put...` or `Register...` API operations not being correctly classified as `Create` op type
 	// OperationType []string `json:"operation_type"`
@@ -59,8 +59,8 @@ func (c *Config) OperationIsIgnored(operation *awssdkmodel.Operation) bool {
 	return util.InStrings(operation.ExportedName, c.Ignore.Operations)
 }
 
-// UnmarshalJSON parses input for a either a string or
-// or a list and returns a StringArray.
+// UnmarshalJSON parses input for either a string or
+// a list and returns a StringArray.
 func (a *StringArray) UnmarshalJSON(b []byte) error {
 	var multi []string
 	err := json.Unmarshal(b, &multi)

pkg/model/sdk_api.go

@@ -78,12 +78,14 @@ func (a *SDKAPI) GetOperationMap(cfg *ackgenconfig.Config) *OperationMap {
 	// create an index of Operations by operation types and resource name
 	opMap := OperationMap{}
 	for opID, op := range a.API.Operations {
-		opTypeArray, resName := getOpTypeAndResourceName(opID, cfg)
+		opTypeArray, resList := getOpTypeAndResourceName(opID, cfg)
 		for _, opType := range opTypeArray {
 			if _, found := opMap[opType]; !found {
 				opMap[opType] = map[string]*awssdkmodel.Operation{}
 			}
-			opMap[opType][resName] = op
+			for _, resName := range resList {
+				opMap[opType][resName] = op
+			}
 		}
 	}
 
@@ -98,16 +100,18 @@ func (a *SDKAPI) GetOperationMap(cfg *ackgenconfig.Config) *OperationMap {
 	//
 	// see: https://github.com/aws-controllers-k8s/community/issues/1555
 	for opID, opCfg := range cfg.Operations {
-		if opCfg.ResourceName == "" {
+		if opCfg.ResourceName == nil {
 			continue
 		}
 		op, found := a.API.Operations[opID]
 		if !found {
 			panic("operation " + opID + " in generator.yaml 'operations:' object does not exist.")
 		}
 		for _, operationType := range opCfg.OperationType {
-			opType := OpTypeFromString(operationType)
-			opMap[opType][opCfg.ResourceName] = op
+			for _, resName := range opCfg.ResourceName {
+				opType := OpTypeFromString(operationType)
+				opMap[opType][resName] = op
+			}
 		}
 	}
 	a.opMap = &opMap
@@ -360,20 +364,19 @@ func NewSDKAPI(api *awssdkmodel.API, apiGroupSuffix string) *SDKAPI {
 }
 
 // Override the operation type and/or resource name, if specified in config
-func getOpTypeAndResourceName(opID string, cfg *ackgenconfig.Config) ([]OpType, string) {
+func getOpTypeAndResourceName(opID string, cfg *ackgenconfig.Config) ([]OpType, []string) {
 	opType, resName := GetOpTypeAndResourceNameFromOpID(opID, cfg)
 	opTypes := []OpType{opType}
+	resList := []string{resName}
 
 	if operationConfig, exists := cfg.GetOperationConfig(opID); exists {
-		if operationConfig.ResourceName != "" {
-			resName = operationConfig.ResourceName
-		}
+		resList = operationConfig.ResourceName
 		for _, operationType := range operationConfig.OperationType {
 			opType = OpTypeFromString(operationType)
 			opTypes = append(opTypes, opType)
 		}
 	}
-	return opTypes, resName
+	return opTypes, resList
 }
 
 // getMemberByPath returns a ShapeRef given a root Shape and a dot-notation