com.tencent.polaris.api.pojo.DefaultServiceInstances Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of polaris-all Show documentation
All in one project for polaris-java
There is a newer version: 2.0.0.0
/*
 * 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.pojo;

import com.tencent.polaris.api.utils.CollectionUtils;
import com.tencent.polaris.client.pojo.Node;
import java.util.Collections;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Objects;

public class DefaultServiceInstances implements ServiceInstances {

    private final ServiceKey serviceKey;

    private final List instances;

    private final Map idMap;

    private final Map nodeMap;

    private final int totalWeight;

    private final int hashCode;

    private final String revision;

    public DefaultServiceInstances(ServiceKey serviceKey, List instances) {
        this.serviceKey = serviceKey;
        this.instances = Collections.unmodifiableList(instances);
        this.totalWeight = getTotalWeight(instances);
        hashCode = Objects.hash(serviceKey, instances);
        revision = Integer.toHexString(hashCode);
        idMap = new HashMap<>();
        nodeMap = new HashMap<>();
        for (Instance instance : instances) {
            idMap.put(instance.getId(), instance);
            nodeMap.put(new Node(instance.getHost(), instance.getPort()), instance);
        }
    }

    private int getTotalWeight(List instances) {
        int totalWeight = 0;
        if (CollectionUtils.isNotEmpty(instances)) {
            for (Instance instance : instances) {
                totalWeight += instance.getWeight();
            }
        }
        return totalWeight;
    }

    @Override
    public ServiceKey getServiceKey() {
        return serviceKey;
    }

    @Override
    public int getTotalWeight() {
        return totalWeight;
    }

    @Override
    public List getInstances() {
        return instances;
    }

    @Override
    public boolean isInitialized() {
        return true;
    }

    @Override
    public String getRevision() {
        return revision;
    }

    @Override
    public Instance getInstance(Node node) {
        return nodeMap.get(node);
    }

    @Override
    public Instance getInstance(String id) {
        return idMap.get(id);
    }

    @Override
    public Map getMetadata() {
        return null;
    }

    @Override
    public String getService() {
        return this.serviceKey.getService();
    }

    @Override
    public String getNamespace() {
        return this.serviceKey.getNamespace();
    }

    @Override
    public boolean equals(Object o) {
        if (this == o) {
            return true;
        }
        if (o == null || getClass() != o.getClass()) {
            return false;
        }
        DefaultServiceInstances that = (DefaultServiceInstances) o;
        return Objects.equals(serviceKey, that.serviceKey) &&
                Objects.equals(instances, that.instances);
    }

    @Override
    public int hashCode() {
        return hashCode;
    }
}