package com.newyetai.usercenter.district.dto;

import com.google.common.collect.Lists;
import com.newyetai.common.dto.BaseDTO;
import com.newyetai.common.util.TreeNode;
import io.swagger.annotations.ApiModel;
import io.swagger.annotations.ApiModelProperty;
import lombok.*;
import lombok.experimental.Accessors;
import lombok.experimental.SuperBuilder;

import java.util.List;
import java.util.Objects;

/**
 * <p>
 * 省份城市
 * </p>
 *
 * @author Carlye
 * @since 2021-01-25
 */
@ApiModel("省份城市树形DTO")
@Data
@SuperBuilder
@NoArgsConstructor
@AllArgsConstructor
@EqualsAndHashCode(callSuper = true)
@Accessors(chain = true)
public class DistrictDTO extends BaseDTO implements TreeNode<Long, DistrictDTO> {

    private static final long serialVersionUID = -3042079366109391854L;

    @ApiModelProperty("城市名称")
    private String name;

    @ApiModelProperty("上级城市ID")
    private Long parentId;

    @ApiModelProperty("首字母")
    private String initial;

    @ApiModelProperty("简拼")
    private String initials;

    @ApiModelProperty("全拼")
    private String pinyin;

    @ApiModelProperty("特别行政区")
    private String extra;

    @ApiModelProperty("后缀")
    private String suffix;

    @ApiModelProperty("城市编码")
    private String code;

    @ApiModelProperty("区域编码")
    private String areaCode;

    @ApiModelProperty("排序")
    private Integer districtOrder;

    private Integer level;

    @ApiModelProperty("下级")
    private List<DistrictDTO> childrenSet;

    public List<DistrictDTO> getChildrenSet() {
        if (Objects.isNull(this.childrenSet)) {
            this.childrenSet = Lists.newArrayList();
        }
        return this.childrenSet;
    }

    @Override
    public void addChildren(DistrictDTO node) {
        getChildrenSet().add(node);
    }
}