【AKKA干货】AKKA-HTTP（JAVA版）踩坑记

private Flow<Message, Message, NotUsed> websocketFlow() {
ActorRef actor =
system.actorOf(GatewayActor.props(Globals.getGameRouter())); //gameRouter是集群路由Group
return socketFlow(actor);
}

private Flow<Message, Message, NotUsed> socketFlow(ActorRef actor) {
// 背压支撑
Source<Message, NotUsed> source = Source.<String>actorRefWithBackpressure("ack", o -> {
if (o == "complete")
return Optional.of(CompletionStrategy.draining());
else
return Optional.empty();
}, o -> Optional.empty()).map(message -> (Message) TextMessage.create(message))
.mapMaterializedValue(textMessage -> {
actor.tell(textMessage, ActorRef.noSender());
return NotUsed.getInstance();
})
// .keepAlive(Duration.ofSeconds(10), () -> TextMessage.create("ping")) //这段代码可让服务端自动向客户端发送ping
; Sink<Message, NotUsed> sink = Flow.<Message>create().to(Sink.actorRef(actor, PoisonPill.getInstance())); return Flow.fromSinkAndSource(sink, source);
}

@Override
protected Route routes() {
// TODO Auto-generated method stub
return path("gateway", () -> get(() -> handleWebSocketMessages(websocketFlow))));

public HttpsConnectionContext useHttps(ActorSystem system) {
HttpsConnectionContext https = null;
try {
// initialise the keystore
// !!! never put passwords into code !!!
final char[] password = "123456789".toCharArray(); final KeyStore ks = KeyStore.getInstance("PKCS12");
final InputStream keystore = KutaHttpApp.class.getClassLoader().getResourceAsStream("xxx-xxx-com-akka-0709085814.pfx");
if (keystore == null) {
throw new RuntimeException("Keystore required!");
}
ks.load(keystore, password);
Enumeration<String> aliases = ks.aliases();
while(aliases.hasMoreElements()) {
String next = aliases.nextElement();
logger.info(next);
java.security.Key key = ks.getKey(next, password);
logger.info("alg:{},format:{}",key.getAlgorithm(),key.getFormat());
} final KeyManagerFactory keyManagerFactory = KeyManagerFactory.getInstance("SunX509");
keyManagerFactory.init(ks, password); final TrustManagerFactory tmf = TrustManagerFactory.getInstance("SunX509");
tmf.init(ks); final SSLContext sslContext = SSLContext.getInstance("TLS");
sslContext.init(keyManagerFactory.getKeyManagers(), tmf.getTrustManagers(), new SecureRandom()); https = ConnectionContext.https(sslContext); } catch (NoSuchAlgorithmException | KeyManagementException e) {
system.log().error("Exception while configuring HTTPS.", e);
} catch (CertificateException | KeyStoreException | UnrecoverableKeyException | IOException e) {
system.log().error("Exception while ", e);
} return https;
}

final Http http = Http.get(system);
HttpsConnectionContext https = app.useHttps(system);
http.setDefaultServerHttpContext(https);
Integer sslPort = PropertyUtils.getInteger("app", "gateway.ssl.port");
http.bindAndHandle(flow, ConnectHttp.toHost(host, sslPort), materializer);

public Route RESTfulAsyncRouter() {
return path(PathMatchers.segment("RESTful"),()->{
complete("hello");
});
}

