Search in sources :

Example 6 with Tracer

use of org.ballerinalang.util.tracer.Tracer in project ballerina by ballerina-lang.

the class Patch method createOutboundRequestMsg.

protected HTTPCarbonMessage createOutboundRequestMsg(Context context) {
    HTTPCarbonMessage outboundRequestMsg = super.createOutboundRequestMsg(context);
    outboundRequestMsg.setProperty(HttpConstants.HTTP_METHOD, HttpConstants.HTTP_METHOD_PATCH);
    Tracer tracer = TraceUtil.getParentTracer(context.getParentWorkerExecutionContext());
    HttpUtil.injectHeaders(outboundRequestMsg, tracer.getProperties());
    tracer.addTags(HttpUtil.extractTraceTags(outboundRequestMsg));
    return outboundRequestMsg;
}
Also used : HTTPCarbonMessage(org.wso2.transport.http.netty.message.HTTPCarbonMessage) Tracer(org.ballerinalang.util.tracer.Tracer)

Example 7 with Tracer

use of org.ballerinalang.util.tracer.Tracer in project ballerina by ballerina-lang.

the class Put method createOutboundRequestMsg.

protected HTTPCarbonMessage createOutboundRequestMsg(Context context) {
    HTTPCarbonMessage outboundRequestMsg = super.createOutboundRequestMsg(context);
    outboundRequestMsg.setProperty(HttpConstants.HTTP_METHOD, HttpConstants.HTTP_METHOD_PUT);
    Tracer tracer = TraceUtil.getParentTracer(context.getParentWorkerExecutionContext());
    HttpUtil.injectHeaders(outboundRequestMsg, tracer.getProperties());
    tracer.addTags(HttpUtil.extractTraceTags(outboundRequestMsg));
    return outboundRequestMsg;
}
Also used : HTTPCarbonMessage(org.wso2.transport.http.netty.message.HTTPCarbonMessage) Tracer(org.ballerinalang.util.tracer.Tracer)

Example 8 with Tracer

use of org.ballerinalang.util.tracer.Tracer in project ballerina by ballerina-lang.

the class Execute method createOutboundRequestMsg.

protected HTTPCarbonMessage createOutboundRequestMsg(Context context) {
    // Extract Argument values
    BStruct bConnector = (BStruct) context.getRefArgument(0);
    String httpVerb = context.getStringArgument(0);
    String path = context.getStringArgument(1);
    BStruct requestStruct = ((BStruct) context.getRefArgument(1));
    // TODO check below line
    HTTPCarbonMessage defaultCarbonMsg = HttpUtil.createHttpCarbonMessage(true);
    HTTPCarbonMessage outboundRequestMsg = HttpUtil.getCarbonMsg(requestStruct, defaultCarbonMsg);
    prepareOutboundRequest(context, bConnector, path, outboundRequestMsg);
    // If the verb is not specified, use the verb in incoming message
    if (httpVerb == null || "".equals(httpVerb)) {
        httpVerb = (String) outboundRequestMsg.getProperty(HttpConstants.HTTP_METHOD);
    }
    outboundRequestMsg.setProperty(HttpConstants.HTTP_METHOD, httpVerb.trim().toUpperCase(Locale.getDefault()));
    if (outboundRequestMsg.getHeader(HttpHeaderNames.ACCEPT_ENCODING.toString()) == null) {
        outboundRequestMsg.setHeader(HttpHeaderNames.ACCEPT_ENCODING.toString(), ENCODING_DEFLATE + ", " + ENCODING_GZIP);
    }
    Tracer tracer = TraceUtil.getParentTracer(context.getParentWorkerExecutionContext());
    HttpUtil.injectHeaders(outboundRequestMsg, tracer.getProperties());
    tracer.addTags(HttpUtil.extractTraceTags(outboundRequestMsg));
    return outboundRequestMsg;
}
Also used : BStruct(org.ballerinalang.model.values.BStruct) HTTPCarbonMessage(org.wso2.transport.http.netty.message.HTTPCarbonMessage) Tracer(org.ballerinalang.util.tracer.Tracer)

Example 9 with Tracer

