source: java/main/src/main/java/com/framsticks/communication/ServerSideManagedConnection.java @ 193

package com.framsticks.communication;

import com.framsticks.communication.queries.*;
import com.framsticks.util.FramsticksException;
import com.framsticks.util.lang.Holder;
import com.framsticks.util.lang.Pair;
import com.framsticks.util.lang.Strings;

import org.apache.logging.log4j.Logger;
import org.apache.logging.log4j.LogManager;

import java.net.Socket;
import com.framsticks.util.dispatching.RunAt;

/**
 * @author Piotr Sniegowski
 */
public class ServerSideManagedConnection extends ManagedConnection {

        private final static Logger log = LogManager.getLogger(ServerSideManagedConnection.class);

        protected final RequestHandler requestHandler;

        public ServerSideManagedConnection(Socket socket, RequestHandler requestHandler) {
                setAddress(new Address(socket.getInetAddress().getHostAddress(), socket.getPort()));
                setDescription("server connection");
                this.socket = socket;
                this.requestHandler = requestHandler;
                // socket.setSoTimeout(500);
                setupStreams();
        }



        @Override
        protected void receiverThreadRoutine() {

                processInputBatchesUntilClosed();
        }

        protected void handleRequest(Request request, ServerSideResponseFuture responseCallback) {
                if (request instanceof ApplicationRequest) {
                        requestHandler.handle((ApplicationRequest) request, responseCallback);
                        return;
                }
                if (request instanceof ProtocolRequest) {
                        if (request instanceof VersionRequest) {
                                responseCallback.pass(new Response(true, null));
                                return;
                        }
                        if (request instanceof UseRequest) {
                                String feature = ((UseRequest)request).getFeature();
                                if (feature.equals("request_id")) {
                                        requestIdEnabled = true;
                                        responseCallback.pass(new Response(true, null));
                                        return;
                                }
                                // if (feature.equals("call_empty_result")) {
                                //      responseCallback.pass(new Response(true, null));
                                //      return;
                                // }
                                if (feature.equals("needfile_id")) {
                                        responseCallback.pass(new Response(true, null));
                                        return;
                                }
                                responseCallback.pass(new Response(false, "unknown feature: " + feature));
                                return;
                        }

                }
                log.error("unhandled request: {}", request);
                responseCallback.pass(new Response(false, "unhandled"));
        }



        protected final void respond(final Response response, final Integer id) {
                senderThread.dispatch(new RunAt<Connection>(requestHandler) {
                        @Override
                        protected void runAt() {
                                if (response.getFiles() != null) {
                                        for (File f : response.getFiles()) {
                                                putFile(f, id);
                                        }
                                }
                                StringBuilder statusLine = new StringBuilder();
                                statusLine.append(response.getOk() ? "ok" : "error").append(idToString(id));
                                if (Strings.notEmpty(response.getComment())) {
                                        Request.quoteValue(statusLine.append(" "), response.getComment());
                                }
                                putLine(statusLine.toString());
                                flushOut();
                        }
                });

        }


        protected void processNextInputBatch() {
                final Holder<Integer> id = new Holder<>();
                final String line = getLine();
                try {
                        Pair<CharSequence, CharSequence> command = Request.takeIdentifier(line);
                        final Pair<Integer, CharSequence> rest = takeRequestId(command.second);
                        id.set(rest.first);

                        final Request request = Request.parse(command.first, rest.second);

                        if (log.isTraceEnabled()) {
                                log.trace("read request: {}", request);
                        }

                        handleRequest(request, new ServerSideResponseFuture() {
                                @Override
                                protected void result(Response response) {
                                        respond(response, rest.first);
                                }
                        });
                } catch (FramsticksException e) {
                        e.arg("id", id.get()).arg("line", line);
                        log.error("error: ", e);
                        respond(new Response(false, "invalid input: " + e.getMsg()), id.get());
                        return;
                }

        }
}