public Route RESTfulRouter() {
return path(PathMatchers.segment("RESTful"),
()-> concat(
post(()->{
return entity(Jackson.unmarshaller(JSONObject.class), json -> {
final JSONObject data = json;
final MessageDispatcher dispatcher = system.dispatchers().lookup(KSFConstants.BLOCKING_DISPATCHER);
try {
return
CompletableFuture.<Route>supplyAsync(()->{
final ActorRef RESTfull = system.actorOf(RESTfulActor.props(Globals.getHallRouter())
.withDispatcher(KSFConstants.BLOCKING_DISPATCHER));
// logger.info("当前运行线程:{}",Thread.currentThread());
CompletionStage<Optional<KutaJsonResponse>> rsp = Patterns
.ask(RESTfull, data, timeout)
.thenApply(a -> {
return Optional.of((KutaJsonResponse) a);
});
return onSuccess(() -> rsp, performed -> {
RESTfull.tell(PoisonPill.getInstance(), ActorRef.noSender());
List<HttpHeader> list = new ArrayList<>();
list.add(HttpHeader.parse("Access-Control-Allow-Origin", "*"));
list.add(HttpHeader.parse("Access-Control-Allow-Credentials", "true"));
list.add(HttpHeader.parse("Access-Control-Allow-Methods", "POST,OPTIONS"));
list.add(HttpHeader.parse("Access-Control-Expose-Headers", "Content-Type, Access-Control-Allow-Origin, Access-Control-Allow-Credentials"));
list.add(HttpHeader.parse("Access-Control-Allow-Headers", "Content-Type,Access-Token,Authorization"));
if (performed.isPresent()) {
if(performed.get().getSize() > (1024 * 50)) {
return encodeResponseWith(
Collections.singletonList(Coder.Gzip),
()->complete(StatusCodes.OK, list, performed.get(), Jackson.marshaller())
);
}
else {
return complete(StatusCodes.OK, list, performed.get(), Jackson.marshaller());
}
}
else {
return complete(StatusCodes.NOT_FOUND);
}
}).orElse(complete(StatusCodes.INTERNAL_SERVER_ERROR));
}, dispatcher).get();
} catch (InterruptedException | ExecutionException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}
return null;
});
})
, options(()->{
List<HttpHeader> list = new ArrayList<>();
list.add(HttpHeader.parse("Access-Control-Allow-Origin", "*"));
list.add(HttpHeader.parse("Access-Control-Allow-Credentials", "true"));
list.add(HttpHeader.parse("Access-Control-Allow-Methods", "POST,OPTIONS"));
list.add(HttpHeader.parse("Access-Control-Expose-Headers", "Content-Type, Access-Control-Allow-Origin, Access-Control-Allow-Credentials,Vary"));
list.add(HttpHeader.parse("Access-Control-Allow-Headers", "Content-Type,Access-Token,Authorization"));
return respondWithHeaders(list,()-> complete(""));
})
)
);
}

public Route staticResourceRouter() {
return path(PathMatchers.remaining(), remain -> get(()-> {
if(remain.endsWith(".properties") || remain.endsWith(".xml") || remain.endsWith(".conf")) {
return complete(StatusCodes.UNAUTHORIZED);
}
return getFromResource(remain);
}));
}

//我自己实现的HttpApp
KutaHttpApp app = new KutaHttpApp(system);
final Http http = Http.get(system);
final ActorMaterializer materializer = ActorMaterializer.create(system);
final Flow<HttpRequest, HttpResponse, NotUsed> flow = app.routes().flow(system, materializer);
//先绑定http
http.bindAndHandle(flow, ConnectHttp.toHost(host, port), materializer);
boolean useHttps = Boolean.parseBoolean(PropertyUtils.getProperty("app", "gateway.usessl"));
if (useHttps) {
HttpsConnectionContext https = app.useHttps(system);
http.setDefaultServerHttpContext(https);
Integer sslPort = PropertyUtils.getInteger("app", "gateway.ssl.port");
http.bindAndHandle(flow, ConnectHttp.toHost(host, sslPort), materializer);
logger.info("启动ssl服务.host:{},port:{}",host,sslPort);
}

public Route RESTfulAsyncRouter() {
return path(PathMatchers.segment("RESTful"),()->{
return entity(Jackson.unmarshaller(JSONObject.class), json -> {
final Set<HttpHeader> headers = new HashSet<>();
return completeWith(Marshaller.entityToOKResponse(headers, Jackson.<KutaJsonResponse>marshaller()), f->{
system.actorOf(RESTfulAsyncActor.props(json, f));
});
});
});
}