DonutModule.java

/*
 * Copyright 2009 Amit Levy, Jeff Prouty, Rylan Hawkins
 * 
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * 
 *      http://www.apache.org/licenses/LICENSE-2.0
 *      
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package edu.washington.cs.cse490h.donut.server;

import java.io.IOException;
import java.net.InetAddress;
import java.util.UUID;

import org.kohsuke.args4j.CmdLineException;
import org.kohsuke.args4j.CmdLineParser;
import org.kohsuke.args4j.Option;

import com.google.inject.Binder;
import com.google.inject.Module;

import edu.washington.cs.cse490h.donut.business.KeyId;
import edu.washington.cs.cse490h.donut.business.Node;
import edu.washington.cs.cse490h.donut.service.DonutHashRequestService;
import edu.washington.cs.cse490h.donut.service.HashService;
import edu.washington.cs.cse490h.donut.service.KeyLocator;
import edu.washington.cs.cse490h.donut.service.LocatorClientFactory;
import edu.washington.cs.cse490h.donut.service.NodeLocator;
import edu.washington.cs.cse490h.donut.service.RemoteLocatorClientFactory;
import edu.washington.cs.cse490h.donut.service.application.DonutHashTableService;
import edu.washington.cs.cse490h.donut.service.application.DonutInMemoryHashTableService;

/**
 * @author alevy
 */
public class DonutModule implements Module {

    @Option(name = "--hostname", usage = "the hostname to use for this Node")
    private String hostname          = InetAddress.getLocalHost().getCanonicalHostName();

    @Option(name = "--locator-port", usage = "the port on which to bind the Node's Locator Server (default: 8081)")
    private int    locatorPort              = 8081;

    @Option(name = "--request-port", usage = "the port on which to bind this Node Hashtable Server (default: 8080)")
    private int    requestPort = 8080;

    private long   key               = UUID.randomUUID().getMostSignificantBits();

    @Option(name = "--key", usage = "the 64-bit key for this Node (default: random)")
    void setKey(String key) {
        this.key = Long.decode(key);
    }

    @Option(name = "--known-host", usage = "the hostname of a known node (default: none)")
    private String knownHostname = null;

    @Option(name = "--known-port", usage = "the port of a known node (default: 8081)")
    private int    knownPort     = 8081;

    public DonutModule() throws Exception {
    }

    public void parseArgs(String[] args) throws IOException {
        CmdLineParser parser = new CmdLineParser(this);
        try {
            parser.parseArgument(args);
            System.out.println("Setup: " + getHostname() + " " + getPort() + " " + getKey());
        } catch (CmdLineException e) {
            int start = e.getMessage().indexOf('"') + 1;
            int end = e.getMessage().lastIndexOf('"');
            String wrongArgument = e.getMessage().substring(start, end);
            System.err.println("Unknown argument: " + wrongArgument);
            parser.printUsage(System.err);
            System.err.println();
            System.exit(1);
        }
    }

    public void configure(Binder binder) {
        Node node = new Node(getHostname(), getPort(), new KeyId(getKey()));

        binder.bind(Node.class).toInstance(node);
        binder.bind(LocatorClientFactory.class).to(RemoteLocatorClientFactory.class);
        binder.bind(KeyLocator.Iface.class).to(NodeLocator.class);
        binder.bind(HashService.Iface.class).to(DonutHashRequestService.class);
        binder.bind(DonutHashTableService.class).to(DonutInMemoryHashTableService.class);
    }

    public void setHostname(String hostname) {
        this.hostname = hostname;
    }

    public String getHostname() {
        return hostname;
    }

    public void setPort(int port) {
        this.locatorPort = port;
    }

    public int getPort() {
        return locatorPort;
    }

    public void setKey(long key) {
        this.key = key;
    }

    public long getKey() {
        return key;
    }

    public void setKnownHostname(String knownHostname) {
        this.knownHostname = knownHostname;
    }

    public String getKnownHostname() {
        return knownHostname;
    }

    public void setKnownPort(int knownPort) {
        this.knownPort = knownPort;
    }

    public int getKnownPort() {
        return knownPort;
    }

    public int getRequestPort() {
        return requestPort;
    }

}