DataFunctionBlacklistResource.java

/**
 * Copyright (C) 2012 - present by OpenGamma Inc. and the OpenGamma group of companies
 * 
 * Please see distribution for license.
 */
package com.opengamma.engine.function.blacklist;

import java.util.Collection;
import java.util.concurrent.ExecutorService;

import javax.ws.rs.GET;
import javax.ws.rs.Path;
import javax.ws.rs.PathParam;
import javax.ws.rs.core.Response;

import org.fudgemsg.FudgeContext;
import org.fudgemsg.FudgeMsg;
import org.fudgemsg.MutableFudgeMsg;
import org.fudgemsg.mapping.FudgeSerializer;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.opengamma.transport.ByteArrayFudgeMessageSender;
import com.opengamma.transport.FudgeMessageSender;
import com.opengamma.transport.jms.JmsByteArrayMessageSender;
import com.opengamma.util.ArgumentChecker;
import com.opengamma.util.jms.JmsConnector;
import com.opengamma.util.rest.AbstractDataResource;

/**
 * Publishes a {@link FunctionBlacklist} to remote clients
 */
public class DataFunctionBlacklistResource extends AbstractDataResource implements FunctionBlacklistRuleListener {

  private static final Logger s_logger = LoggerFactory.getLogger(DataFunctionBlacklistResource.class);

  /**
   * Field containing the blacklist name when included in a response message.
   */
  public static final String NAME_FIELD = "name";
  /**
   * Field containing the blacklist modification count when included in a response message.
   */
  public static final String MODIFICATION_COUNT_FIELD = "modificationCount";
  /**
   * Field containing the blacklist rules when included in a response message.
   */
  public static final String RULES_FIELD = "rules";
  /**
   * Field containing new blacklist rules when published to remote listeners.
   */
  public static final String RULES_ADDED_FIELD = "add";
  /**
   * Field containing removed blacklist rules when published to remote listeners.
   */
  public static final String RULES_REMOVED_FIELD = "remove";
  /**
   * Field containing the JMS topic name updates will be published on.
   */
  public static final String JMS_TOPIC_FIELD = "jms";

  private final FunctionBlacklist _underlying;
  private final FudgeContext _fudgeContext;
  private final String _jmsTopic;
  private final FudgeMessageSender _publish;

  public DataFunctionBlacklistResource(final FunctionBlacklist underlying, final FudgeContext fudgeContext, final JmsConnector jmsConnector) {
    ArgumentChecker.notNull(underlying, "underlying");
    ArgumentChecker.notNull(fudgeContext, "fudgeContext");
    _underlying = underlying;
    _fudgeContext = fudgeContext;
    String name = jmsConnector.getTopicName();
    if (name == null) {
      _jmsTopic = underlying.getName() + "_BLACKLIST";
    } else {
      _jmsTopic = name + "_" + underlying.getName() + "_BLACKLIST";
    }
    JmsByteArrayMessageSender jmsSender = new JmsByteArrayMessageSender(_jmsTopic, jmsConnector.getJmsTemplateTopic());
    _publish = new ByteArrayFudgeMessageSender(jmsSender, fudgeContext);
    underlying.addRuleListener(this);
  }

  protected FunctionBlacklist getUnderlying() {
    return _underlying;
  }

  protected FudgeContext getFudgeContext() {
    return _fudgeContext;
  }

  @GET
  public Response info() {
    final FudgeSerializer fsc = new FudgeSerializer(getFudgeContext());
    final MutableFudgeMsg info = fsc.newMessage();
    info.add(NAME_FIELD, getUnderlying().getName());
    info.add(MODIFICATION_COUNT_FIELD, getUnderlying().getModificationCount());
    final MutableFudgeMsg rules = info.addSubMessage(RULES_FIELD, null);
    for (FunctionBlacklistRule rule : getUnderlying().getRules()) {
      fsc.addToMessage(rules, null, null, rule);
    }
    info.add(JMS_TOPIC_FIELD, _jmsTopic);
    return responseOk(info);
  }

  @GET
  @Path("mod/{mod}")
  public Response info(@PathParam("mod") final int mod) {
    final FudgeSerializer fsc = new FudgeSerializer(getFudgeContext());
    final MutableFudgeMsg info = fsc.newMessage();
    final int modificationCount = getUnderlying().getModificationCount();
    if (modificationCount != mod) {
      info.add(MODIFICATION_COUNT_FIELD, modificationCount);
      final MutableFudgeMsg rules = info.addSubMessage(RULES_FIELD, null);
      for (FunctionBlacklistRule rule : getUnderlying().getRules()) {
        fsc.addToMessage(rules, null, null, rule);
      }
    }
    return responseOk(info);
  }

  private final class Publish implements Runnable {

    private final FudgeMsg _msg;

    public Publish(final FudgeMsg msg) {
      _msg = msg;
    }

    @Override
    public void run() {
      try {
        _publish.send(_msg);
      } catch (RuntimeException e) {
        s_logger.warn("Error publishing JMS message", e);
      }
    }

  }

  @Override
  public void ruleAdded(final int modificationCount, final FunctionBlacklistRule rule, final ExecutorService defer) {
    final FudgeSerializer fsc = new FudgeSerializer(getFudgeContext());
    final MutableFudgeMsg msg = fsc.newMessage();
    msg.add(MODIFICATION_COUNT_FIELD, modificationCount);
    final MutableFudgeMsg rulesMessage = msg.addSubMessage(RULES_ADDED_FIELD, null);
    fsc.addToMessage(rulesMessage, null, null, rule);
    defer.submit(new Publish(msg));
  }

  @Override
  public void rulesAdded(final int modificationCount, final Collection<FunctionBlacklistRule> rules, final ExecutorService defer) {
    final FudgeSerializer fsc = new FudgeSerializer(getFudgeContext());
    final MutableFudgeMsg msg = fsc.newMessage();
    msg.add(MODIFICATION_COUNT_FIELD, modificationCount);
    final MutableFudgeMsg rulesMessage = msg.addSubMessage(RULES_ADDED_FIELD, null);
    for (FunctionBlacklistRule rule : rules) {
      fsc.addToMessage(rulesMessage, null, null, rule);
    }
    defer.submit(new Publish(msg));
  }

  @Override
  public void ruleRemoved(final int modificationCount, final FunctionBlacklistRule rule, final ExecutorService defer) {
    final FudgeSerializer fsc = new FudgeSerializer(getFudgeContext());
    final MutableFudgeMsg msg = fsc.newMessage();
    msg.add(MODIFICATION_COUNT_FIELD, modificationCount);
    final MutableFudgeMsg rulesMessage = msg.addSubMessage(RULES_REMOVED_FIELD, null);
    fsc.addToMessage(rulesMessage, null, null, rule);
    defer.submit(new Publish(msg));
  }

  @Override
  public void rulesRemoved(final int modificationCount, final Collection<FunctionBlacklistRule> rules, final ExecutorService defer) {
    final FudgeSerializer fsc = new FudgeSerializer(getFudgeContext());
    final MutableFudgeMsg msg = fsc.newMessage();
    msg.add(MODIFICATION_COUNT_FIELD, modificationCount);
    final MutableFudgeMsg rulesMessage = msg.addSubMessage(RULES_REMOVED_FIELD, null);
    for (FunctionBlacklistRule rule : rules) {
      fsc.addToMessage(rulesMessage, null, null, rule);
    }
    defer.submit(new Publish(msg));
  }

}