Skip to content

Commit

Permalink
[#noissue] Cleanup
Browse files Browse the repository at this point in the history
  • Loading branch information
emeroad authored and ga-ram committed Nov 28, 2022
1 parent a025fae commit 22adb82
Show file tree
Hide file tree
Showing 5 changed files with 17 additions and 22 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -159,10 +159,11 @@ public void destroyed(REQ request, final Throwable throwable, final int statusCo

// TODO STATDISABLE this logic was added to disable statistics tracing
if (!trace.canSampled()) {
Trace copy = traceContext.removeTraceObject();
traceContext.removeTraceObject();
trace.close();
boolean status = isNotFailedStatus(statusCode);
uriStatRecorder.record(copy, request, rpcName, status, trace.getStartTime(), trace.getEndTime());
String uriTemplate = trace.getUriTemplate();
uriStatRecorder.record(uriTemplate, request, rpcName, status, trace.getStartTime(), trace.getEndTime());

return;
}
Expand All @@ -177,10 +178,11 @@ public void destroyed(REQ request, final Throwable throwable, final int statusCo
this.parameterRecorder.record(recorder, request, throwable);
} finally {
trace.traceBlockEnd();
Trace copy = this.traceContext.removeTraceObject();
this.traceContext.removeTraceObject();
trace.close();
boolean status = isNotFailedStatus(statusCode);
uriStatRecorder.record(copy, request, rpcName, status, trace.getStartTime(), trace.getEndTime());
String uriTemplate = trace.getUriTemplate();
uriStatRecorder.record(uriTemplate, request, rpcName, status, trace.getStartTime(), trace.getEndTime());
}
}

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,6 @@

package com.navercorp.pinpoint.bootstrap.plugin.uri;

import com.navercorp.pinpoint.bootstrap.context.Trace;

/**
* @author Taejin Koo
Expand All @@ -30,7 +29,7 @@ public static <T> UriStatRecorder<T> create() {
}

@Override
public void record(Trace trace, T request, String rawUri, boolean status, long startTime, long endTime) {
public void record(String uriTemplate, T request, String rawUri, boolean status, long startTime, long endTime) {
// do nothing
}

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -16,16 +16,14 @@

package com.navercorp.pinpoint.bootstrap.plugin.uri;

import com.navercorp.pinpoint.bootstrap.context.Trace;

/**
* Class that store information that obtained from Request and URIExtractor inside Pinpoint
*
* @author Taejin Koo
*/
public interface UriStatRecorder<T> {

void record(Trace trace, T request, String rawUri, boolean status, long startTime, long endTime);
void record(String uriTemplate, T request, String rawUri, boolean status, long startTime, long endTime);

}

Original file line number Diff line number Diff line change
Expand Up @@ -31,11 +31,11 @@
import com.navercorp.pinpoint.bootstrap.logging.PLoggerFactory;
import com.navercorp.pinpoint.bootstrap.plugin.RequestRecorderFactory;
import com.navercorp.pinpoint.bootstrap.plugin.proxy.ProxyRequestRecorder;
import com.navercorp.pinpoint.bootstrap.plugin.request.util.ParameterRecorder;
import com.navercorp.pinpoint.bootstrap.plugin.request.util.RemoteAddressResolverFactory;
import com.navercorp.pinpoint.bootstrap.plugin.request.RequestAdaptor;
import com.navercorp.pinpoint.bootstrap.plugin.request.RequestTraceReader;
import com.navercorp.pinpoint.bootstrap.plugin.request.ServerRequestRecorder;
import com.navercorp.pinpoint.bootstrap.plugin.request.util.ParameterRecorder;
import com.navercorp.pinpoint.bootstrap.plugin.request.util.RemoteAddressResolverFactory;
import com.navercorp.pinpoint.bootstrap.plugin.uri.BypassingUriExtractorService;
import com.navercorp.pinpoint.bootstrap.plugin.uri.UriStatRecorder;
import com.navercorp.pinpoint.bootstrap.plugin.uri.UriStatRecorderFactory;
Expand Down Expand Up @@ -261,7 +261,7 @@ private void recordUriStat(Object[] args, Trace trace) {
final String urlTemplate = ((VertxUrlTemplate)scope.getCurrentInvocation().getAttachment()).getUrlTemplate();
String uri = StringUtils.isEmpty(urlTemplate)? request.uri() : urlTemplate;
boolean status = isNotFailedStatus(request.response().getStatusCode());
uriStatRecorder.record(trace, request, uri, status, trace.getStartTime(), trace.getEndTime());
uriStatRecorder.record(urlTemplate, request, uri, status, trace.getStartTime(), trace.getEndTime());
}
}

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -16,14 +16,11 @@

package com.navercorp.pinpoint.profiler.context.recorder;

import com.navercorp.pinpoint.bootstrap.context.Trace;
import com.navercorp.pinpoint.bootstrap.context.TraceContext;
import com.navercorp.pinpoint.bootstrap.plugin.uri.UriExtractor;
import com.navercorp.pinpoint.bootstrap.plugin.uri.UriStatRecorder;
import com.navercorp.pinpoint.profiler.context.storage.UriStatStorage;

import org.apache.logging.log4j.Logger;
import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;

import java.util.Objects;

Expand All @@ -33,30 +30,29 @@
public class DefaultUriStatRecorder<T> implements UriStatRecorder<T> {

private final Logger logger = LogManager.getLogger(this.getClass());
private static final String NOT_FOUNDED = "/NOT_FOUND_URI";

private final UriExtractor<T> uriExtractor;
private final UriStatStorage uriStatStorage;

private final String NOT_FOUNDED = "/NOT_FOUND_URI";

public DefaultUriStatRecorder(UriExtractor<T> uriExtractor, UriStatStorage uriStatStorage) {
this.uriExtractor = Objects.requireNonNull(uriExtractor, "uriExtractor");
this.uriStatStorage = Objects.requireNonNull(uriStatStorage, "uriStatStorage");
}

@Override
public void record(Trace trace, T request, String rawUri, boolean status, long startTime, long endTime) {
public void record(String uriTemplate, T request, String rawUri, boolean status, long startTime, long endTime) {
String uri;

String userAttributeUri = uriExtractor.getUri(request, rawUri);
String interceptedUri = trace.getUriTemplate();
if (userAttributeUri != null) {
uri = userAttributeUri;
} else if (interceptedUri != null) {
uri = interceptedUri;
} else if (uriTemplate != null) {
uri = uriTemplate;
} else {
uri = NOT_FOUNDED;
logger.warn("can not extract uri. request:{}, rawUri:{}", request, rawUri);
logger.debug("can not extract uri. request:{}, rawUri:{}", request, rawUri);
}

uriStatStorage.store(uri, status, startTime, endTime);
Expand Down

0 comments on commit 22adb82

Please sign in to comment.