v1

type BatchTransformJob struct {
	metav1.TypeMeta   `json:",inline"`
	metav1.ObjectMeta `json:"metadata,omitempty"`

	Spec   BatchTransformJobSpec   `json:"spec,omitempty"`
	Status BatchTransformJobStatus `json:"status,omitempty"`
}

type BatchTransformJobList struct {
	metav1.TypeMeta `json:",inline"`
	metav1.ListMeta `json:"metadata,omitempty"`
	Items           []BatchTransformJob `json:"items"`
}

type BatchTransformJobSpec struct {
	// The SageMaker batchtransform job name. This is optional for the SageMaker K8s operator. If it is empty,
	// the operator will populate it with a generated name.
	// +kubebuilder:validation:MaxLength=63
	TransformJobName *string `json:"transformJobName,omitempty"`

	BatchStrategy commonv1.BatchStrategy `json:"batchStrategy,omitempty"`

	DataProcessing *commonv1.DataProcessing `json:"dataProcessing,omitempty"`

	Environment []*commonv1.KeyValuePair `json:"environment,omitempty"`

	MaxConcurrentTransforms *int64 `json:"maxConcurrentTransforms,omitempty"`

	MaxPayloadInMB *int64 `json:"maxPayloadInMB,omitempty"`

	ModelName *string `json:"modelName"`

	Tags []commonv1.Tag `json:"tags,omitempty"`

	TransformInput *commonv1.TransformInput `json:"transformInput"`

	TransformOutput *commonv1.TransformOutput `json:"transformOutput"`

	TransformResources *commonv1.TransformResources `json:"transformResources"`

	// +kubebuilder:validation:MinLength=1
	Region *string `json:"region"`

	// A custom SageMaker endpoint to use when communicating with SageMaker.
	// +kubebuilder:validation:Pattern=^(https|http)://.*$
	SageMakerEndpoint *string `json:"sageMakerEndpoint,omitempty"`
}

type BatchTransformJobStatus struct {
	// The status of the transform job.
	// https://docs.aws.amazon.com/sagemaker/latest/dg/API_DescribeTransformJob.html
	TransformJobStatus string `json:"transformJobStatus,omitempty"`

	// Field to store additional information, for example if
	// we are unable to check the status we update this.
	Additional string `json:"additional,omitempty"`

	// The last time that we checked the status of the SageMaker job.
	LastCheckTime *metav1.Time `json:"lastCheckTime,omitempty"`

	//SageMaker TransformJobName job name
	SageMakerTransformJobName string `json:"sageMakerTransformJobName,omitempty"`
}