com.hotels.styx.api.client.OriginsSnapshot Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of styx-api Show documentation
There is a newer version: 1.0.0.beta9
/*
  Copyright (C) 2013-2018 Expedia Inc.

  Licensed under the Apache License, Version 2.0 (the "License");
  you may not use this file except in compliance with the License.
  You may obtain a copy of the License at

  http://www.apache.org/licenses/LICENSE-2.0

  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.hotels.styx.api.client;

import com.hotels.styx.api.Id;

import java.util.Collection;
import java.util.HashMap;
import java.util.Map;
import java.util.Set;

import static com.google.common.base.MoreObjects.toStringHelper;
import static com.google.common.base.Preconditions.checkNotNull;
import static com.hotels.styx.api.Id.id;
import static com.hotels.styx.api.client.Origin.newOriginBuilder;
import static java.util.stream.Collectors.toSet;

/**
 * Holds the state of currently configured origins, i.e. whether the origin is active, inactive or disabled.
 */
public final class OriginsSnapshot {
    private final Id appId;
    private final Set activeOrigins;
    private final Set inactiveOrigins;
    private final Set disabledOrigins;
    private final Map allOriginsById = new HashMap<>();

    OriginsSnapshot(String appId,
                    Collection activeOrigins,
                    Collection inactiveOrigins,
                    Collection disabledOrigins) {
        this.appId = id(appId);
        this.activeOrigins = withAppId(activeOrigins, appId);
        this.inactiveOrigins = withAppId(inactiveOrigins, appId);
        this.disabledOrigins = withAppId(disabledOrigins, appId);
        mapOriginsById();
    }

    /**
     * Construct a snapshot from application ID, and three sets of connection pools: active, inactive and disabled.
     *
     * @param appId           application ID
     * @param activeOrigins   connection pools for active origins
     * @param inactiveOrigins connection pools for inactive origins
     * @param disabledOrigins connection pools for disabled origins
     */
    public OriginsSnapshot(Id appId,
                           Collection activeOrigins,
                           Collection inactiveOrigins,
                           Collection disabledOrigins) {
        this.appId = checkNotNull(appId);
        this.activeOrigins = mapToOrigins(activeOrigins);
        this.inactiveOrigins = mapToOrigins(inactiveOrigins);
        this.disabledOrigins = mapToOrigins(disabledOrigins);
        mapOriginsById();
    }

    private void mapOriginsById() {
        this.activeOrigins.forEach(origin -> allOriginsById.put(origin.id(), origin));
        this.inactiveOrigins.forEach(origin -> allOriginsById.put(origin.id(), origin));
        this.disabledOrigins.forEach(origin -> allOriginsById.put(origin.id(), origin));
    }

    private Set mapToOrigins(Collection activeOrigins) {
        return activeOrigins.stream()
                .map(RemoteHost::origin)
                .collect(toSet());
    }

    private static Set withAppId(Collection origins, String appId) {
        return origins.stream().map(origin -> newOriginBuilder(origin)
                .applicationId(appId)
                .build())
                .collect(toSet());
    }

    /**
     * ID of the application that all of the origins in the inventory belong to.
     *
     * @return application ID
     */
    public Id appId() {
        return appId;
    }

    String appIdAsString() {
        return appId.toString();
    }

    /**
     * Set of active origins.
     *
     * @return active origins
     */
    public Set activeOrigins() {
        return activeOrigins;
    }

    /**
     * Set of inactive origins.
     *
     * @return inactive origins
     */
    public Set inactiveOrigins() {
        return inactiveOrigins;
    }

    /**
     * Set of disabled origins.
     *
     * @return disabled origins
     */
    public Set disabledOrigins() {
        return disabledOrigins;
    }

    /**
     * Finds out whether an origin belongs to this origin inventory, i.e. if it is present within any of the three sets: active, inactive, disabled.
     *
     * @param originId origin ID
     * @return true if the origin belongs to this origin inventory.
     */
    public boolean containsOrigin(Id originId) {
        return allOriginsById.containsKey(originId);
    }

    @Override
    public String toString() {
        return toStringHelper(this)
                .add("appId", appId)
                .add("activeOrigins", activeOrigins)
                .add("inactiveOrigins", inactiveOrigins)
                .add("disabledOrigins", disabledOrigins)
                .toString();
    }
}