EchoServer.java

/*******************************************************************************
 * Copyright (c) 2004, 2013 Steve Flasby
 * All rights reserved.
 * Redistribution and use in source and binary forms, with or without modification,
 * are permitted provided that the following conditions are met:
 * <ul>
 *     <li>Redistributions of source code must retain the above copyright notice,
 *         this list of conditions and the following disclaimer.</li>
 *     <li>Redistributions in binary form must reproduce the above copyright notice,
 *         this list of conditions and the following disclaimer in the documentation
 *         and/or other materials provided with the distribution.</li>
 * </ul>
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
 * IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT,
 * INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
 * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
 * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE
 * OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
 * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 ******************************************************************************/
package org.flasby.net;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.net.Socket;
import java.nio.charset.StandardCharsets;
import org.flasby.cli.CommandLine;
import org.flasby.cli.IntegerOption;
import org.flasby.cli.InvalidOptionFormatException;
import org.flasby.cli.MissingRequiredOptionException;
import org.flasby.cli.Options;
import org.flasby.cli.UnknownArgumentException;
import org.flasby.cli.impl.CommandLineImpl;
import org.flasby.server.Instance;
import org.flasby.server.Reporter;
import org.flasby.server.SimpleServer;
import org.flasby.util.Message;

public class EchoServer {

  public static void main(String[] args) {
    Options o = new Options(new Message(), "org.flasby.");
    CommandLine l;
    Integer v;
    try {
      l = new CommandLineImpl(o, args);
      IntegerOption bd = new IntegerOption(o.create("port", "1234"));
      v = bd.getIntegerValue(l);
    } catch (UnknownArgumentException
        | MissingRequiredOptionException
        | InvalidOptionFormatException e) {
      System.out.println(e.getLocalizedMessage());
      return;
    }

    new SimpleServer(v) {
      @Override
      protected Instance create(final Socket s, final Reporter reporter) {
        return new Instance(s, getReporter()) {
          @Override
          public void start() throws StartFailedException {
            try (BufferedReader reader =
                new BufferedReader(
                    new InputStreamReader(s.getInputStream(), StandardCharsets.UTF_8))) {
              String in = reader.readLine();
              if (in != null) {
                s.getOutputStream().write(in.getBytes(StandardCharsets.UTF_8));
              }
            } catch (IOException e) {
              throw new StartFailedException(e);
            }
            stop();
          }
        };
      }
    }.start();
  }
}