Added ping and pong methods

Author: LatvianModder

src/main/java/dev/latvian/apps/tinyserver/ws/Frame.java

@@ -19,6 +19,10 @@ public static Frame binary(byte[] buffer) {
 		return simple(Opcode.BINARY, null, buffer);
 	}
 
+	public static Frame ping(byte[] buffer) {
+		return simple(Opcode.PING, null, buffer);
+	}
+
 	public Frame appendTo(@Nullable Frame previous) {
 		if (previous != null) {
 			var newLen = previous.info.size() + info.size();

src/main/java/dev/latvian/apps/tinyserver/ws/RXThread.java

@@ -48,6 +48,7 @@ public void run() {
 						session.onPing(frame.payload());
 						session.send(new Frame(new FrameInfo(Opcode.PONG, info.mask(), info.fin(), info.rsv1(), info.rsv2(), info.rsv3(), info.maskKey(), info.size()), frame.payload()));
 					}
+					case PONG -> session.onPong(frame.payload());
 					case CLOSING -> {
 						if (info.size() > 0) {
 							var payload = ByteBuffer.wrap(frame.payload());

src/main/java/dev/latvian/apps/tinyserver/ws/WSHandler.java

@@ -70,4 +70,28 @@ default void broadcastBinary(Supplier<byte[]> payload) {
 			}
 		}
 	}
+
+	default void broadcastPing(byte[] payload) {
+		var s = sessions().values();
+
+		if (!s.isEmpty()) {
+			var p = Frame.ping(payload);
+
+			for (var session : s) {
+				session.send(p);
+			}
+		}
+	}
+
+	default void broadcastPing(Supplier<byte[]> payload) {
+		var s = sessions().values();
+
+		if (!s.isEmpty()) {
+			var p = Frame.ping(payload.get());
+
+			for (var session : s) {
+				session.send(p);
+			}
+		}
+	}
 }

src/main/java/dev/latvian/apps/tinyserver/ws/WSSession.java

@@ -58,6 +58,10 @@ public final void sendBinary(byte[] payload) {
 		send(Frame.binary(payload));
 	}
 
+	public final void sendPing(byte[] payload) {
+		send(Frame.ping(payload));
+	}
+
 	public void onOpen(REQ req) {
 	}
 
@@ -76,6 +80,9 @@ public void onBinaryMessage(byte[] message) {
 	public void onPing(byte[] payload) {
 	}
 
+	public void onPong(byte[] payload) {
+	}
+
 	public final void close(WSCloseStatus status, String reason) {
 		txThread.remoteClosed = false;
 		txThread.closeReason = new StatusCode(status.statusCode.code(), reason);

src/test/java/dev/latvian/apps/tinyserver/test/TestWSSession.java

@@ -26,4 +26,9 @@ public void onTextMessage(String message) {
 	public void onPing(byte[] payload) {
 		System.out.println("WS Ping: " + new String(payload, StandardCharsets.UTF_8));
 	}
+
+	@Override
+	public void onPong(byte[] payload) {
+		System.out.println("WS Pong: " + new String(payload, StandardCharsets.UTF_8));
+	}
 }