60 lines
1.6 KiB
Java
60 lines
1.6 KiB
Java
package io.github.ehlxr.did;
|
||
|
||
import io.github.ehlxr.did.client.SdkClient;
|
||
import io.github.ehlxr.did.common.SdkProto;
|
||
import org.junit.After;
|
||
import org.junit.Before;
|
||
import org.junit.Test;
|
||
|
||
import java.util.concurrent.CountDownLatch;
|
||
import java.util.concurrent.TimeUnit;
|
||
|
||
/**
|
||
* @author ehlxr
|
||
*/
|
||
public class DidSdkTest {
|
||
private static final int NUM = 10;
|
||
SdkClient client;
|
||
|
||
@Before
|
||
public void init() {
|
||
client = new SdkClient("127.0.0.1", 16831, 5000);
|
||
client.start();
|
||
}
|
||
|
||
@After
|
||
public void destroy() {
|
||
client.shutdown();
|
||
}
|
||
|
||
@Test
|
||
public void didSdkTest() throws Exception {
|
||
// 测试同步请求，关注rqid是否对应
|
||
for (int i = 0; i < NUM; i++) {
|
||
SdkProto resultProto = client.invokeSync();
|
||
System.out.println(i + " resultProto: " + resultProto);
|
||
}
|
||
System.out.println("invokeync test finish");
|
||
|
||
// 测试异步请求，关注rqid是否对应
|
||
final CountDownLatch countDownLatch = new CountDownLatch(NUM);
|
||
for (int i = 0; i < NUM; i++) {
|
||
final int finalI = i;
|
||
client.invokeAsync(responseFuture -> {
|
||
countDownLatch.countDown();
|
||
System.out.println(finalI + " resultProto: " + responseFuture.getSdkProto());
|
||
});
|
||
}
|
||
countDownLatch.await(10, TimeUnit.SECONDS);
|
||
System.out.println("invokeAsync test finish");
|
||
}
|
||
|
||
@Test
|
||
public void testInvoke() throws Exception {
|
||
System.out.println(client.invoke());
|
||
|
||
client.setTimeoutMillis(3000);
|
||
System.out.println(client.invoke());
|
||
}
|
||
}
|