com.volcengine.service.vikingDB.common.SearchParam Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of volc-sdk-java Show documentation

The VOLC Engine SDK for Java

There is a newer version: 1.0.192

package com.volcengine.service.vikingDB.common;

import java.util.List;
import java.util.Map;

import com.volcengine.service.vikingDB.VikingDBException;

import lombok.Data;
@Data
public class SearchParam {
    private VectorOrder vectorOrder = null;
    private ScalarOrder scalarOrder = null;
    private Map filter = null;
    private Integer limit = 10;
    private List outputFields = null;
    private Object partition = "default";
    private Integer isBuild = 0;
    private Double denseWeight = null;
    public SearchParam setVectorOrder(VectorOrder vectorOrder){
        this.vectorOrder = vectorOrder;
        return this;
    }
    public SearchParam setScalarOrder(ScalarOrder scalarOrder){
        this.scalarOrder = scalarOrder;
        return this;
    }
    public SearchParam setFilter(Map filter){
        this.filter = filter;
        return this;
    }
    public SearchParam setLimit(Integer limit){
        this.limit = limit;
        return this;
    }
    public SearchParam setOutputFields(List outputFields){
        this.outputFields = outputFields;
        return this;
    }
    public SearchParam setPartition(Object partition){
        this.partition = partition;
        return this;
    }
    public SearchParam setDenseWeight(Double denseWeight){
        this.denseWeight = denseWeight;
        return this;
    }
    public SearchParam build() throws Exception{
        VikingDBException vikingDBException = new VikingDBException(1000030, null, "Params does not exist");
        if(this.scalarOrder != null && this.scalarOrder.getIsBuild() == 0){
            throw vikingDBException.getErrorCodeException(1000030, null, "scalarOrder does not build");
        }
        this.isBuild = 1;
        return this;
    }
}