commonMain.aws.sdk.kotlin.services.neptunedata.serde.StartMLModelTrainingJobOperationSerializer.kt Maven / Gradle / Ivy

Go to download
// Code generated by smithy-kotlin-codegen. DO NOT EDIT!

package aws.sdk.kotlin.services.neptunedata.serde

import aws.sdk.kotlin.services.neptunedata.model.CustomModelTrainingParameters
import aws.sdk.kotlin.services.neptunedata.model.StartMlModelTrainingJobRequest
import aws.smithy.kotlin.runtime.http.HttpBody
import aws.smithy.kotlin.runtime.http.HttpMethod
import aws.smithy.kotlin.runtime.http.operation.HttpSerializer
import aws.smithy.kotlin.runtime.http.request.HttpRequestBuilder
import aws.smithy.kotlin.runtime.http.request.url
import aws.smithy.kotlin.runtime.operation.ExecutionContext
import aws.smithy.kotlin.runtime.serde.SdkFieldDescriptor
import aws.smithy.kotlin.runtime.serde.SdkObjectDescriptor
import aws.smithy.kotlin.runtime.serde.SerialKind
import aws.smithy.kotlin.runtime.serde.asSdkSerializable
import aws.smithy.kotlin.runtime.serde.deserializeList
import aws.smithy.kotlin.runtime.serde.deserializeMap
import aws.smithy.kotlin.runtime.serde.deserializeStruct
import aws.smithy.kotlin.runtime.serde.field
import aws.smithy.kotlin.runtime.serde.json.JsonDeserializer
import aws.smithy.kotlin.runtime.serde.json.JsonSerialName
import aws.smithy.kotlin.runtime.serde.json.JsonSerializer
import aws.smithy.kotlin.runtime.serde.serializeList
import aws.smithy.kotlin.runtime.serde.serializeMap
import aws.smithy.kotlin.runtime.serde.serializeStruct


internal class StartMLModelTrainingJobOperationSerializer: HttpSerializer.NonStreaming {
    override fun serialize(context: ExecutionContext, input: StartMlModelTrainingJobRequest): HttpRequestBuilder {
        val builder = HttpRequestBuilder()
        builder.method = HttpMethod.POST

        builder.url {
            path.encoded = "/ml/modeltraining"
        }

        val payload = serializeStartMLModelTrainingJobOperationBody(context, input)
        builder.body = HttpBody.fromBytes(payload)
        if (builder.body !is HttpBody.Empty) {
            builder.headers.setMissing("Content-Type", "application/json")
        }
        return builder
    }
}

