io.tilt.minka.domain.Shard.java Source code

Introduction

Here is the source code for io.tilt.minka.domain.Shard.java

Source

/*
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You 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 io.tilt.minka.domain;

import java.util.Comparator;
import java.util.List;

import org.apache.commons.lang.builder.HashCodeBuilder;
import org.joda.time.DateTime;
import org.joda.time.DateTimeZone;

import io.tilt.minka.broker.EventBroker.BrokerChannel;
import io.tilt.minka.utils.SlidingSortedSet;

/**
 * Compound information about a node, maintained by the Leader
 * 
 * @author Cristian Gonzalez
 * @since Nov 5, 2015
 */
public class Shard implements Comparator<Shard> {

    private static final int MAX_HEARBEATS_TO_EVALUATE = 50;

    private final BrokerChannel brokerChannel;
    private final NetworkShardID shardId;
    private final DateTime firstTimeSeen;
    private DateTime lastStatusChange;
    private final SlidingSortedSet<Heartbeat> cardiacLapse;
    private Workload workLoad;
    private ShardState serviceState;

    public Shard(final BrokerChannel channel, final NetworkShardID memberId) {
        super();
        this.brokerChannel = channel;
        this.shardId = memberId;
        this.serviceState = ShardState.JOINING;
        this.cardiacLapse = new SlidingSortedSet<>(MAX_HEARBEATS_TO_EVALUATE);
        this.firstTimeSeen = new DateTime(DateTimeZone.UTC);
        this.lastStatusChange = new DateTime(DateTimeZone.UTC);
    }

    public DateTime getLastStatusChange() {
        return this.lastStatusChange;
    }

    public DateTime getFirstTimeSeen() {
        return this.firstTimeSeen;
    }

    public BrokerChannel getBrokerChannel() {
        return this.brokerChannel;
    }

    public NetworkShardID getShardID() {
        return this.shardId;
    }

    public void addHeartbeat(final Heartbeat hb) {
        if (hb.getStateChange() != null) {
            this.serviceState = hb.getStateChange();
        }
        this.cardiacLapse.add(hb);
    }

    public List<Heartbeat> getHeartbeats() {
        return this.cardiacLapse.values();
    }

    public Workload getWorkLoad() {
        return this.workLoad;
    }

    public void setWorkLoad(Workload workLoad) {
        this.workLoad = workLoad;
    }

    public ShardState getState() {
        return this.serviceState;
    }

    public void setState(ShardState serviceState) {
        this.serviceState = serviceState;
        this.lastStatusChange = new DateTime(DateTimeZone.UTC);
    }

    public int hashCode() {
        return new HashCodeBuilder().append(this.getShardID()).toHashCode();
    }

    @Override
    public boolean equals(Object obj) {
        if (obj instanceof Shard) {
            Shard other = (Shard) obj;
            return other.getShardID().equals(getShardID());
        } else {
            return false;
        }
    }

    @Override
    public String toString() {
        return this.shardId.toString();
    }

    @Override
    public int compare(Shard o1, Shard o2) {
        return o1.getFirstTimeSeen().compareTo(o2.getFirstTimeSeen());
    }
}