/*
* Copyright 2014-2016 CyberVision, 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 org.kaaproject.kaa.server.operations.service.event;
import java.util.HashMap;
import java.util.Map;
public class EventDeliveryTable {
Map<EndpointEvent, Map<RouteTableAddress, DeliveryState>> data;
public EventDeliveryTable() {
super();
this.data = new HashMap<>();
}
/**
* Register new delivery attempt.
*
* @param event the event to delivery
* @param addresses the destination of event
*/
public void registerDeliveryAttempt(EndpointEvent event, RouteTableAddress... addresses) {
Map<RouteTableAddress, DeliveryState> attempts = data.get(event);
if (attempts == null) {
attempts = new HashMap<>();
data.put(event, attempts);
}
for (RouteTableAddress address : addresses) {
attempts.put(address, DeliveryState.PENDING);
}
}
public boolean isDeliveryStarted(EndpointEvent event, RouteTableAddress address) {
Map<RouteTableAddress, DeliveryState> attempts = data.get(event);
return attempts != null && attempts.containsKey(address);
}
public boolean clear(EndpointEvent event) {
return data.remove(event) != null;
}
/**
* Register success delivery attempt.
*
* @param event the event to delivery
* @param address the destination of event
*/
public void registerDeliverySuccess(EndpointEvent event, RouteTableAddress address) {
Map<RouteTableAddress, DeliveryState> attempts = data.get(event);
if (attempts == null) {
attempts = new HashMap<>();
data.put(event, attempts);
}
attempts.put(address, DeliveryState.DELIVERED);
}
/**
* Register fail delivery attempt.
*
* @param event the event to delivery
* @param address the destination of event
*/
public void registerDeliveryFailure(EndpointEvent event, RouteTableAddress address) {
Map<RouteTableAddress, DeliveryState> attempts = data.get(event);
if (attempts != null) {
attempts.remove(address);
}
}
private static enum DeliveryState {
PENDING,
DELIVERED
}
}