begin working on HTTP tracker server. parse config.json for tracker

endpoints and create appropriate protocol servers
This commit is contained in:
mykola2312 2024-11-03 00:07:03 +02:00
parent 067f10ff69
commit 59cf29c2ce
7 changed files with 189 additions and 0 deletions

View file

@ -1,9 +1,15 @@
package com.mykola2312.retracker;
import java.util.LinkedList;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import com.mykola2312.retracker.config.Config;
import com.mykola2312.retracker.tracker.Tracker;
import com.mykola2312.retracker.tracker.TrackerEndpoint;
import com.mykola2312.retracker.tracker.TrackerHTTPServer;
import com.mykola2312.retracker.tracker.TrackerServer;
/* Class responsible for managing running server threads
* and repeating timer-based tasks
@ -13,7 +19,81 @@ public class Application {
private Config config;
private Tracker tracker;
private LinkedList<TrackerServer> servers;
public Application(Config config) {
this.config = config;
}
private class TrackerServerFactory {
private Tracker tracker;
private String endpointUrl;
public TrackerServerFactory(Tracker tracker, String endpointUrl) {
this.tracker = tracker;
this.endpointUrl = endpointUrl;
}
public TrackerServer create() throws Exception {
TrackerEndpoint endpoint = TrackerEndpoint.parseEndpoint(endpointUrl);
switch (endpoint.getProtocol()) {
case HTTP: return new TrackerHTTPServer(tracker, endpoint);
default:
throw new UnsupportedOperationException("protocol not implemented yet " + endpoint.getProtocol().toString());
}
}
}
public void init() {
// create tracker
tracker = new Tracker();
// create endpoint servers
servers = new LinkedList<TrackerServer>();
for (String endpointUrl : config.tracker.endpoints) {
try {
TrackerServer server = new TrackerServerFactory(tracker, endpointUrl)
.create();
servers.add(server);
} catch (Exception e) {
logger.error("failed to create endpoint {}: {}", endpointUrl, e.toString());
}
}
}
public void start() {
// start servers
for (TrackerServer server : servers) {
try {
server.start();
} catch (Exception e) {
final String endpointString = server.getEndpoint().getAddress().toString();
logger.error("failed to start server {}: {}", endpointString, e.toString());
}
}
}
public void stop() {
// stop servers
for (TrackerServer server : servers) {
try {
server.stop();
} catch (Exception e) {
final String endpointString = server.getEndpoint().getAddress().toString();
logger.error("failed to stop server {}: {}", endpointString, e.toString());
}
}
}
public void mainLoop() {
// TODO: only for debugging purposes
while (true) {
try {
Thread.sleep(1000);
} catch (InterruptedException e) {
logger.error("main thread interrupted: {}", e.toString());
}
}
}
}

View file

@ -85,7 +85,11 @@ public class Main {
}
Application app = new Application(config);
app.init();
app.start();
log.info("retracker started!");
app.mainLoop();
}
}

View file

@ -2,10 +2,17 @@ package com.mykola2312.retracker.config;
import java.io.File;
import java.io.IOException;
import java.util.List;
import com.fasterxml.jackson.databind.ObjectMapper;
public class Config {
public static class TrackerConfig {
public List<String> endpoints;
}
public TrackerConfig tracker;
public static Config loadConfig(String configPath) throws IOException {
return new ObjectMapper()
.readerFor(Config.class)

View file

@ -0,0 +1,77 @@
package com.mykola2312.retracker.tracker;
import java.io.IOException;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import com.sun.net.httpserver.HttpExchange;
import com.sun.net.httpserver.HttpHandler;
import com.sun.net.httpserver.HttpServer;
public class TrackerHTTPServer implements TrackerServer {
private static final Logger logger = LoggerFactory.getLogger(TrackerHTTPServer.class);
private Tracker tracker;
private TrackerEndpoint endpoint;
private HttpServer httpServer = null;
public TrackerHTTPServer(Tracker tracker, TrackerEndpoint endpoint) {
this.tracker = tracker;
this.endpoint = endpoint;
}
private class AnnounceHTTPHandler implements HttpHandler {
private static final Logger logger = LoggerFactory.getLogger(AnnounceHTTPHandler.class);
private Tracker tracker;
public AnnounceHTTPHandler(Tracker tracker) {
this.tracker = tracker;
}
@Override
public void handle(HttpExchange exchange) throws IOException {
if (exchange.getRequestMethod().equals("GET")) {
handleAnnounce(exchange);
}
}
private void handleAnnounce(HttpExchange exchange) throws IOException {
// here we need parse URI parameters and pass them to PeerLocalData
logger.info(exchange.getRequestURI().toString());
}
}
@Override
public void start() {
try {
httpServer = HttpServer.create(endpoint.getAddress(), 0);
AnnounceHTTPHandler handler = new AnnounceHTTPHandler(getTracker());
httpServer.createContext("/", handler);
httpServer.start();
} catch (IOException e) {
logger.error("HttpServer: {}", e.toString());
}
}
@Override
public void stop() {
if (httpServer != null) {
httpServer.stop(0);
httpServer = null;
}
}
@Override
public Tracker getTracker() {
return tracker;
}
@Override
public TrackerEndpoint getEndpoint() {
return endpoint;
}
}

View file

@ -0,0 +1,7 @@
package com.mykola2312.retracker.tracker;
public interface TrackerJob {
void start();
void stop();
Tracker getTracker();
}

View file

@ -11,4 +11,13 @@ public enum TrackerProtocol {
default: return 0;
}
}
@Override
public String toString() {
switch (this) {
case HTTP: return "http";
case UDP: return "udp";
default: return "<unknown>";
}
}
}

View file

@ -0,0 +1,5 @@
package com.mykola2312.retracker.tracker;
public interface TrackerServer extends TrackerJob {
TrackerEndpoint getEndpoint();
}