package com.facebook.presto.cli;

import com.facebook.presto.Main;
import com.facebook.presto.server.ServerMainModule;
import com.google.inject.Injector;
import io.airlift.bootstrap.Bootstrap;
import io.airlift.command.Command;
import io.airlift.discovery.client.Announcer;
import io.airlift.discovery.client.DiscoveryModule;
import io.airlift.event.client.HttpEventModule;
import io.airlift.http.server.HttpServerModule;
import io.airlift.jaxrs.JaxrsModule;
import io.airlift.jmx.JmxHttpModule;
import io.airlift.jmx.JmxModule;
import io.airlift.json.JsonModule;
import io.airlift.log.LogJmxModule;
import io.airlift.log.Logger;
import io.airlift.node.NodeModule;
import io.airlift.tracetoken.TraceTokenModule;
import org.weakref.jmx.guice.MBeanModule;

@Command(name = "server", description = "Run the server")
public class Server
        implements Runnable
{

    public void run()
    {
        Logger log = Logger.get(Main.class);
        Bootstrap app = new Bootstrap(
                new NodeModule(),
                new DiscoveryModule(),
                new HttpServerModule(),
                new JsonModule(),
                new JaxrsModule(),
                new MBeanModule(),
                new JmxModule(),
                new JmxHttpModule(),
                new LogJmxModule(),
                new HttpEventModule(),
                new TraceTokenModule(),
                new ServerMainModule());

        try {
            Injector injector = app.strictConfig().initialize();
            injector.getInstance(Announcer.class).start();
        }
        catch (Throwable e) {
            log.error(e);
            System.exit(1);
        }
    }
}

@@ -31,41 +31,8 @@
 i;%0A%0A
-import com.facebook.presto.Main;%0A
 impo
@@ -823,16 +823,29 @@
 nable%0A%7B%0A
+    @Override
 %0A    pub
@@ -897,20 +897,22 @@
 ger.get(
-Main
+Server
 .class);

Fix logger name