private fun serializeStartMLModelTrainingJobOperationBody(context: ExecutionContext, input: StartMlModelTrainingJobRequest): ByteArray {
    val serializer = JsonSerializer()
    val BASEPROCESSINGINSTANCETYPE_DESCRIPTOR = SdkFieldDescriptor(SerialKind.String, JsonSerialName("baseProcessingInstanceType"))
    val CUSTOMMODELTRAININGPARAMETERS_DESCRIPTOR = SdkFieldDescriptor(SerialKind.Struct, JsonSerialName("customModelTrainingParameters"))
    val DATAPROCESSINGJOBID_DESCRIPTOR = SdkFieldDescriptor(SerialKind.String, JsonSerialName("dataProcessingJobId"))
    val ENABLEMANAGEDSPOTTRAINING_DESCRIPTOR = SdkFieldDescriptor(SerialKind.Boolean, JsonSerialName("enableManagedSpotTraining"))
    val ID_DESCRIPTOR = SdkFieldDescriptor(SerialKind.String, JsonSerialName("id"))
    val MAXHPONUMBEROFTRAININGJOBS_DESCRIPTOR = SdkFieldDescriptor(SerialKind.Integer, JsonSerialName("maxHPONumberOfTrainingJobs"))
    val MAXHPOPARALLELTRAININGJOBS_DESCRIPTOR = SdkFieldDescriptor(SerialKind.Integer, JsonSerialName("maxHPOParallelTrainingJobs"))
    val NEPTUNEIAMROLEARN_DESCRIPTOR = SdkFieldDescriptor(SerialKind.String, JsonSerialName("neptuneIamRoleArn"))
    val PREVIOUSMODELTRAININGJOBID_DESCRIPTOR = SdkFieldDescriptor(SerialKind.String, JsonSerialName("previousModelTrainingJobId"))
    val S3OUTPUTENCRYPTIONKMSKEY_DESCRIPTOR = SdkFieldDescriptor(SerialKind.String, JsonSerialName("s3OutputEncryptionKMSKey"))
    val SAGEMAKERIAMROLEARN_DESCRIPTOR = SdkFieldDescriptor(SerialKind.String, JsonSerialName("sagemakerIamRoleArn"))
    val SECURITYGROUPIDS_DESCRIPTOR = SdkFieldDescriptor(SerialKind.List, JsonSerialName("securityGroupIds"))
    val SUBNETS_DESCRIPTOR = SdkFieldDescriptor(SerialKind.List, JsonSerialName("subnets"))
    val TRAINMODELS3LOCATION_DESCRIPTOR = SdkFieldDescriptor(SerialKind.String, JsonSerialName("trainModelS3Location"))
    val TRAININGINSTANCETYPE_DESCRIPTOR = SdkFieldDescriptor(SerialKind.String, JsonSerialName("trainingInstanceType"))
    val TRAININGINSTANCEVOLUMESIZEINGB_DESCRIPTOR = SdkFieldDescriptor(SerialKind.Integer, JsonSerialName("trainingInstanceVolumeSizeInGB"))
    val TRAININGTIMEOUTINSECONDS_DESCRIPTOR = SdkFieldDescriptor(SerialKind.Integer, JsonSerialName("trainingTimeOutInSeconds"))
    val VOLUMEENCRYPTIONKMSKEY_DESCRIPTOR = SdkFieldDescriptor(SerialKind.String, JsonSerialName("volumeEncryptionKMSKey"))
    val OBJ_DESCRIPTOR = SdkObjectDescriptor.build {
        field(BASEPROCESSINGINSTANCETYPE_DESCRIPTOR)
        field(CUSTOMMODELTRAININGPARAMETERS_DESCRIPTOR)
        field(DATAPROCESSINGJOBID_DESCRIPTOR)
        field(ENABLEMANAGEDSPOTTRAINING_DESCRIPTOR)
        field(ID_DESCRIPTOR)
        field(MAXHPONUMBEROFTRAININGJOBS_DESCRIPTOR)
        field(MAXHPOPARALLELTRAININGJOBS_DESCRIPTOR)
        field(NEPTUNEIAMROLEARN_DESCRIPTOR)
        field(PREVIOUSMODELTRAININGJOBID_DESCRIPTOR)
        field(S3OUTPUTENCRYPTIONKMSKEY_DESCRIPTOR)
        field(SAGEMAKERIAMROLEARN_DESCRIPTOR)
        field(SECURITYGROUPIDS_DESCRIPTOR)
        field(SUBNETS_DESCRIPTOR)
        field(TRAINMODELS3LOCATION_DESCRIPTOR)
        field(TRAININGINSTANCETYPE_DESCRIPTOR)
        field(TRAININGINSTANCEVOLUMESIZEINGB_DESCRIPTOR)
        field(TRAININGTIMEOUTINSECONDS_DESCRIPTOR)
        field(VOLUMEENCRYPTIONKMSKEY_DESCRIPTOR)
    }

    serializer.serializeStruct(OBJ_DESCRIPTOR) {
        input.baseProcessingInstanceType?.let { field(BASEPROCESSINGINSTANCETYPE_DESCRIPTOR, it) }
        input.customModelTrainingParameters?.let { field(CUSTOMMODELTRAININGPARAMETERS_DESCRIPTOR, it, ::serializeCustomModelTrainingParametersDocument) }
        input.dataProcessingJobId?.let { field(DATAPROCESSINGJOBID_DESCRIPTOR, it) }
        input.enableManagedSpotTraining?.let { field(ENABLEMANAGEDSPOTTRAINING_DESCRIPTOR, it) }
        input.id?.let { field(ID_DESCRIPTOR, it) }
        input.maxHpoNumberOfTrainingJobs?.let { field(MAXHPONUMBEROFTRAININGJOBS_DESCRIPTOR, it) }
        input.maxHpoParallelTrainingJobs?.let { field(MAXHPOPARALLELTRAININGJOBS_DESCRIPTOR, it) }
        input.neptuneIamRoleArn?.let { field(NEPTUNEIAMROLEARN_DESCRIPTOR, it) }
        input.previousModelTrainingJobId?.let { field(PREVIOUSMODELTRAININGJOBID_DESCRIPTOR, it) }
        input.s3OutputEncryptionKmsKey?.let { field(S3OUTPUTENCRYPTIONKMSKEY_DESCRIPTOR, it) }
        input.sagemakerIamRoleArn?.let { field(SAGEMAKERIAMROLEARN_DESCRIPTOR, it) }
        if (input.securityGroupIds != null) {
            listField(SECURITYGROUPIDS_DESCRIPTOR) {
                for (el0 in input.securityGroupIds) {
                    serializeString(el0)
                }
            }
        }
        if (input.subnets != null) {
            listField(SUBNETS_DESCRIPTOR) {
                for (el0 in input.subnets) {
                    serializeString(el0)
                }
            }
        }
        input.trainModelS3Location?.let { field(TRAINMODELS3LOCATION_DESCRIPTOR, it) }
        input.trainingInstanceType?.let { field(TRAININGINSTANCETYPE_DESCRIPTOR, it) }
        input.trainingInstanceVolumeSizeInGb?.let { field(TRAININGINSTANCEVOLUMESIZEINGB_DESCRIPTOR, it) }
        input.trainingTimeOutInSeconds?.let { field(TRAININGTIMEOUTINSECONDS_DESCRIPTOR, it) }
        input.volumeEncryptionKmsKey?.let { field(VOLUMEENCRYPTIONKMSKEY_DESCRIPTOR, it) }
    }
    return serializer.toByteArray()
}