org.nustaq.kontraktor.services.rlserver.RLJsonServer Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of service-support Show documentation
service registry
There is a newer version: 5.2.0
package org.nustaq.kontraktor.services.rlserver;

import org.nustaq.kontraktor.Actor;
import org.nustaq.kontraktor.IPromise;
import org.nustaq.kontraktor.Scheduler;
import org.nustaq.kontraktor.annotations.CallerSideMethod;
import org.nustaq.kontraktor.annotations.Local;
import org.nustaq.kontraktor.apputil.*;
import org.nustaq.kontraktor.impl.SimpleScheduler;
import org.nustaq.kontraktor.remoting.encoding.Coding;
import org.nustaq.kontraktor.remoting.encoding.SerializerType;
import org.nustaq.kontraktor.remoting.http.undertow.Http4K;
import org.nustaq.kontraktor.services.rlclient.DataClient;
import org.nustaq.kontraktor.util.Log;
import org.nustaq.reallive.messages.*;
import org.nustaq.reallive.records.MapRecord;

import java.io.File;
import java.util.Random;
import java.util.concurrent.TimeUnit;
import java.util.stream.IntStream;

public class RLJsonServer extends Actor {

    public static final Class[] CLAZZES = new Class[]{
        LoginData.class,
        MapRecord.class,
        AddMessage.class,
        UpdateMessage.class,
        RemoveMessage.class,
        QueryDoneMessage.class,
        SessionEvent.class,
        RLJsonAuthResult.class,
        Diff.class,
    };

    public static SimpleRLConfig Cfg() {
        synchronized (RLJsonServer.class) {
            if ( cfg == null )
                cfg = SimpleRLConfig.read();
            return cfg;
        }
    }

    private static SimpleRLConfig cfg;

    // threads to dispatch session onto
    protected Scheduler clientThreads[];
    protected Random rand = new Random();
    protected RLJsonServerService service;
    protected DataClient dclient;

    @Local
    public IPromise init(String args[]) {
        clientThreads = new Scheduler[Cfg().getNumSessionThreads()];
        IntStream.range(0,Cfg().getNumSessionThreads())
            .forEach( i -> clientThreads[i] = new SimpleScheduler(10000, true /*Important!*/ ));
        service = RLJsonServerService.start(args);
        service.setWebServer(self());
        dclient = service.getDClient();
        return resolve();
    }

    public IPromise authenticate(String user, String pwd ) {
        Log.Info(this,"authenticate session");
        return createSession(new Object[]{ user,pwd });
    }

    protected IPromise createSession(Object customSessionData) {
        // FIXME, check credentials
        RLJsonSession session = AsActor(
            getSessionActorClazz(customSessionData),
            // randomly distribute session actors among clientThreads
            clientThreads[rand.nextInt(clientThreads.length)]
        );
        session.init(self(),dclient,customSessionData);
        return resolve(new RLJsonAuthResult().session(session));
    }

    protected Class getSessionActorClazz(Object authData) {
        return RLJsonSession.class;
    }

    @CallerSideMethod
    protected void createServer(RLJsonServer app, Class[] CLAZZES) {
        Http4K.Build(Cfg().getBindIp(), Cfg().getBindPort())
            .httpAPI("/api", app) // could also be websocket based (see IntrinsicReactJSX github project)
            .coding(new Coding(SerializerType.JsonNoRef, CLAZZES))
            .setSessionTimeout(TimeUnit.MINUTES.toMillis(cfg.getSessionTimeoutMinutes() ))
            .buildHttpApi()
            .websocket("/ws", app)
                .coding(new Coding(SerializerType.JsonNoRef, CLAZZES))
                .buildWebsocket()
            .build();
    }

    @CallerSideMethod @Local
    public DataClient getDClient() {
        return getActor().dclient;
    }

    public static void main(String[] args) throws InterruptedException {
        Class appClazz = RLJsonServer.class;

        startUp(args, appClazz);

    }

    public static void startUp(String[] args, Class appClazz) throws InterruptedException {
        if ( ! new File("./etc").exists() ) {
            System.out.println("Please run with project working dir");
            System.exit(-1);
        }

        if ( Cfg().runDataClusterInsideWebserver ) {
            SingleProcessRLCluster.main(new String[0]);
        }

        RLJsonServer app = (RLJsonServer) AsActor(appClazz);
        app.init(args).await(60_000);

        Log.Info(appClazz,"listening on http://"+Cfg().getBindIp()+":"+Cfg().getBindPort());

        app.createServer(app, CLAZZES);
    }


}