/*
* Copyright 2014 NAVER Corp.
*
* 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 com.navercorp.pinpoint.profiler.sender;
import com.navercorp.pinpoint.profiler.context.Span;
import com.navercorp.pinpoint.profiler.context.SpanChunk;
import com.navercorp.pinpoint.rpc.FutureListener;
import com.navercorp.pinpoint.rpc.ResponseMessage;
import com.navercorp.pinpoint.rpc.client.PinpointClientReconnectEventListener;
import org.apache.thrift.TBase;
import java.util.concurrent.atomic.AtomicInteger;
/**
* @author emeroad
*/
public class CountingDataSender implements EnhancedDataSender {
private final AtomicInteger requestCounter = new AtomicInteger();
private final AtomicInteger requestRetryCounter = new AtomicInteger();
private final AtomicInteger requestResponseListenerCounter = new AtomicInteger();
private final AtomicInteger senderCounter = new AtomicInteger();
private final AtomicInteger spanCounter = new AtomicInteger();
private final AtomicInteger spanChunkCounter = new AtomicInteger();
@Override
public boolean request(TBase<?, ?> data) {
requestCounter.incrementAndGet();
return false;
}
@Override
public boolean request(TBase<?, ?> data, int retry) {
requestRetryCounter.incrementAndGet();
return false;
}
@Override
public boolean request(TBase<?, ?> data, FutureListener<ResponseMessage> listener) {
return false;
}
@Override
public boolean addReconnectEventListener(PinpointClientReconnectEventListener eventListener) {
return false;
}
@Override
public boolean removeReconnectEventListener(PinpointClientReconnectEventListener eventListener) {
return false;
}
@Override
public boolean send(TBase<?, ?> data) {
senderCounter.incrementAndGet();
if (data instanceof Span) {
this.spanCounter.incrementAndGet();
} else if (data instanceof SpanChunk) {
this.spanChunkCounter.incrementAndGet();
}
return false;
}
@Override
public void stop() {
this.requestCounter.set(0);
this.requestRetryCounter.set(0);
this.requestResponseListenerCounter.set(0);
this.senderCounter.set(0);
this.spanCounter.set(0);
this.spanChunkCounter.set(0);
}
public int getRequestCounter() {
return requestCounter.get();
}
public int getRequestRetryCounter() {
return requestRetryCounter.get();
}
public int getRequestResponseListenerCounter() {
return requestResponseListenerCounter.get();
}
public int getSenderCounter() {
return senderCounter.get();
}
public int getSpanChunkCounter() {
return spanChunkCounter.get();
}
public int getSpanCounter() {
return spanCounter.get();
}
public int getTotalCount() {
return requestCounter.get() + requestRetryCounter.get() + requestResponseListenerCounter.get() + senderCounter.get();
}
@Override
public String toString() {
return "CountingDataSender{" +
"requestCounter=" + requestCounter +
", requestRetryCounter=" + requestRetryCounter +
", requestResponseListenerCounter=" + requestResponseListenerCounter +
", senderCounter=" + senderCounter +
", spanCounter=" + spanCounter +
", spanChunkCounter=" + spanChunkCounter +
'}';
}
}