use of org.ballerinalang.util.tracer.Tracer in project ballerina by ballerina-lang.

the class Forward method createOutboundRequestMsg.

protected HTTPCarbonMessage createOutboundRequestMsg(Context context) {
    BStruct bConnector = (BStruct) context.getRefArgument(0);
    String path = context.getStringArgument(0);
    BStruct requestStruct = ((BStruct) context.getRefArgument(1));
    if (requestStruct.getNativeData(HttpConstants.REQUEST) == null) {
        throw new BallerinaException("invalid inbound request parameter");
    }
    HTTPCarbonMessage outboundRequestMsg = HttpUtil.getCarbonMsg(requestStruct, HttpUtil.createHttpCarbonMessage(true));
    prepareOutboundRequest(context, bConnector, path, outboundRequestMsg);
    String httpVerb = (String) outboundRequestMsg.getProperty(HttpConstants.HTTP_METHOD);
    outboundRequestMsg.setProperty(HttpConstants.HTTP_METHOD, httpVerb.trim().toUpperCase(Locale.getDefault()));
    Tracer tracer = TraceUtil.getParentTracer(context.getParentWorkerExecutionContext());
    HttpUtil.injectHeaders(outboundRequestMsg, tracer.getProperties());
    tracer.addTags(HttpUtil.extractTraceTags(outboundRequestMsg));
    return outboundRequestMsg;
}
Also used : BStruct(org.ballerinalang.model.values.BStruct) HTTPCarbonMessage(org.wso2.transport.http.netty.message.HTTPCarbonMessage) Tracer(org.ballerinalang.util.tracer.Tracer) BallerinaException(org.ballerinalang.util.exceptions.BallerinaException)

Example 10 with Tracer

use of org.ballerinalang.util.tracer.Tracer in project ballerina by ballerina-lang.

the class Post method createOutboundRequestMsg.

protected HTTPCarbonMessage createOutboundRequestMsg(Context context) {
    HTTPCarbonMessage outboundRequestMsg = super.createOutboundRequestMsg(context);
    outboundRequestMsg.setProperty(HttpConstants.HTTP_METHOD, HttpConstants.HTTP_METHOD_POST);
    Tracer tracer = TraceUtil.getParentTracer(context.getParentWorkerExecutionContext());
    HttpUtil.injectHeaders(outboundRequestMsg, tracer.getProperties());
    tracer.addTags(HttpUtil.extractTraceTags(outboundRequestMsg));
    return outboundRequestMsg;
}
Also used : HTTPCarbonMessage(org.wso2.transport.http.netty.message.HTTPCarbonMessage) Tracer(org.ballerinalang.util.tracer.Tracer)

Aggregations

Tracer (org.ballerinalang.util.tracer.Tracer)12 HTTPCarbonMessage (org.wso2.transport.http.netty.message.HTTPCarbonMessage)11 BStruct (org.ballerinalang.model.values.BStruct)4 HashMap (java.util.HashMap)2 Map (java.util.Map)2 CallableUnitCallback (org.ballerinalang.bre.bvm.CallableUnitCallback)2 BLangConnectorSPIUtil (org.ballerinalang.connector.api.BLangConnectorSPIUtil)2 BallerinaConnectorException (org.ballerinalang.connector.api.BallerinaConnectorException)2 Executor (org.ballerinalang.connector.api.Executor)2 BValue (org.ballerinalang.model.values.BValue)2 PROTOCOL_PACKAGE_HTTP (org.ballerinalang.net.http.HttpConstants.PROTOCOL_PACKAGE_HTTP)2 BallerinaException (org.ballerinalang.util.exceptions.BallerinaException)2 TraceConstants (org.ballerinalang.util.tracer.TraceConstants)2 TraceManagerWrapper (org.ballerinalang.util.tracer.TraceManagerWrapper)2 Logger (org.slf4j.Logger)2 LoggerFactory (org.slf4j.LoggerFactory)2 HashSet (java.util.HashSet)1 List (java.util.List)1 Semaphore (java.util.concurrent.Semaphore)1 AtomicBoolean (java.util.concurrent.atomic.AtomicBoolean)1