SEEC-II-Demo/src/main/java/nju/seec/SEECdemo/web/dto/ContainerDTO.java

package nju.seec.SEECdemo.web.dto;

import lombok.Data;
import nju.seec.SEECdemo.logic.api.k8s.model.Container;

import javax.validation.constraints.NotEmpty;
import java.util.HashSet;
import java.util.Set;
import java.util.stream.Collectors;

@Data
public class ContainerDTO {

    @NotEmpty
    private String name;

    @NotEmpty
    private String image;

    private Set<PortDTO> ports = new HashSet<>();

    public Container toContainerDTO() {
        Container container = new Container();
        container.setName(name);
        container.setImage(image);
        container.setPorts(ports.stream().map(PortDTO::toContainerPortDTO).collect(Collectors.toSet()));
        return container;
    }

}