59 lines
2.4 KiB
Java
59 lines
2.4 KiB
Java
import com.a.eye.skywalking.network.Client;
|
|
import com.a.eye.skywalking.network.grpc.AckSpan;
|
|
import com.a.eye.skywalking.network.grpc.RequestSpan;
|
|
import com.a.eye.skywalking.network.grpc.TraceId;
|
|
import com.a.eye.skywalking.network.grpc.client.SpanStorageClient;
|
|
import com.a.eye.skywalking.storage.util.NetUtils;
|
|
|
|
import java.util.ArrayList;
|
|
import java.util.List;
|
|
import java.util.concurrent.CountDownLatch;
|
|
|
|
public class StorageThread extends Thread {
|
|
|
|
private SpanStorageClient client;
|
|
private long count;
|
|
private CountDownLatch countDownLatch;
|
|
|
|
StorageThread(long count, CountDownLatch countDownLatch) {
|
|
client = new Client("10.128.7.241", 34000).newSpanStorageClient();
|
|
this.count = count;
|
|
this.countDownLatch = countDownLatch;
|
|
}
|
|
|
|
@Override
|
|
public void run() {
|
|
RequestSpan[] requestSpanList = new RequestSpan[10];
|
|
AckSpan[] ackSpanList = new AckSpan[10];
|
|
int cycle = 0;
|
|
for (int i = 0; i < count; i++) {
|
|
|
|
long value = System.currentTimeMillis();
|
|
RequestSpan requestSpan = RequestSpan.newBuilder().setSpanType(1).setAddress(NetUtils.getLocalAddress().toString()).setApplicationId("1").setCallType("1").setLevelId(0)
|
|
.setProcessNo("19287").setStartDate(System.currentTimeMillis())
|
|
.setTraceId(TraceId.newBuilder().addSegments(201611).addSegments(value).addSegments(8504828).addSegments(2277).addSegments(53).addSegments(3).build())
|
|
.setUserId("1").setViewPointId("http://localhost:8080/wwww/test/helloWorld").build();
|
|
|
|
AckSpan ackSpan = AckSpan.newBuilder().setLevelId(0).setCost(10).setTraceId(
|
|
TraceId.newBuilder().addSegments(201611).addSegments(value).addSegments(8504828).addSegments(2277).addSegments(Thread.currentThread().getId()).addSegments(3)
|
|
.build()).setStatusCode(0).setViewpointId("http://localhost:8080/wwww/test/helloWorld").build();
|
|
|
|
if (cycle == 10) {
|
|
client.sendACKSpan(ackSpanList);
|
|
client.sendRequestSpan(requestSpanList);
|
|
cycle = 0;
|
|
}
|
|
|
|
requestSpanList[cycle] = requestSpan;
|
|
ackSpanList[cycle] = ackSpan;
|
|
cycle++;
|
|
|
|
if (i % 10_000 == 0) {
|
|
System.out.println(i + " " + value);
|
|
}
|
|
}
|
|
|
|
countDownLatch.countDown();
|
|
}
|
|
}
|