Unverified Commit cee0525f authored by 吴晟's avatar 吴晟 Committed by GitHub
Browse files

Make a test case for span limit. (#2157)

parent b5eb9e30

apm-sniffer/apm-agent-core/src/main/java/org/apache/skywalking/apm/agent/core/remote/GRPCChannelManager.java

+8 −17
Original line number Diff line number Diff line
@@ -18,23 +18,12 @@ 


package org.apache.skywalking.apm.agent.core.remote;



import io.grpc.Channel;

import io.grpc.Status;

import io.grpc.StatusRuntimeException;

import java.util.Arrays;

import java.util.Collections;

import java.util.LinkedList;

import java.util.List;

import java.util.Random;

import java.util.concurrent.Executors;

import java.util.concurrent.ScheduledFuture;

import java.util.concurrent.TimeUnit;

import org.apache.skywalking.apm.agent.core.boot.BootService;

import org.apache.skywalking.apm.agent.core.boot.DefaultImplementor;

import org.apache.skywalking.apm.agent.core.boot.DefaultNamedThreadFactory;

import io.grpc.*;

import java.util.*;

import java.util.concurrent.*;

import org.apache.skywalking.apm.agent.core.boot.*;

import org.apache.skywalking.apm.agent.core.conf.Config;

import org.apache.skywalking.apm.agent.core.logging.api.ILog;

import org.apache.skywalking.apm.agent.core.logging.api.LogManager;

import org.apache.skywalking.apm.agent.core.logging.api.*;

import org.apache.skywalking.apm.util.RunnableWithExceptionProtection;



/**
@@ -81,7 +70,9 @@ public class GRPCChannelManager implements BootService, Runnable { 


    @Override

    public void shutdown() throws Throwable {

        if (connectCheckFuture != null) {

            connectCheckFuture.cancel(true);

        }

        if (managedChannel != null) {

            managedChannel.shutdownNow();

        }

apm-sniffer/apm-agent-core/src/test/java/org/apache/skywalking/apm/agent/core/context/TracingContextTest.java

0 → 100644
+66 −0
Original line number Diff line number Diff line 
/*

 * Licensed to the Apache Software Foundation (ASF) under one or more

 * contributor license agreements.  See the NOTICE file distributed with

 * this work for additional information regarding copyright ownership.

 * The ASF licenses this file to You 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 org.apache.skywalking.apm.agent.core.context;



import org.apache.skywalking.apm.agent.core.boot.ServiceManager;

import org.apache.skywalking.apm.agent.core.conf.RemoteDownstreamConfig;

import org.apache.skywalking.apm.agent.core.context.trace.*;

import org.junit.*;



public class TracingContextTest {

    @BeforeClass

    public static void setup() {

        ServiceManager.INSTANCE.boot();

        RemoteDownstreamConfig.Agent.SERVICE_INSTANCE_ID = 5;

    }



    @AfterClass

    public static void clear() {

        RemoteDownstreamConfig.Agent.SERVICE_INSTANCE_ID = 0;

        ServiceManager.INSTANCE.shutdown();

    }



    @Test

    public void testSpanLimit() {

        final boolean[] dataReceived = {false};

        TracingContext.ListenerManager.add(new TracingContextListener() {

            @Override public void afterFinished(TraceSegment traceSegment) {

                dataReceived[0] = true;

            }

        });

        TracingContext tracingContext = new TracingContext();

        AbstractSpan span = tracingContext.createEntrySpan("/url");



        for (int i = 0; i < 10; i++) {

            AbstractSpan localSpan = tracingContext.createLocalSpan("/java-bean");



            for (int j = 0; j < 100; j++) {

                AbstractSpan exitSpan = tracingContext.createExitSpan("/redis","localhost");

                tracingContext.stopSpan(exitSpan);

            }



            tracingContext.stopSpan(localSpan);

        }



        tracingContext.stopSpan(span);





        Assert.assertTrue(dataReceived[0]);

    }

}