pekko/akka-docs/src/test/java/jdocs/io/UdpConnectedDocTest.java
2022-11-30 16:45:15 +01:00

96 lines
2.9 KiB
Java

/*
* Copyright (C) 2009-2022 Lightbend Inc. <https://www.lightbend.com>
*/
package jdocs.io;
import org.apache.pekko.japi.pf.ReceiveBuilder;
import org.junit.Test;
import org.apache.pekko.actor.ActorSystem;
import org.apache.pekko.actor.AbstractActor;
// #imports
import java.net.InetSocketAddress;
import java.util.ArrayList;
import java.util.List;
import org.apache.pekko.actor.ActorRef;
import org.apache.pekko.io.Inet;
import org.apache.pekko.io.UdpConnected;
import org.apache.pekko.io.UdpConnectedMessage;
import org.apache.pekko.io.UdpSO;
import org.apache.pekko.util.ByteString;
import static org.apache.pekko.util.ByteString.emptyByteString;
// #imports
public class UdpConnectedDocTest {
public static class Demo extends AbstractActor {
ActorRef connectionActor = null;
ActorRef handler = getSelf();
ActorSystem system = getContext().getSystem();
@Override
public Receive createReceive() {
ReceiveBuilder builder = receiveBuilder();
builder.matchEquals(
"connect",
message -> {
// #manager
final ActorRef udp = UdpConnected.get(system).manager();
// #manager
// #connect
final InetSocketAddress remoteAddr = new InetSocketAddress("127.0.0.1", 12345);
udp.tell(UdpConnectedMessage.connect(handler, remoteAddr), getSelf());
// #connect
// #connect-with-options
final InetSocketAddress localAddr = new InetSocketAddress("127.0.0.1", 1234);
final List<Inet.SocketOption> options = new ArrayList<Inet.SocketOption>();
options.add(UdpSO.broadcast(true));
udp.tell(
UdpConnectedMessage.connect(handler, remoteAddr, localAddr, options), getSelf());
// #connect-with-options
});
// #connected
builder.match(
UdpConnected.Connected.class,
conn -> {
connectionActor = getSender(); // Save the worker ref for later use
});
// #connected
// #received
builder
.match(
UdpConnected.Received.class,
recv -> {
final ByteString data = recv.data();
// and do something with the received data ...
})
.match(
UdpConnected.CommandFailed.class,
failed -> {
final UdpConnected.Command command = failed.cmd();
// react to failed connect, etc.
})
.match(
UdpConnected.Disconnected.class,
x -> {
// do something on disconnect
});
// #received
builder.matchEquals(
"send",
x -> {
ByteString data = emptyByteString();
// #send
connectionActor.tell(UdpConnectedMessage.send(data), getSelf());
// #send
});
return builder.build();
}
}
@Test
public void demonstrateConnect() {}
}