change grpc param

This commit is contained in:
jackdaw@loafle.com 2017-04-28 11:16:33 +09:00
parent 919b4bc4be
commit f7bb1e3e20
3 changed files with 26 additions and 14 deletions

View File

@ -143,12 +143,14 @@ public class Server {
Object obj = null; Object obj = null;
for (Init in : list) { for (Init in : list) {
Crawler crawler = crawlerMap.get(in.getName().name()); Crawler crawler = crawlerMap.get(in.getName().name());
if (crawler != null) {
try { try {
obj = crawler.init(in.getData().toByteArray()); obj = crawler.init(in.getData().toByteArray());
} catch (Exception e) { } catch (Exception e) {
e.printStackTrace(); e.printStackTrace();
} }
} }
}
try { try {
Output reply = Output.newBuilder() Output reply = Output.newBuilder()

View File

@ -1,5 +1,8 @@
package com.loafle.overflow; package com.loafle.overflow;
import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.google.protobuf.ByteString;
import com.loafle.overflow.crawler.Crawler; import com.loafle.overflow.crawler.Crawler;
import com.loafle.overflow.crawler.config.Config; import com.loafle.overflow.crawler.config.Config;
import com.loafle.overflow.rpc.*; import com.loafle.overflow.rpc.*;
@ -11,6 +14,9 @@ import org.junit.Before;
import org.junit.Test; import org.junit.Test;
import java.io.*; import java.io.*;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.util.logging.Logger; import java.util.logging.Logger;
import static org.junit.Assert.assertEquals; import static org.junit.Assert.assertEquals;
@ -54,14 +60,16 @@ public class ServerTest {
return ManagedChannelBuilder.forAddress(ip,port).usePlaintext(true).build(); return ManagedChannelBuilder.forAddress(ip,port).usePlaintext(true).build();
} }
public Output init() { public Output init() throws IOException {
ConfigGrpc.ConfigBlockingStub client = newConfigClient("localhost",50052); ConfigGrpc.ConfigBlockingStub client = newConfigClient("localhost",50052);
ClassLoader classLoader = getClass().getClassLoader(); ClassLoader classLoader = getClass().getClassLoader();
Path path = Paths.get(classLoader.getResource("config/example.json").getFile());
byte[] data = Files.readAllBytes(path);
ByteString sss = ByteString.copyFrom(data);
// path is test resources/config/example.json
Init input = Init.newBuilder(). Init input = Init.newBuilder().
setPath(classLoader.getResource("config").getFile()). setData(sss).
setName(Crawlers.REDIS).build(); setName(Crawlers.REDIS).build();
InputArray arr = InputArray.newBuilder().addIn(input).build(); InputArray arr = InputArray.newBuilder().addIn(input).build();
@ -89,13 +97,15 @@ public class ServerTest {
ConfigGrpc.ConfigBlockingStub client = newConfigClient("localhost",50052); ConfigGrpc.ConfigBlockingStub client = newConfigClient("localhost",50052);
ClassLoader classLoader = getClass().getClassLoader();
// path is test resources/config/example.json // path is test resources/config/example.json
Input input = Input.newBuilder(). ClassLoader classLoader = getClass().getClassLoader();
setId("example.json"). Path path = Paths.get(classLoader.getResource("config/example.json").getFile());
setName(Crawlers.REDIS).build(); byte[] data = Files.readAllBytes(path);
ByteString sss = ByteString.copyFrom(data);
InputAdd input = InputAdd.newBuilder().
setData(sss).
setName(Crawlers.REDIS).build();
Output out = client.add(input); Output out = client.add(input);
@ -141,7 +151,7 @@ public class ServerTest {
DataGrpc.DataBlockingStub client = newDataClient("localhost",50052); DataGrpc.DataBlockingStub client = newDataClient("localhost",50052);
Input input = Input.newBuilder(). Input input = Input.newBuilder().
setId("example.json"). setId("SOEJWEOJWOEJOSDJFOASDJFOSDFO2903870928734").
setName(Crawlers.REDIS).build(); setName(Crawlers.REDIS).build();
Output out = client.get(input); Output out = client.get(input);

View File

@ -35,7 +35,7 @@ public class TestClient {
/** Say hello to server. */ /** Say hello to server. */
public void add() { public void add() {
Input request = Input.newBuilder().setId("test111").setName(Crawlers.SNMP).build(); InputAdd request = InputAdd.newBuilder().setData(null).setName(Crawlers.SNMP).build();
Output response; Output response;
try { try {
response = configStub.add(request); response = configStub.add(request);