com.ajjpj.asysmon.datasink.offloadhttpjson.AJsonOffloadingEntity Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of a-sysmon-core Show documentation
There is a newer version: 1.0-pre28
package com.ajjpj.asysmon.datasink.offloadhttpjson;

import com.ajjpj.abase.io.AJsonSerHelper;
import com.ajjpj.asysmon.data.ACorrelationId;
import com.ajjpj.asysmon.data.AHierarchicalData;
import com.ajjpj.asysmon.data.AHierarchicalDataRoot;
import com.ajjpj.asysmon.data.AScalarDataPoint;
import org.apache.http.entity.AbstractHttpEntity;

import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.util.ArrayList;
import java.util.Collection;
import java.util.List;


/**
 * @author arno
 */
class AJsonOffloadingEntity extends AbstractHttpEntity {
    private final String sender;
    private final String senderInstance;

    private final List traces = new ArrayList();
    private final List scalarData = new ArrayList();

    AJsonOffloadingEntity(List traces, Collection scalarData, String sender, String senderInstance) {
        setChunked(true);
        //TODO content type, encoding

        this.sender = sender;
        this.senderInstance = senderInstance;

        this.traces.addAll(traces);
        this.scalarData.addAll(scalarData);
    }

    @Override public boolean isRepeatable() {
        return false;
    }

    @Override public long getContentLength() {
        return -1; // streaming sending means chunked HTTP
    }

    @Override public InputStream getContent() throws IOException, IllegalStateException {
        throw new UnsupportedOperationException();
    }

    @Override public void writeTo(OutputStream outstream) throws IOException {
        final AJsonSerHelper ser = new AJsonSerHelper(outstream);

        ser.startObject(); // start 'RootNode'

        ser.writeKey("sender");
        ser.writeStringLiteral(sender);

        ser.writeKey("senderInstance");
        ser.writeStringLiteral(senderInstance);

        ser.writeKey("senderTimestamp");
        ser.writeNumberLiteral(System.currentTimeMillis(), 0);

        ser.writeKey("traces");
        ser.startArray();
        for(AHierarchicalDataRoot trace: traces) {
            writeTraceRoot(ser, trace);
        }
        ser.endArray();

        ser.writeKey("scalars");
        ser.startArray();
        for(AScalarDataPoint scalar: scalarData) {
            writeScalar(ser, scalar);
        }
        ser.endArray();

        ser.endObject();
    }

    private void writeTraceRoot(AJsonSerHelper ser, AHierarchicalDataRoot trace) throws IOException {
        ser.startObject(); // start 'TraceRootNode'

        ser.writeKey("uuid");
        ser.writeStringLiteral(trace.getUuid().toString());

        ser.writeKey("startedFlows");
        ser.startArray();
        for(ACorrelationId flow: trace.getStartedFlows()) {
            writeCorrelationId(ser, flow);
        }
        ser.endArray();

        ser.writeKey("joinedFlows");
        ser.startArray();
        for(ACorrelationId flow: trace.getJoinedFlows()) {
            writeCorrelationId(ser, flow);
        }
        ser.endArray();

        ser.writeKey("trace");
        writeTraceRec(ser, trace.getRootNode());

        ser.endObject();
    }

    private void writeCorrelationId(AJsonSerHelper ser, ACorrelationId flow) throws IOException {
        ser.startObject(); // start 'CorrelationId'

        ser.writeKey("kind");
        ser.writeStringLiteral(flow.getQualifier());

        ser.writeKey("ident");
        ser.writeStringLiteral(flow.getId());

        ser.endObject();
    }

    private void writeTraceRec(AJsonSerHelper ser, AHierarchicalData trace) throws IOException {
        ser.startObject(); // start 'TraceNode'

        ser.writeKey("isSerial");
        ser.writeBooleanLiteral(trace.isSerial());

        ser.writeKey("senderStartTimeMillis");
        ser.writeNumberLiteral(trace.getStartTimeMillis(), 0);

        ser.writeKey("durationNanos");
        ser.writeNumberLiteral(trace.getDurationNanos(), 0);

        ser.writeKey("identifier");
        ser.writeStringLiteral(trace.getIdentifier());

        ser.writeKey("parameters");
        ser.startObject();
        for(String key: trace.getParameters().keySet()) {
            ser.writeKey(key);
            ser.writeStringLiteral(trace.getParameters().get(key));
        }
        ser.endObject();

        ser.writeKey("children");
        ser.startArray();
        for(AHierarchicalData child: trace.getChildren()) {
            writeTraceRec(ser, child);
        }
        ser.endArray();

        ser.endObject();
    }

    private void writeScalar(AJsonSerHelper ser, AScalarDataPoint scalar) throws IOException {
        ser.startObject(); // start 'ScalarNode'

        ser.writeKey("uuid");
        ser.writeStringLiteral(scalar.getUuid().toString());

        ser.writeKey("senderTimestamp");
        ser.writeNumberLiteral(scalar.getTimestamp(), 0);

        ser.writeKey("name");
        ser.writeStringLiteral(scalar.getName());

        ser.writeKey("value");
        ser.writeNumberLiteral(scalar.getValueRaw(), scalar.getNumFracDigits());

        ser.endObject();
    }

    @Override public boolean isStreaming() {
        return true;
    }
}