@@ -71,6 +71,8 @@ func (rm *resourceManager) syncEventInvokeConfig(
71
71
return r , nil
72
72
}
73
73
74
+ // updateProvisionedConcurrency calls `PutProvisionedConcurrencyConfig` to update the fields
75
+ // or `DeleteProvisionedConcurrencyConfig` if users removes the fields
74
76
func (rm * resourceManager ) updateProvisionedConcurrency (
75
77
ctx context.Context ,
76
78
desired * resource ,
@@ -80,20 +82,29 @@ func (rm *resourceManager) updateProvisionedConcurrency(
80
82
exit := rlog .Trace ("rm.updateProvisionedConcurrency" )
81
83
defer exit (err )
82
84
85
+ // Check if the user deleted the 'ProvisionedConcurrency' configuration
86
+ // If yes, delete ProvisionedConcurrencyConfig
87
+ if desired .ko .Spec .ProvisionedConcurrencyConfig == nil {
88
+ input_delete := & svcsdk.DeleteProvisionedConcurrencyConfigInput {
89
+ FunctionName : aws .String (* desired .ko .Spec .FunctionName ),
90
+ Qualifier : aws .String (* desired .ko .Spec .Name ),
91
+ }
92
+ _ , err = rm .sdkapi .DeleteProvisionedConcurrencyConfigWithContext (ctx , input_delete )
93
+ rm .metrics .RecordAPICall ("DELETE" , "DeleteProvisionedConcurrency" , err )
94
+ if err != nil {
95
+ return err
96
+ }
97
+ return nil
98
+ }
99
+
83
100
dspec := desired .ko .Spec
84
101
input := & svcsdk.PutProvisionedConcurrencyConfigInput {
85
102
FunctionName : aws .String (* dspec .FunctionName ),
86
103
Qualifier : aws .String (* dspec .Name ),
87
104
}
88
105
89
- if desired .ko .Spec .ProvisionedConcurrencyConfig != nil {
90
- if desired .ko .Spec .ProvisionedConcurrencyConfig .ProvisionedConcurrentExecutions != nil {
91
- input .ProvisionedConcurrentExecutions = aws .Int64 (* desired .ko .Spec .ProvisionedConcurrencyConfig .ProvisionedConcurrentExecutions )
92
- } else {
93
- input .ProvisionedConcurrentExecutions = aws .Int64 (0 )
94
- }
95
- } else {
96
- input .ProvisionedConcurrentExecutions = aws .Int64 (0 )
106
+ if dspec .ProvisionedConcurrencyConfig .ProvisionedConcurrentExecutions != nil {
107
+ input .ProvisionedConcurrentExecutions = aws .Int64 (* dspec .ProvisionedConcurrencyConfig .ProvisionedConcurrentExecutions )
97
108
}
98
109
99
110
_ , err = rm .sdkapi .PutProvisionedConcurrencyConfigWithContext (ctx , input )
@@ -104,7 +115,9 @@ func (rm *resourceManager) updateProvisionedConcurrency(
104
115
return nil
105
116
}
106
117
107
- func (rm * resourceManager ) getProvisionedConcurrencyConfig (
118
+ // setProvisionedConcurrencyConfig sets the Provisioned Concurrency
119
+ // for the Function's Alias
120
+ func (rm * resourceManager ) setProvisionedConcurrencyConfig (
108
121
ctx context.Context ,
109
122
ko * svcapitypes.Alias ,
110
123
) (err error ) {
@@ -126,12 +139,17 @@ func (rm *resourceManager) getProvisionedConcurrencyConfig(
126
139
return err
127
140
}
128
141
} else {
129
- ko .Spec .ProvisionedConcurrencyConfig .ProvisionedConcurrentExecutions = getProvisionedConcurrencyConfigOutput .RequestedProvisionedConcurrentExecutions
142
+ // creating ProvisionedConcurrency object to store the values returned from `Get` call
143
+ cloudProvisionedConcurrency := & svcapitypes.PutProvisionedConcurrencyConfigInput {}
144
+ cloudProvisionedConcurrency .ProvisionedConcurrentExecutions = getProvisionedConcurrencyConfigOutput .RequestedProvisionedConcurrentExecutions
145
+ ko .Spec .ProvisionedConcurrencyConfig = cloudProvisionedConcurrency
130
146
}
131
147
132
148
return nil
133
149
}
134
150
151
+ // getFunctionEventInvokeConfig will describe the fields that are
152
+ // custom to the Alias resource
135
153
func (rm * resourceManager ) getFunctionEventInvokeConfig (
136
154
ctx context.Context ,
137
155
ko * svcapitypes.Alias ,
@@ -183,6 +201,8 @@ func (rm *resourceManager) getFunctionEventInvokeConfig(
183
201
return nil
184
202
}
185
203
204
+ // setResourceAdditionalFields will describe the fields that are not return by the
205
+ // getFunctionConfiguration API call
186
206
func (rm * resourceManager ) setResourceAdditionalFields (
187
207
ctx context.Context ,
188
208
ko * svcapitypes.Alias ,
@@ -191,14 +211,16 @@ func (rm *resourceManager) setResourceAdditionalFields(
191
211
exit := rlog .Trace ("rm.setResourceAdditionalFields" )
192
212
defer exit (err )
193
213
214
+ // To set Asynchronous Invocations for the function's alias
194
215
eic_err := rm .getFunctionEventInvokeConfig (ctx , ko )
195
216
if eic_err != nil {
196
217
return eic_err
197
218
}
198
219
199
- pc_err := rm .getProvisionedConcurrencyConfig (ctx , ko )
200
- if pc_err != nil {
201
- return pc_err
220
+ // To set Provisioned Concurrency for the function's alias
221
+ err = rm .setProvisionedConcurrencyConfig (ctx , ko )
222
+ if err != nil {
223
+ return err
202
224
}
203
225
204
226
return nil
0 commit comments