Examples with SofaRpcException com.alipay.sofa.rpc.core.exception.SofaRpcException used on opensource projects

Example 6 with SofaRpcException

use of com.alipay.sofa.rpc.core.exception.SofaRpcException in project sofa-rpc by sofastack.

the class AsyncCallbackTest method testCallbackCallerHandleException.

@Test
public void testCallbackCallerHandleException() {
    serverConfig = new ServerConfig().setPort(22223).setDaemon(false);
    // RpcServer for C
    CProvider = new ProviderConfig<HelloService>().setInterfaceId(HelloService.class.getName()).setRef(new HelloServiceImpl(0)).setServer(serverConfig);
    CProvider.export();
    // RpcClient For B invoke C
    Filter filter = new TestAsyncFilter();
    BConsumer = new ConsumerConfig<HelloService>().setInterfaceId(HelloService.class.getName()).setInvokeType(RpcConstants.INVOKER_TYPE_CALLBACK).setTimeout(3000).setFilterRef(Arrays.asList(filter)).setRejectedExecutionPolicy(RejectedExecutionPolicy.CALLER_HANDLE_EXCEPTION.name()).setDirectUrl("bolt://127.0.0.1:22223");
    HelloService helloService = BConsumer.refer();
    int maxsize = RpcConfigs.getIntValue(RpcOptions.ASYNC_POOL_MAX);
    int queuesize = RpcConfigs.getIntValue(RpcOptions.ASYNC_POOL_QUEUE);
    int invokeCount = (maxsize + queuesize) * 2;
    final CountDownLatch latch = new CountDownLatch(invokeCount);
    AtomicInteger sofaExceptionCount = new AtomicInteger(0);
    SofaResponseCallback callback = new SofaResponseCallback() {

        @Override
        public void onAppResponse(Object appResponse, String methodName, RequestBase request) {
            LOGGER.info("B get result: {}", appResponse);
            try {
                Thread.sleep(100);
            } catch (Exception e) {
            }
            latch.countDown();
        }

        @Override
        public void onAppException(Throwable throwable, String methodName, RequestBase request) {
            LOGGER.info("B get app exception: {}", throwable);
            latch.countDown();
        }

        @Override
        public void onSofaException(SofaRpcException sofaException, String methodName, RequestBase request) {
            LOGGER.info("B get sofa exception: {}", sofaException);
            latch.countDown();
            sofaExceptionCount.addAndGet(1);
        }
    };
    for (int i = 0; i < invokeCount; i++) {
        RpcInvokeContext.getContext().setResponseCallback(callback);
        helloService.sayHello("" + i, 33);
    }
    try {
        latch.await(100L * invokeCount, TimeUnit.MILLISECONDS);
    } catch (InterruptedException ignore) {
    }
    // Exception callbacks are triggered by IO threads after the thread pool is full, so the total number of responses received is equal to the total number of calls
    assertEquals(0, latch.getCount());
    // The number of exception callbacks triggered by IO threads must exist
    assertTrue(sofaExceptionCount.get() > 0);
    RpcInvokeContext.removeContext();
}

Example 7 with SofaRpcException

use of com.alipay.sofa.rpc.core.exception.SofaRpcException in project sofa-rpc by sofastack.

the class AsyncCallbackTest method testCallbackDiscard.

@Test
public void testCallbackDiscard() {
    serverConfig = new ServerConfig().setPort(22225).setDaemon(false);
    // RpcServer For C
    CProvider = new ProviderConfig<HelloService>().setInterfaceId(HelloService.class.getName()).setRef(new HelloServiceImpl(0)).setServer(serverConfig);
    CProvider.export();
    // RpcClient For B invoke C
    Filter filter = new TestAsyncFilter();
    BConsumer = new ConsumerConfig<HelloService>().setInterfaceId(HelloService.class.getName()).setInvokeType(RpcConstants.INVOKER_TYPE_CALLBACK).setTimeout(3000).setFilterRef(Arrays.asList(filter)).setRejectedExecutionPolicy(RejectedExecutionPolicy.DISCARD.name()).setDirectUrl("bolt://127.0.0.1:22225");
    HelloService helloService = BConsumer.refer();
    int maxsize = RpcConfigs.getIntValue(RpcOptions.ASYNC_POOL_MAX);
    int queuesize = RpcConfigs.getIntValue(RpcOptions.ASYNC_POOL_QUEUE);
    int invokeCount = (maxsize + queuesize) * 2;
    final CountDownLatch latch = new CountDownLatch(invokeCount);
    SofaResponseCallback callback = new SofaResponseCallback() {

        @Override
        public void onAppResponse(Object appResponse, String methodName, RequestBase request) {
            LOGGER.info("B get result: {}", appResponse);
            try {
                Thread.sleep(100);
            } catch (Exception e) {
            }
            latch.countDown();
        }

        @Override
        public void onAppException(Throwable throwable, String methodName, RequestBase request) {
            LOGGER.info("B get app exception: {}", throwable);
            latch.countDown();
        }

        @Override
        public void onSofaException(SofaRpcException sofaException, String methodName, RequestBase request) {
            LOGGER.info("B get sofa exception: {}", sofaException);
            latch.countDown();
        }
    };
    for (int i = 0; i < invokeCount; i++) {
        RpcInvokeContext.getContext().setResponseCallback(callback);
        helloService.sayHello("" + i, 33);
    }
    try {
        latch.await(100L * invokeCount, TimeUnit.MILLISECONDS);
    } catch (InterruptedException ignore) {
    }
    LOGGER.info("Discard response callback: " + (invokeCount - latch.getCount()));
    // The total number of callbacks received in discard mode must be less than the total number of requests
    assertTrue(latch.getCount() > 0);
    RpcInvokeContext.removeContext();
}

