LookoutVisionInferenceClient::OperationStatus LookoutVisionInferenceClient::StartModel()

in src/lookoutvision-client/LookoutVisionInferenceClient.cc [113:139]

• 23 lines of code
• 4 McCabe index (conditional complexity)

LookoutVisionInferenceClient::OperationStatus LookoutVisionInferenceClient::StartModel(std::string model_component,
                                                                                       int model_status_timeout) {
    AWS::LookoutVision::StartModelRequest request;
    AWS::LookoutVision::StartModelResponse reply;
    grpc::ClientContext context;

    try {
        request.set_model_component(model_component);
        grpc::Status status = stub->StartModel(&context, request, &reply);

        if (!status.ok()) {
            std::cout << "StartModel returned error "
            << status.error_code() << ": " << status.error_message() << std::endl;
        }
    } catch (std::exception& e) {
        std::cout << "Exception: " << e.what() << std::endl;
    }

    if (waitForModelStatusWithTimeout(model_component, AWS::LookoutVision::ModelStatus::RUNNING,
                                      model_status_timeout)) {
        return OperationStatus::SUCCESSFUL;
    } else {
        std::cout << "Model didn't reach RUNNING state within " << model_status_timeout << " seconds" << std::endl;
    }

    return OperationStatus::FAILED;
}