Skip to content

Commit a0573da

Browse files
CLOUDP-368428-Stream-Processor Constants and Aligning with Flex Cluster
1 parent 64d662a commit a0573da

2 files changed

Lines changed: 19 additions & 19 deletions

File tree

cfn-resources/stream-processor/cmd/resource/callbacks.go

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -147,15 +147,15 @@ func handleCreateCallback(ctx context.Context, client *util.MongoDBClient, curre
147147
if peErr := startStreamProcessor(ctx, client.AtlasSDK, callbackCtx.ProjectID, callbackCtx.WorkspaceOrInstanceName, callbackCtx.ProcessorName); peErr != nil {
148148
return *peErr
149149
}
150-
return createInProgressEvent("Starting stream processor", currentModel, callbackContext)
150+
return createInProgressEvent(constants.Pending, currentModel, callbackContext)
151151
}
152-
return finalizeModel(streamProcessor, currentModel, "Create Completed")
152+
return finalizeModel(streamProcessor, currentModel, constants.Complete)
153153

154154
case StartedState:
155-
return finalizeModel(streamProcessor, currentModel, "Create Completed")
155+
return finalizeModel(streamProcessor, currentModel, constants.Complete)
156156

157157
case InitiatingState, CreatingState:
158-
return createInProgressEvent(fmt.Sprintf("Creating stream processor (current state: %s)", currentState), currentModel, callbackContext)
158+
return createInProgressEvent(constants.Pending, currentModel, callbackContext)
159159

160160
case FailedState:
161161
return handler.ProgressEvent{
@@ -214,14 +214,14 @@ func handleUpdateCallback(ctx context.Context, client *util.MongoDBClient, curre
214214
if peErr := startStreamProcessor(ctx, client.AtlasSDK, callbackCtx.ProjectID, callbackCtx.WorkspaceOrInstanceName, callbackCtx.ProcessorName); peErr != nil {
215215
return *peErr
216216
}
217-
return createInProgressEvent("Starting stream processor", currentModel, callbackContext)
217+
return createInProgressEvent(constants.Pending, currentModel, callbackContext)
218218
}
219219

220-
return finalizeModel(streamProcessorResp, currentModel, "Update Completed")
220+
return finalizeModel(streamProcessorResp, currentModel, constants.Complete)
221221

222222
case StartedState:
223223
if desiredState == StartedState {
224-
return finalizeModel(streamProcessor, currentModel, "Update Completed")
224+
return finalizeModel(streamProcessor, currentModel, constants.Complete)
225225
}
226226

227227
_, err := client.AtlasSDK.StreamsApi.StopStreamProcessorWithParams(ctx,
@@ -237,7 +237,7 @@ func handleUpdateCallback(ctx context.Context, client *util.MongoDBClient, curre
237237
Message: fmt.Sprintf("Error stopping stream processor: %s", err.Error()),
238238
}
239239
}
240-
return createInProgressEvent("Stopping stream processor", currentModel, callbackContext)
240+
return createInProgressEvent(constants.Pending, currentModel, callbackContext)
241241

242242
case FailedState:
243243
return handler.ProgressEvent{
@@ -246,6 +246,6 @@ func handleUpdateCallback(ctx context.Context, client *util.MongoDBClient, curre
246246
}
247247

248248
default:
249-
return createInProgressEvent(fmt.Sprintf("Updating stream processor (current state: %s)", currentState), currentModel, callbackContext)
249+
return createInProgressEvent(constants.Pending, currentModel, callbackContext)
250250
}
251251
}

cfn-resources/stream-processor/cmd/resource/share.go

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -28,13 +28,13 @@ import (
2828
"github.com/mongodb/mongodbatlas-cloudformation-resources/util/constants"
2929
)
3030