Example 8 with SofaRpcException

use of com.alipay.sofa.rpc.core.exception.SofaRpcException in project sofa-rpc by sofastack.

the class AsyncCallbackTest method testAll.

@Test
public void testAll() {
    serverConfig = new ServerConfig().setPort(22220).setDaemon(false);
    // C服务的服务端
    CProvider = new ProviderConfig<HelloService>().setInterfaceId(HelloService.class.getName()).setRef(new HelloServiceImpl(1000)).setServer(serverConfig);
    CProvider.export();
    // B调C的客户端
    Filter filter = new TestAsyncFilter();
    BConsumer = new ConsumerConfig<HelloService>().setInterfaceId(HelloService.class.getName()).setInvokeType(RpcConstants.INVOKER_TYPE_CALLBACK).setTimeout(50000).setFilterRef(Arrays.asList(filter)).setDirectUrl("bolt://127.0.0.1:22220");
    HelloService helloService = BConsumer.refer();
    final CountDownLatch latch = new CountDownLatch(1);
    final String[] ret = { null };
    RpcInvokeContext.getContext().setResponseCallback(new SofaResponseCallback() {

        @Override
        public void onAppResponse(Object appResponse, String methodName, RequestBase request) {
            LOGGER.info("B get result: {}", appResponse);
            ret[0] = (String) appResponse;
            latch.countDown();
        }

        @Override
        public void onAppException(Throwable throwable, String methodName, RequestBase request) {
            LOGGER.info("B get app exception: {}", throwable);
            latch.countDown();
        }

        @Override
        public void onSofaException(SofaRpcException sofaException, String methodName, RequestBase request) {
            LOGGER.info("B get sofa exception: {}", sofaException);
            latch.countDown();
        }
    });
    String ret0 = helloService.sayHello("xxx", 22);
    // 第一次返回null
    Assert.assertNull(ret0);
    try {
        latch.await(60000, TimeUnit.MILLISECONDS);
    } catch (InterruptedException ignore) {
    }
    Assert.assertNotNull(ret[0]);
    // 过滤器生效
    assertTrue(ret[0].endsWith("append by async filter"));
    RpcInvokeContext.removeContext();
}

Example 9 with SofaRpcException

use of com.alipay.sofa.rpc.core.exception.SofaRpcException in project sofa-rpc by sofastack.

the class AsyncCallbackTest method testCallbackCallerRuns.

@Test
public void testCallbackCallerRuns() {
    serverConfig = new ServerConfig().setPort(22224).setDaemon(false);
    // RpcServer for C
    CProvider = new ProviderConfig<HelloService>().setInterfaceId(HelloService.class.getName()).setRef(new HelloServiceImpl(0)).setServer(serverConfig);
    CProvider.export();
    // RpcClient For B invoke C
    Filter filter = new TestAsyncFilter();
    BConsumer = new ConsumerConfig<HelloService>().setInterfaceId(HelloService.class.getName()).setInvokeType(RpcConstants.INVOKER_TYPE_CALLBACK).setTimeout(3000).setFilterRef(Arrays.asList(filter)).setRejectedExecutionPolicy(RejectedExecutionPolicy.CALLER_RUNS.name()).setDirectUrl("bolt://127.0.0.1:22224");
    HelloService helloService = BConsumer.refer();
    int maxsize = RpcConfigs.getIntValue(RpcOptions.ASYNC_POOL_MAX);
    int queuesize = RpcConfigs.getIntValue(RpcOptions.ASYNC_POOL_QUEUE);
    int invokeCount = (maxsize + queuesize) * 2;
    final CountDownLatch latch = new CountDownLatch(invokeCount);
    AtomicInteger runCount = new AtomicInteger(0);
    SofaResponseCallback callback = new SofaResponseCallback() {

        @Override
        public void onAppResponse(Object appResponse, String methodName, RequestBase request) {
            LOGGER.info("B get result: {}", appResponse);
            try {
                Thread.sleep(100);
            } catch (Exception e) {
            }
            runCount.addAndGet(1);
            latch.countDown();
        }

        @Override
        public void onAppException(Throwable throwable, String methodName, RequestBase request) {
            LOGGER.info("B get app exception: {}", throwable);
            latch.countDown();
        }

        @Override
        public void onSofaException(SofaRpcException sofaException, String methodName, RequestBase request) {
            LOGGER.info("B get sofa exception: {}", sofaException);
            latch.countDown();
        }
    };
    for (int i = 0; i < invokeCount; i++) {
        RpcInvokeContext.getContext().setResponseCallback(callback);
        helloService.sayHello("" + i, 33);
    }
    try {
        latch.await(100L * invokeCount, TimeUnit.MILLISECONDS);
    } catch (InterruptedException ignore) {
    }
    assertEquals(0, latch.getCount());
    // Invoke callbacks are triggered by IO threads after the thread pool is full, so the total number of normal responses received is equal to the total number of calls
    assertEquals(invokeCount, runCount.get());
    RpcInvokeContext.removeContext();
}

