seecoder-devcloud/api/src/main/java/seecoder/devcloud/api/k8s/model/CpuQuantity.java

package seecoder.devcloud.api.k8s.model;

import io.kubernetes.client.custom.Quantity;
import lombok.Getter;

/**
 * CPU配额数量，默认单位为m，即CPU时间
 */
@Getter
public class CpuQuantity {

    private final Quantity quantity;

    private final Unit unit;

    private static final Unit DEFAULT_UNIT = Unit.Mi;
    //Cpu=0
    public static final CpuQuantity EMPTY = new CpuQuantity(0);
    //Cpu的默认分配额度
    public static final CpuQuantity DEFAULT = new CpuQuantity(50);

    //Cpu单位
    public enum Unit{
        Mi("m");

        private String code;

        Unit(String code) {
            this.code = code;
        }

        public String getCode(){
            return this.code;
        }
    }



    public CpuQuantity(int quantity){
        this.quantity = Quantity.fromString(quantity + DEFAULT_UNIT.getCode());
        this.unit = DEFAULT_UNIT;
    }

    public CpuQuantity(long quantity) {
        this.quantity = Quantity.fromString(quantity + DEFAULT_UNIT.getCode());
        this.unit = DEFAULT_UNIT;
    }


    public CpuQuantity(Quantity quantity) {
        this.quantity = quantity;
        this.unit = DEFAULT_UNIT;
    }

    public Quantity toQuantity(){
        return quantity;
    }

    public String getQuantity() {
        return quantity.toSuffixedString();
    }

    public static CpuQuantity getCpuQuantity(Quantity quantity) {
        if (quantity == null) {
            return null;
        } else {
            return new CpuQuantity(quantity);
        }
    }


}