BEServer.java

import java.net.InetAddress;

import org.apache.log4j.BasicConfigurator;
import org.apache.log4j.Logger;

import org.apache.thrift.TProcessorFactory;
import org.apache.thrift.protocol.TBinaryProtocol;
import org.apache.thrift.protocol.TProtocol;
import org.apache.thrift.server.TThreadPoolServer;
import org.apache.thrift.transport.TServerSocket;
import org.apache.thrift.transport.TSocket;
import org.apache.thrift.transport.TTransport;
import org.apache.thrift.transport.layered.TFramedTransport;

public class BEServer {
    static Logger log;

    public static void main(String [] args) throws Exception {
        if (args.length != 3) {
            System.err.println("Usage: java BENode FE_host FE_port BE_port");
            System.exit(-1);
        }

        // initialize log4j
        BasicConfigurator.configure();
        log = Logger.getLogger(BEServer.class.getName());

        String hostFE = args[0];
        int portFE = Integer.parseInt(args[1]);
        int portBE = Integer.parseInt(args[2]);
        log.info("Launching BE node on port " + portBE + " at host " + getHostName());

        // launch Thrift server in a separate thread
        MiningPoolService.Processor processor = new MiningPoolService.Processor<MiningPoolService.Iface>(new MiningPoolServiceHandler(true));
        TServerSocket socket = new TServerSocket(portBE);
        TThreadPoolServer.Args sargs = new TThreadPoolServer.Args(socket);
        sargs.protocolFactory(new TBinaryProtocol.Factory());
        sargs.transportFactory(new TFramedTransport.Factory());
        sargs.processorFactory(new TProcessorFactory(processor));
        TThreadPoolServer server = new TThreadPoolServer(sargs);

        Thread registrationThread = new Thread(() -> {
            while (true) {
                try {
                    TSocket beSocket = new TSocket(hostFE, portFE);
                    TTransport transport = new TFramedTransport(beSocket);
                    TProtocol protocol = new TBinaryProtocol(transport);
                    MiningPoolService.Client beClient = new MiningPoolService.Client(protocol);
                    transport.open();
                    beClient.registerBEServer(hostFE, portBE, Runtime.getRuntime().availableProcessors());
                    transport.close();
                    break;
                } catch (Exception e) {
                    try {
                        Thread.sleep(50);
                    } catch (InterruptedException ie) {
                        Thread.currentThread().interrupt();
                    }
                }
            }
        });
        registrationThread.start();
        System.out.println("Starting BE server.");
        server.serve();
    }

    static String getHostName() {
        try {
            return InetAddress.getLocalHost().getHostName();
        } catch (Exception e) {
            return "localhost";
        }
    }
}