package com.ai.da.model.vo;

import io.swagger.annotations.ApiModel;
import io.swagger.annotations.ApiModelProperty;
import lombok.Data;

import java.util.List;

@Data
@ApiModel("generate响应vo")
public class GenerateCollectionVO {

    @ApiModelProperty("generateId")
    private Long generateId;

    @ApiModelProperty("collection")
    private Long collectionId;

    @ApiModelProperty("生成的图片信息")
    private List<GenerateCollectionItemVO> generatedCollectionItems;

    @ApiModelProperty("在当前队列中的排序")
    private Long rankPosition;

    public GenerateCollectionVO(Long rankPosition) {
        this.rankPosition = rankPosition;
    }

    public GenerateCollectionVO(Long generateId, Long collectionId, List<GenerateCollectionItemVO> generatedCollectionItems) {
        this.generateId = generateId;
        this.collectionId = collectionId;
        this.generatedCollectionItems = generatedCollectionItems;
    }

    public GenerateCollectionVO() {
    }
}