Example 10 with SofaRpcException

use of com.alipay.sofa.rpc.core.exception.SofaRpcException in project sofa-rpc by sofastack.

the class RpcContextTest method testAll.

@Test
public void testAll() {
    // 只有1个线程 执行
    ServerConfig serverConfig = new ServerConfig().setPort(22222).setProtocol(RpcConstants.PROTOCOL_TYPE_BOLT).setQueues(100).setCoreThreads(1).setMaxThreads(2);
    // 发布一个服务，每个请求要执行1秒
    CtxHelloServiceImpl helloServiceImpl = new CtxHelloServiceImpl();
    ProviderConfig<HelloService> providerConfig = new ProviderConfig<HelloService>().setInterfaceId(HelloService.class.getName()).setApplication(new ApplicationConfig().setAppName("test-server")).setRef(helloServiceImpl).setServer(serverConfig).setRegister(false);
    providerConfig.export();
    {
        ConsumerConfig<HelloService> consumerConfig = new ConsumerConfig<HelloService>().setInterfaceId(HelloService.class.getName()).setApplication(new ApplicationConfig().setAppName("test-client")).setDirectUrl("bolt://127.0.0.1:22222?appName=test-server").setTimeout(30000).setRegister(false);
        final HelloService helloService = consumerConfig.refer();
        String str = helloService.sayHello("xxx", 123);
        RpcServiceContext serviceContext = RpcContextManager.currentServiceContext(false);
        RpcReferenceContext referenceContext = RpcContextManager.lastReferenceContext(false);
        Assert.assertNull(serviceContext);
        Assert.assertNotNull(referenceContext);
        serviceContext = helloServiceImpl.serviceContext;
        Assert.assertNotNull(serviceContext);
        Assert.assertEquals(serviceContext.getCallerAppName(), "test-client");
        Assert.assertEquals(referenceContext.getTargetAppName(), "test-server");
        Assert.assertNotNull(referenceContext.getClientIP());
        Assert.assertTrue(referenceContext.getClientPort() > 0);
    }
    {
        final CountDownLatch latch = new CountDownLatch(1);
        final String[] ret = { null };
        ConsumerConfig<HelloService> consumerConfig2 = new ConsumerConfig<HelloService>().setInterfaceId(HelloService.class.getName()).setApplication(new ApplicationConfig().setAppName("test-client")).setDirectUrl("bolt://127.0.0.1:22222?appName=test-server").setTimeout(2000).setInvokeType("callback").setOnReturn(new SofaResponseCallback() {

            @Override
            public void onAppResponse(Object appResponse, String methodName, RequestBase request) {
                ret[0] = (String) appResponse;
                latch.countDown();
            }

            @Override
            public void onAppException(Throwable throwable, String methodName, RequestBase request) {
                latch.countDown();
            }

            @Override
            public void onSofaException(SofaRpcException sofaException, String methodName, RequestBase request) {
                latch.countDown();
            }
        }).setRegister(false);
        final HelloService helloServiceCallback = consumerConfig2.refer();
        String ret0 = helloServiceCallback.sayHello("xxx", 22);
        // 第一次返回null
        Assert.assertNull(ret0);
        RpcServiceContext serviceContext = RpcContextManager.currentServiceContext(false);
        RpcReferenceContext referenceContext = RpcContextManager.lastReferenceContext(false);
        Assert.assertNull(serviceContext);
        Assert.assertNotNull(referenceContext);
        serviceContext = helloServiceImpl.serviceContext;
        Assert.assertNotNull(serviceContext);
        Assert.assertEquals(serviceContext.getCallerAppName(), "test-client");
        Assert.assertEquals(referenceContext.getTargetAppName(), "test-server");
        Assert.assertNotNull(referenceContext.getClientIP());
        Assert.assertTrue(referenceContext.getClientPort() > 0);
        try {
            latch.await(5000, TimeUnit.MILLISECONDS);
        } catch (InterruptedException ignore) {
        }
        Assert.assertNotNull(ret[0]);
    }
}