/*
* Copyright 2002-2010 the original author or authors.
*
* 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.springframework.amqp.rabbit.log4j.web.controller;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.springframework.amqp.core.Binding;
import org.springframework.amqp.core.BindingBuilder;
import org.springframework.amqp.core.Queue;
import org.springframework.amqp.core.TopicExchange;
import org.springframework.amqp.rabbit.core.RabbitAdmin;
import org.springframework.amqp.rabbit.log4j.listener.AmqpLogMessage;
import org.springframework.amqp.rabbit.log4j.listener.AmqpLogMessageListener;
import org.springframework.beans.factory.DisposableBean;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestMethod;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.ResponseBody;
/**
* @author Tomas Lukosius
* @author Dave Syer
*
*/
@Controller
public class LogsController implements DisposableBean {
protected Log logger = LogFactory.getLog(this.getClass());
@Autowired
private TopicExchange exchange;
@Autowired
private Queue logQueue;
@Autowired
private RabbitAdmin admin;
@Autowired
private Binding binding;
@Autowired
private AmqpLogMessageListener messageListener;
@RequestMapping(value="/logs", method=RequestMethod.GET)
@ResponseBody
public List<AmqpLogMessage> logs(@RequestParam(required = false) Long timestamp) {
if (timestamp == null) {
timestamp = 0L;
}
ArrayList<AmqpLogMessage> list = new ArrayList<AmqpLogMessage>();
for (AmqpLogMessage log : messageListener.getLogs()) {
if (log.getTimestamp() > timestamp) {
list.add(log);
}
}
Collections.reverse(list);
return list;
}
@RequestMapping(value="/binding", method=RequestMethod.POST)
@ResponseBody
public Binding addBinding(@RequestParam String routingKey) {
try {
admin.removeBinding(binding);
Binding binding = BindingBuilder.bind(logQueue).to(exchange).with(routingKey);
admin.declareBinding(binding);
this.binding = binding;
} catch (RuntimeException e) {
// TODO: create a new BindingResult object to carry back error message
logger.error("Failed to declare queue or bind it with exchage", e);
}
return this.binding;
}
@RequestMapping(value="/binding", method=RequestMethod.GET)
@ResponseBody
public Binding getBinding() {
return binding;
}
public void destroy() throws Exception {
admin.removeBinding(binding);
}
}