31-
func isCallback(req *handler.Request) bool {
31+
func IsCallback(req *handler.Request) bool {
3232
_, found := req.CallbackContext["callbackStreamProcessor"]
3333
return found
3434
}
3535

3636
func HandleCreate(req *handler.Request, client *util.MongoDBClient, model *Model) handler.ProgressEvent {
37-
if isCallback(req) {
37+
if IsCallback(req) {
3838
callbackCtx := getCallbackData(*req)
3939
if peErr := validateCallbackData(callbackCtx); peErr != nil {
4040
return *peErr
@@ -107,7 +107,7 @@ func HandleCreate(req *handler.Request, client *util.MongoDBClient, model *Model
107107

108108
return handler.ProgressEvent{
109109
OperationStatus: handler.InProgress,
110-
Message: "Creating stream processor",
110+
Message: constants.Pending,
111111
ResourceModel: inProgressModel,
112112
CallbackDelaySeconds: defaultCallbackDelaySeconds,
113113
CallbackContext: buildCallbackContext(projectID, workspaceOrInstanceName, processorName, map[string]any{
@@ -160,13 +160,13 @@ func HandleRead(req *handler.Request, client *util.MongoDBClient, model *Model)
160160

161161
return handler.ProgressEvent{
162162
OperationStatus: handler.Success,
163-
Message: "Read Completed",
163+
Message: constants.ReadComplete,
164164
ResourceModel: resourceModel,
165165
}
166166
}
167167

168168
func HandleUpdate(req *handler.Request, client *util.MongoDBClient, prevModel *Model, model *Model) handler.ProgressEvent {
169-
if isCallback(req) {
169+
if IsCallback(req) {
170170
callbackCtx := getCallbackData(*req)
171171
if peErr := validateCallbackData(callbackCtx); peErr != nil {
172172
return *peErr
@@ -249,7 +249,7 @@ func HandleUpdate(req *handler.Request, client *util.MongoDBClient, prevModel *M
249249

250250
return handler.ProgressEvent{
251251
OperationStatus: handler.InProgress,
252-
Message: "Stopping stream processor",
252+
Message: constants.Pending,
253253
ResourceModel: inProgressModel,
254254
CallbackDelaySeconds: defaultCallbackDelaySeconds,
255255
CallbackContext: buildCallbackContext(projectID, workspaceOrInstanceName, processorName, map[string]any{
@@ -295,7 +295,7 @@ func HandleUpdate(req *handler.Request, client *util.MongoDBClient, prevModel *M
295295

296296
return handler.ProgressEvent{
297297
OperationStatus: handler.InProgress,
298-
Message: "Starting stream processor",
298+
Message: constants.Pending,
299299
ResourceModel: inProgressModel,
300300
CallbackDelaySeconds: defaultCallbackDelaySeconds,
301301
CallbackContext: buildCallbackContext(projectID, workspaceOrInstanceName, processorName, map[string]any{
@@ -304,7 +304,7 @@ func HandleUpdate(req *handler.Request, client *util.MongoDBClient, prevModel *M
304304
}
305305
}
306306

307-
return finalizeModel(streamProcessorResp, model, "Update Completed")
307+
return finalizeModel(streamProcessorResp, model, constants.Complete)
308308
}
309309

310310
func HandleDelete(req *handler.Request, client *util.MongoDBClient, model *Model) handler.ProgressEvent {
@@ -337,7 +337,7 @@ func HandleDelete(req *handler.Request, client *util.MongoDBClient, model *Model
337337

338338
return handler.ProgressEvent{
339339
OperationStatus: handler.Success,
340-
Message: "Delete Completed",
340+
Message: constants.Complete,
341341
}
342342
}
343343

@@ -374,7 +374,7 @@ func HandleList(req *handler.Request, client *util.MongoDBClient, model *Model)
374374

375375
return handler.ProgressEvent{
376376
OperationStatus: handler.Success,
377-
Message: "List Completed",
377+
Message: constants.Complete,
378378
ResourceModels: response,
379379
}
380380
}

0 commit comments

Comments
 (0)