com.aol.micro.server.events.RequestsBeingExecuted Maven / Gradle / Ivy

Go to download
package com.aol.micro.server.events;

import javax.xml.bind.annotation.XmlAccessType;
import javax.xml.bind.annotation.XmlAccessorType;

import lombok.AllArgsConstructor;
import lombok.Getter;
import lombok.experimental.Builder;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Component;

import com.google.common.eventbus.EventBus;
import com.google.common.eventbus.Subscribe;

@Component
public class RequestsBeingExecuted {

	private final EventBus bus;
	private final ActiveEvents> events = new ActiveEvents();
	@Getter
	private final String type;
	
	@Autowired
	public RequestsBeingExecuted(EventBus bus,  boolean queryCapture){
		this.bus = bus;
		this.type = "default";
		if(queryCapture)
			bus.register(this);
		
	}
	
	public RequestsBeingExecuted(EventBus bus, String type){
		this.bus = bus;
		this.type = type;
		bus.register(this);
		
	}
	
	public int events(){
		return events.events();
	}
	
	public int size(){
		return events.size();
	}
	
	public String toString(){
		return events.toString();
	}
	@Subscribe
	public void finished(RemoveQuery data) {
		if(type.equals(data.getData().type))
			events.finished(buildId(data.getData()));
 
	}
	@Subscribe
	public void processing(AddQuery data) {
		if(type.equals(data.getData().type)){
			String id = buildId(data.getData());
			events.active(id, data.getData());
		}
		
	}
	

	private String buildId(RequestData data) {
		String id =  data.correlationId+ "-"+data.getProcessingThread();
		return id;
	}
	
	
	public static class AddQuery extends AddEvent>{

		public AddQuery(RequestData data) {
			super(data);
		}
		
	}
	
	public static class RemoveQuery extends RemoveEvent>{

		public RemoveQuery(RequestData data) {
			super(data);
		}
		
	}

	
	@AllArgsConstructor
	@Builder
	@XmlAccessorType(XmlAccessType.FIELD)
	static class RequestData  extends BaseEventInfo {

		private final long correlationId;
		
		private final T query;
		
		private final String type;
		private final Object additionalData;
	}

}