60 lines
2 KiB
Java
60 lines
2 KiB
Java
/*
|
|
* Copyright (C) 2009-2015 Typesafe Inc. <http://www.typesafe.com>
|
|
*/
|
|
|
|
package docs.http.javadsl.server;
|
|
|
|
import akka.http.javadsl.server.Route;
|
|
|
|
import akka.japi.JavaPartialFunction;
|
|
|
|
import akka.stream.javadsl.Flow;
|
|
import akka.stream.javadsl.Source;
|
|
|
|
import akka.http.javadsl.model.ws.Message;
|
|
import akka.http.javadsl.model.ws.TextMessage;
|
|
|
|
import akka.http.javadsl.server.HttpApp;
|
|
|
|
public class WebsocketRoutingExample extends HttpApp {
|
|
//#websocket-route
|
|
@Override
|
|
public Route createRoute() {
|
|
return
|
|
path("greeter").route(
|
|
handleWebsocketMessages(greeter())
|
|
);
|
|
}
|
|
//#websocket-route
|
|
|
|
/**
|
|
* A handler that treats incoming messages as a name,
|
|
* and responds with a greeting to that name
|
|
*/
|
|
public static Flow<Message, Message, Object> greeter() {
|
|
return
|
|
upcastMaterializerToObject(Flow.<Message>create())
|
|
.collect(new JavaPartialFunction<Message, Message>() {
|
|
@Override
|
|
public Message apply(Message msg, boolean isCheck) throws Exception {
|
|
if (isCheck)
|
|
if (msg.isText()) return null;
|
|
else throw noMatch();
|
|
else
|
|
return handleTextMessage(msg.asTextMessage());
|
|
}
|
|
});
|
|
}
|
|
public static TextMessage handleTextMessage(TextMessage msg) {
|
|
if (msg.isStrict()) // optimization that directly creates a simple response...
|
|
return TextMessage.create("Hello "+msg.getStrictText());
|
|
else // ... this would suffice to handle all text messages in a streaming fashion
|
|
return TextMessage.create(Source.single("Hello ").concat(msg.getStreamedText()));
|
|
}
|
|
|
|
// needed because of #18028, see https://github.com/akka/akka/issues/18028
|
|
@SuppressWarnings("unchecked")
|
|
public static <T, U> Flow<T, U, Object> upcastMaterializerToObject(Flow< T, U, ?> flow) {
|
|
return (Flow<T, U, Object>) flow;
|
|
}
|
|
}
|