WebSocket начало

This commit is contained in:
Gravit 2018-09-17 16:30:19 +07:00
parent cf3c186eb7
commit 96507dfee3
2 changed files with 88 additions and 0 deletions

View file

@ -6,4 +6,5 @@
compileOnly 'org.javassist:javassist:3.23.1-GA'
compile files('../compat/authlib/authlib-clean.jar')
compile 'com.eclipsesource.minimal-json:minimal-json:0.9.4'
compile group: 'org.eclipse.jetty.websocket', name: 'websocket-client', version: '9.4.12.v20180830'
}

View file

@ -0,0 +1,87 @@
package ru.gravit.utils.websocket;
import java.io.ByteArrayInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.nio.ByteBuffer;
import java.util.concurrent.*;
import jdk.nashorn.internal.ir.RuntimeNode;
import org.eclipse.jetty.websocket.api.Session;
import org.eclipse.jetty.websocket.api.StatusCode;
import org.eclipse.jetty.websocket.api.annotations.OnWebSocketClose;
import org.eclipse.jetty.websocket.api.annotations.OnWebSocketConnect;
import org.eclipse.jetty.websocket.api.annotations.OnWebSocketMessage;
import org.eclipse.jetty.websocket.api.annotations.WebSocket;
import ru.gravit.launcher.Launcher;
import ru.gravit.launcher.serialize.HInput;
import ru.gravit.utils.helper.IOHelper;
/**
* Basic Echo Client Socket
*/
@WebSocket(maxTextMessageSize = 64 * 1024)
public class WebSocketClient
{
private final CountDownLatch closeLatch;
@SuppressWarnings("unused")
private Session session;
public WebSocketClient()
{
this.closeLatch = new CountDownLatch(1);
}
public boolean awaitClose(int duration, TimeUnit unit) throws InterruptedException
{
return this.closeLatch.await(duration,unit);
}
@OnWebSocketClose
public void onClose(int statusCode, String reason)
{
System.out.printf("Connection closed: %d - %s%n",statusCode,reason);
this.session = null;
this.closeLatch.countDown(); // trigger latch
}
@OnWebSocketConnect
public void onConnect(Session session)
{
System.out.printf("Got connect: %s%n",session);
this.session = session;
try
{
Future<Void> fut;
fut = session.getRemote().sendStringByFuture("Hello");
fut.get(2,TimeUnit.SECONDS); // wait for send to complete.
fut = session.getRemote().sendStringByFuture("Thanks for the conversation.");
fut.get(2,TimeUnit.SECONDS); // wait for send to complete.
session.close(StatusCode.NORMAL,"I'm done");
}
catch (Throwable t)
{
t.printStackTrace();
}
}
public void request(ByteBuffer buffer) throws InterruptedException, ExecutionException, TimeoutException {
Future<Void> fut;
fut = session.getRemote().sendBytesByFuture(buffer);
fut.get(2,TimeUnit.SECONDS);
}
@OnWebSocketMessage
public void onMessage(String msg) throws IOException {
byte[] bytes = msg.getBytes();
InputStream stream = new ByteArrayInputStream(bytes);
HInput input = new HInput(stream);
long handshake = input.readLong();
if(handshake != Launcher.PROTOCOL_MAGIC)
{
throw new IOException("INVALID HANDSHAKE");
}
int type = input.readInt();
System.out.printf("Got msg: %l %d",handshake,type);
}
}