com.tencent.polaris.api.rpc.InstancesResponse Maven / Gradle / Ivy

Go to download
/*
 * Tencent is pleased to support the open source community by making Polaris available.
 *
 * Copyright (C) 2019 THL A29 Limited, a Tencent company. All rights reserved.
 *
 * Licensed under the BSD 3-Clause License (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * https://opensource.org/licenses/BSD-3-Clause
 *
 * Unless required by applicable law or agreed to in writing, software distributed
 * under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR
 * CONDITIONS OF ANY KIND, either express or implied. See the License for the
 * specific language governing permissions and limitations under the License.
 */

package com.tencent.polaris.api.rpc;

import com.tencent.polaris.api.pojo.Instance;
import com.tencent.polaris.api.pojo.ServiceInstances;
import com.tencent.polaris.api.pojo.ServiceInstancesWrap;
import com.tencent.polaris.api.utils.StringUtils;
import com.tencent.polaris.specification.api.v1.model.ModelProto.MatchString;
import java.util.Arrays;
import java.util.Collection;
import java.util.Map;

/**
 * Response for instances query request.
 *
 * @author andrewshan
 * @date 2019/8/21
 */
public class InstancesResponse extends BaseEntity {

    private final ServiceInstances serviceInstances;

    private final Map metadata;

    private final int totalWeight;

    private final String subset;

    private final Map subsetMetadata;

    private final Instance[] instances;

    public InstancesResponse(ServiceInstances serviceInstances, String subset,
            Map subsetMetadata) {
        this.serviceInstances = serviceInstances;
        this.metadata = serviceInstances.getMetadata();
        this.setService(serviceInstances.getService());
        this.setNamespace(serviceInstances.getNamespace());
        Collection svcInstances = serviceInstances.getInstances();
        this.instances = svcInstances.toArray(new Instance[svcInstances.size()]);
        this.totalWeight = serviceInstances.getTotalWeight();
        this.subset = subset;
        this.subsetMetadata = subsetMetadata;
    }

    public InstancesResponse(ServiceInstances serviceInstances, Instance singleInstance, String subset,
            Map subsetMetadata) {
        this.serviceInstances = serviceInstances;
        this.metadata = serviceInstances.getMetadata();
        this.setService(serviceInstances.getService());
        this.setNamespace(serviceInstances.getNamespace());
        this.instances = new Instance[]{singleInstance};
        this.totalWeight = serviceInstances.getTotalWeight();
        this.subset = subset;
        this.subsetMetadata = subsetMetadata;
    }

    public Map getMetadata() {
        return metadata;
    }

    public Instance[] getInstances() {
        return instances;
    }

    public Instance getInstance() {
        if (null != instances && instances.length > 0) {
            return instances[0];
        }
        return null;
    }

    public ServiceInstances toServiceInstances() {
        return new ServiceInstancesWrap(serviceInstances, Arrays.asList(getInstances()), totalWeight);
    }

    public int getTotalWeight() {
        return totalWeight;
    }

    public ServiceInstances getServiceInstances() {
        return serviceInstances;
    }

    public boolean isServiceExist() {
        return StringUtils.isNotBlank(this.serviceInstances.getRevision());
    }

    public String getSubset() {
        return subset;
    }

    @Override
    public String toString() {
        return "InstancesResponse{" +
                "serviceInstances=" + serviceInstances +
                ", metadata=" + metadata +
                ", totalWeight=" + totalWeight +
                ", subset='" + subset + '\'' +
                ", instances=" + Arrays.toString(instances) +
                "} " + super.toString();
    }
}