Allow OperatorType to be a list such that one API can map to multiple operations

Author: mbaijal

pkg/generate/config/operation.go

@@ -37,7 +37,7 @@ type OperationConfig struct {
 	ResourceName string `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"`
+	OperationType []string `json:"operation_type"`
 	// PrimaryIdentifierFieldName provides the name of the field that should be
 	// interpreted as the "primary" identifier field. This field will be used as
 	// the primary field for resource adoption.

pkg/model/sdk_helper.go

@@ -206,11 +206,13 @@ 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 {
-		opType, resName := getOpTypeAndResourceName(opID, cfg)
-		if _, found := opMap[opType]; !found {
-			opMap[opType] = map[string]*awssdkmodel.Operation{}
-		}
-		opMap[opType][resName] = op
+		opTypeArray, resName := getOpTypeAndResourceName(opID, cfg)
+		for _, opType := range opTypeArray {
+			if _, found := opMap[opType]; !found {
+				opMap[opType] = map[string]*awssdkmodel.Operation{}
+			}
+			opMap[opType][resName] = op
+		}	
 	}
 	a.opMap = &opMap
 	return &opMap
@@ -382,20 +384,27 @@ func (a *SDKAPI) SDKAPIInterfaceTypeName() string {
 }
 
 // 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)
-
+	opTypeArray := make([]OpType, 0)
+	// TODO: or use var opTypeArray []OpType
 	if cfg != nil {
 		if operationConfig, exists := cfg.Operations[opID]; exists {
-			if operationConfig.OperationType != "" {
-				opType = OpTypeFromString(operationConfig.OperationType)
-			}
-
 			if operationConfig.ResourceName != "" {
 				resName = operationConfig.ResourceName
 			}
-		}
-	}
 
-	return opType, resName
+			if len(operationConfig.OperationType) > 0 {
+				for _, operationType := range operationConfig.OperationType {
+					opType = OpTypeFromString(operationType)
+					opTypeArray = append(opTypeArray, opType)
+				}	
+			} else {
+				opTypeArray = append(opTypeArray, opType)
+			}
+		} else {
+			opTypeArray = append(opTypeArray, opType)
+		}
+	} 
+	return opTypeArray, resName
 }