SupportServer.java

package com.workshare.msnos.integration;

import com.workshare.msnos.core.Cloud;
import com.workshare.msnos.core.Cloud.Listener;
import com.workshare.msnos.core.LocalAgent;
import com.workshare.msnos.core.Message;
import com.workshare.msnos.core.Message.Payload;
import com.workshare.msnos.core.MsnosException;
import com.workshare.msnos.core.payloads.GenericPayload;
import com.workshare.msnos.soup.Shorteners;
import com.workshare.msnos.soup.json.Json;

import java.io.File;
import java.io.FileWriter;
import java.io.IOException;
import java.io.PrintWriter;
import java.util.Arrays;
import java.util.UUID;

public class SupportServer implements IntegrationActor {

    protected LocalAgent masterAgent;
    protected Cloud masterCloud;

    protected LocalAgent testAgent;
    protected Cloud testCloud;

    static {
        System.setProperty("com.ws.nsnos.time.local", "true");
    }

    public SupportServer() throws IOException {
        bootstrap();

        masterCloud.addListener(new Listener() {
            @Override
            public void onMessage(Message message) {
                if (message.getType() == Message.Type.APP) {
                    final Payload payload = message.getData();
                    if (payload instanceof GenericPayload)
                        try {
                            processCommand(CommandPayload.create((GenericPayload) payload), message);
                        } catch (MsnosException e) {
                            log("Unexpected exception while processing a command from the test client! " + e.getMessage());
                        }
                }
            }
        });

    }

    public void bootstrap() throws MsnosException {
        masterAgent = new LocalAgent(UUID.randomUUID());
        masterCloud = new Cloud(MASTER_CLOUD_UUID);
        masterAgent.join(masterCloud);

        testCloud = new Cloud(GENERIC_CLOUD_UUID);
        testAgent = new LocalAgent(UUID.randomUUID());
    }

    protected void processCommand(CommandPayload payload, Message message) throws MsnosException {
        log("Received message "+Shorteners.shorten(message.getUuid())+": " + Json.toJsonString(payload));
        switch (payload.getCommand()) {
            case INIT:
                log("");
                log("--------------------------------------------");
                break;
                
            case TERM:
                log("");
                break;
                
            case AGENT_JOIN:
                if (testAgent.getCloud() == null) {
                    log("Agent joining the cloud");
                    testAgent.join(testCloud);
                } else {
                    log("Agent already in the cloud :)");
                }
                break;

            case AGENT_LEAVE:
                if (testAgent.getCloud() != null) {
                    log("Agent leaving the cloud...");
                    testAgent.leave();
                } else {
                    log("Agent already out of the cloud :)");
                }
                break;

            case SELF_KILL:
                log("Suiciding ;(");
                System.exit(-1);
                break;
        }

    }

    private void run() throws InterruptedException {
        final int total = 100;
        for (int i = 0; i < total; i++) {
            log("Integration test server is alive :) step " + (i + 1) + " out of " + total);
            Thread.sleep(5000L);
        }

        log("Goodbye!");
    }

    public static void main(String[] args) throws Exception {
        boolean okay = false;

        if (args.length == 1) {
            if ("now".equalsIgnoreCase(args[0])) {
                okay = true;
                runNow();
            } else if ("fork".equalsIgnoreCase(args[0])) {
                okay = true;
                truncateLogfile();
                runFork();
            }
        }

        if (!okay) {
            log("Invalid parameters specified: " + Arrays.asList(args));
            log("Please use either \"fork\" or \"now\"");
        }

        log("");
    }

    private static void runNow() throws Exception {
        log("Running in process");
        new SupportServer().run();
    }

    private static void runFork() throws Exception {
        log("Running forked");
        String javacmd = new File(new File(System.getProperty("java.home"), "bin"), "java").getAbsolutePath();

        String[] args = new String[]{javacmd, "-cp", System.getProperty("java.class.path"), SupportServer.class.getCanonicalName(), "now"};
        log("args: " + Arrays.asList(args));
        Runtime.getRuntime().exec(args);
    }

    private static void log(String what) {
        try {
            final File logfile = logfile();
            PrintWriter log = new PrintWriter(new FileWriter(logfile, true));
            try {
                log.println(what);
            } finally {
                log.close();
            }
        } catch (IOException ex) {
            ex.printStackTrace();
        }
    }

    private static void truncateLogfile() {
        try {
            new FileWriter(logfile(), false).close();
        } catch (Exception ignore) {
        }
    }

    private static File logfile() {
        return new File(System.getProperty("java.io.tmpdir"), "tmp.log");
    }

}