package org.example.client;
|
|
|
import okhttp3.Call;
|
import okhttp3.WebSocket;
|
import org.example.parser.Parser;
|
|
import java.net.URI;
|
import java.net.URISyntaxException;
|
import java.util.concurrent.ConcurrentHashMap;
|
import java.util.logging.Level;
|
import java.util.logging.Logger;
|
|
|
public class IO {
|
|
private static final Logger logger = Logger.getLogger(IO.class.getName());
|
|
private static final ConcurrentHashMap<String, Manager> managers = new ConcurrentHashMap<>();
|
|
/**
|
* Protocol version.
|
*/
|
public static int protocol = Parser.protocol;
|
|
public static void setDefaultOkHttpWebSocketFactory(WebSocket.Factory factory) {
|
Manager.defaultWebSocketFactory = factory;
|
}
|
|
public static void setDefaultOkHttpCallFactory(Call.Factory factory) {
|
Manager.defaultCallFactory = factory;
|
}
|
|
private IO() {}
|
|
public static Socket socket(String uri) throws URISyntaxException {
|
return socket(uri, null);
|
}
|
|
public static Socket socket(String uri, Options opts) throws URISyntaxException {
|
return socket(new URI(uri), opts);
|
}
|
|
public static Socket socket(URI uri) {
|
return socket(uri, null);
|
}
|
|
/**
|
* Initializes a {@link Socket} from an existing {@link Manager} for multiplexing.
|
*
|
* @param uri uri to connect.
|
* @param opts options for socket.
|
* @return {@link Socket} instance.
|
*/
|
public static Socket socket(URI uri, Options opts) {
|
if (opts == null) {
|
opts = new Options();
|
}
|
|
Url.ParsedURI parsed = Url.parse(uri);
|
URI source = parsed.uri;
|
String id = parsed.id;
|
|
boolean sameNamespace = managers.containsKey(id)
|
&& managers.get(id).nsps.containsKey(source.getPath());
|
boolean newConnection = opts.forceNew || !opts.multiplex || sameNamespace;
|
Manager io;
|
|
String query = source.getQuery();
|
if (query != null && (opts.query == null || opts.query.isEmpty())) {
|
opts.query = query;
|
}
|
|
if (newConnection) {
|
if (logger.isLoggable(Level.FINE)) {
|
logger.fine(String.format("ignoring socket cache for %s", source));
|
}
|
io = new Manager(source, opts);
|
} else {
|
if (!managers.containsKey(id)) {
|
if (logger.isLoggable(Level.FINE)) {
|
logger.fine(String.format("new io instance for %s", source));
|
}
|
managers.putIfAbsent(id, new Manager(source, opts));
|
}
|
io = managers.get(id);
|
}
|
|
return io.socket(source.getPath(), opts);
|
}
|
|
|
public static class Options extends Manager.Options {
|
|
public boolean forceNew;
|
|
/**
|
* Whether to enable multiplexing. Default is true.
|
*/
|
public boolean multiplex = true;
|
|
/**
|
* <p>
|
* Retrieve new builder class that helps creating socket option as builder pattern.
|
* This method returns exactly same result as :
|
* </p>
|
* <code>
|
* SocketOptionBuilder builder = SocketOptionBuilder.builder();
|
* </code>
|
*
|
* @return builder class that helps creating socket option as builder pattern.
|
* @see SocketOptionBuilder#builder()
|
*/
|
public static SocketOptionBuilder builder() {
|
return SocketOptionBuilder.builder();
|
}
|
}
|
}
|
