| /* |
| * Licensed to the Apache Software Foundation (ASF) under one |
| * or more contributor license agreements. See the NOTICE file |
| * distributed with this work for additional information |
| * regarding copyright ownership. The ASF licenses this file |
| * to you under the Apache License, Version 2.0 (the |
| * "License"); you may not use this file except in compliance |
| * with the License. You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, |
| * software distributed under the License is distributed on an |
| * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY |
| * KIND, either express or implied. See the License for the |
| * specific language governing permissions and limitations |
| * under the License. |
| */ |
| |
| package com.epam.datalab.dto.exploratory; |
| |
| import com.epam.datalab.dto.aws.computational.ClusterConfig; |
| import com.fasterxml.jackson.annotation.JsonProperty; |
| import com.google.common.base.MoreObjects.ToStringHelper; |
| import lombok.Getter; |
| import lombok.Setter; |
| |
| import java.util.List; |
| import java.util.Map; |
| |
| @Setter |
| @Getter |
| public class ExploratoryCreateDTO<T extends ExploratoryCreateDTO<?>> extends ExploratoryBaseDTO<T> { |
| |
| |
| @JsonProperty("git_creds") |
| private List<ExploratoryGitCreds> gitCreds; |
| @JsonProperty("notebook_image_name") |
| private String imageName; |
| @JsonProperty("spark_configurations") |
| private List<ClusterConfig> clusterConfig; |
| @JsonProperty("tags") |
| private Map<String, String> tags; |
| @JsonProperty("endpoint_name") |
| private String endpoint; |
| @JsonProperty("conf_shared_image_enabled") |
| private String sharedImageEnabled; |
| @JsonProperty("gpu_enabled") |
| private Boolean enabledGPU; |
| @JsonProperty("gpuType") |
| private String gpuType; |
| @JsonProperty("gpuCount") |
| private String gpuCount; |
| |
| /** |
| * Return the list of GIT credentials. |
| */ |
| public List<ExploratoryGitCreds> getGitCreds() { |
| return gitCreds; |
| } |
| |
| /** |
| * Set the list of GIT credentials. |
| */ |
| public void setGitCreds(List<ExploratoryGitCreds> gitCreds) { |
| this.gitCreds = gitCreds; |
| } |
| |
| /** |
| * Set the list of GIT credentials and return this object. |
| */ |
| @SuppressWarnings("unchecked") |
| public T withGitCreds(List<ExploratoryGitCreds> gitCreds) { |
| setGitCreds(gitCreds); |
| return (T) this; |
| } |
| |
| /** |
| * Set the image name and return this object. |
| */ |
| @SuppressWarnings("unchecked") |
| public T withImageName(String imageName) { |
| setImageName(imageName); |
| return (T) this; |
| } |
| |
| @SuppressWarnings("unchecked") |
| public T withTags(Map<String, String> tags) { |
| this.tags = tags; |
| return (T) this; |
| } |
| |
| @SuppressWarnings("unchecked") |
| |
| @Override |
| public T withEndpoint(String endpoint) { |
| this.endpoint = endpoint; |
| return (T) this; |
| } |
| |
| @SuppressWarnings("unchecked") |
| |
| public T withSharedImageEnabled(String sharedImageEnabled) { |
| this.sharedImageEnabled = sharedImageEnabled; |
| return (T) this; |
| } |
| |
| public String getImageName() { |
| return imageName; |
| } |
| |
| public void setImageName(String imageName) { |
| this.imageName = imageName; |
| } |
| |
| @SuppressWarnings("unchecked") |
| |
| public T withClusterConfig(List<ClusterConfig> config) { |
| this.clusterConfig = config; |
| return (T) this; |
| } |
| |
| @SuppressWarnings("unchecked") |
| |
| public T withEnabledGPU(Boolean enabledGPU) { |
| setEnabledGPU(enabledGPU); |
| return (T) this; |
| } |
| |
| @SuppressWarnings("unchecked") |
| public T withGPUCount(String gpuCount) { |
| setGpuCount(gpuCount); |
| return (T) this; |
| } |
| |
| @SuppressWarnings("unchecked") |
| public T withGPUType(String gpuType) { |
| setGpuType(gpuType); |
| return (T) this; |
| } |
| |
| @Override |
| public ToStringHelper toStringHelper(Object self) { |
| return super.toStringHelper(self) |
| .add("gitCreds", gitCreds) |
| .add("imageName", imageName); |
| } |
| } |