net.mms_projects.copy_it.server.Main.java Source code

Introduction

Here is the source code for net.mms_projects.copy_it.server.Main.java

Source

/*  copyit-server
 *  Copyright (C) 2013  Toon Schoenmakers
 *
 *  This program is free software: you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation, either version 3 of the License, or
 *  (at your option) any later version.
 *
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

package net.mms_projects.copy_it.server;

import io.netty.bootstrap.ServerBootstrap;
import io.netty.channel.Channel;
import io.netty.channel.ChannelFuture;
import io.netty.channel.EventLoopGroup;
import io.netty.channel.nio.NioEventLoopGroup;
import io.netty.channel.socket.nio.NioServerSocketChannel;
import net.mms_projects.copy_it.api.http.Initializer;
import net.mms_projects.copy_it.api.http.Page;
import net.mms_projects.copy_it.server.config.Config;
import net.mms_projects.copy_it.server.database.DatabasePool;
import net.mms_projects.copy_it.server.database.MySQL;
import sun.misc.Signal;
import sun.misc.SignalHandler;

import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.InputStream;

public class Main {
    public static void main(String[] args) throws Exception {
        if (args.length > 0) {
            if ("generate-pages".equals(args[0])) {
                new Config(new File((args.length > 1 ? args[1] : "copyit.config")));
                System.exit(PageGenerator.generate() ? 0 : 1);
            } else
                new Config(new File(args[0]));
        } else
            new Config(new File("copyit.config"));
        if (PageGenerator.checkPages()) {
            Messages.printWarning("Outdated pages detected! Updating them now!");
            PageGenerator.generate();
        }
        printPid();
        new DatabasePool(MySQL.class, Config.getMaxConnectionsDatabasePool());
        EventLoopGroup bossGroup = new NioEventLoopGroup();
        EventLoopGroup workerGroup = new NioEventLoopGroup();
        try {
            Signal.handle(new Signal("USR2"), new SignalHandler() {
                public void handle(Signal signal) {
                    FileCache.clear();
                    Messages.printOK("Cleared file cache.");
                }
            });
            ServerBootstrap b = new ServerBootstrap();
            b.group(bossGroup, workerGroup).channel(NioServerSocketChannel.class).childHandler(new Initializer());
            ChannelFuture channelFuture = b.bind(Config.getHTTPAPIPort());
            if (channelFuture.await().isSuccess()) {
                Messages.printOK("Bound to port " + Config.getHTTPAPIPort());
                Page.initPages();
                Channel channel = channelFuture.sync().channel();
                channel.closeFuture().sync();
            } else
                Messages.printError("Failed to listen on " + Config.getHTTPAPIPort());
        } finally {
            bossGroup.shutdownGracefully();
            workerGroup.shutdownGracefully();
        }
    }

    public static void printPid() {
        try {
            byte[] bo = new byte[256];
            InputStream is = new FileInputStream("/proc/self/stat");
            is.read(bo);
            String pid = null;
            for (int i = 0; i < bo.length; i++) {
                if ((bo[i] < '0') || (bo[i] > '9')) {
                    pid = new String(bo, 0, i);
                    break;
                }
            }
            is.close();
            Messages.printOK("Pid of server is " + pid);
            if (Config.hasString(Config.Keys.PID_FILE)) {
                File pidfile = new File(Config.getString(Config.Keys.PID_FILE));
                pidfile.deleteOnExit();
                FileOutputStream pidout = new FileOutputStream(pidfile);
                pidout.write(pid.getBytes());
                pidout.flush();
                pidout.close();
            }
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
}