From e9dac2e9ead1decb8abb1bf9df3cd638a2f5eb04 Mon Sep 17 00:00:00 2001 From: James Moger <james.moger@gitblit.com> Date: Tue, 15 Oct 2013 08:13:04 -0400 Subject: [PATCH] Merge pull request #118 from Mrbytes/master --- src/main/java/com/gitblit/GitBlitServer.java | 78 +++++++++++++++++++++++--------------- 1 files changed, 47 insertions(+), 31 deletions(-) diff --git a/src/main/java/com/gitblit/GitBlitServer.java b/src/main/java/com/gitblit/GitBlitServer.java index 08c0d2a..ace1f2f 100644 --- a/src/main/java/com/gitblit/GitBlitServer.java +++ b/src/main/java/com/gitblit/GitBlitServer.java @@ -75,9 +75,9 @@ * simplify command line parameter processing. This class also automatically * generates a self-signed certificate for localhost, if the keystore does not * already exist. - * + * * @author James Moger - * + * */ public class GitBlitServer { @@ -85,7 +85,7 @@ public static void main(String... args) { GitBlitServer server = new GitBlitServer(); - + // filter out the baseFolder parameter List<String> filtered = new ArrayList<String>(); String folder = "data"; @@ -103,7 +103,7 @@ filtered.add(arg); } } - + Params.baseFolder = folder; Params params = new Params(); JCommander jc = new JCommander(params); @@ -125,7 +125,7 @@ /** * Display the command line usage of Gitblit GO. - * + * * @param jc * @param t */ @@ -172,7 +172,7 @@ FileSettings settings = params.FILESETTINGS; if (!StringUtils.isEmpty(params.settingsfile)) { if (new File(params.settingsfile).exists()) { - settings = new FileSettings(params.settingsfile); + settings = new FileSettings(params.settingsfile); } } logger = LoggerFactory.getLogger(GitBlitServer.class); @@ -198,12 +198,12 @@ String osname = System.getProperty("os.name"); String osversion = System.getProperty("os.version"); logger.info("Running on " + osname + " (" + osversion + ")"); - + List<Connector> connectors = new ArrayList<Connector>(); // conditionally configure the http connector if (params.port > 0) { - Connector httpConnector = createConnector(params.useNIO, params.port); + Connector httpConnector = createConnector(params.useNIO, params.port, settings.getInteger(Keys.server.threadPoolSize, 50)); String bindInterface = settings.getString(Keys.server.httpBindInterface, null); if (!StringUtils.isEmpty(bindInterface)) { logger.warn(MessageFormat.format("Binding connector on port {0,number,0} to {1}", @@ -236,7 +236,7 @@ NewCertificateConfig certificateConfig = NewCertificateConfig.KEY.parse(config); certificateConfig.update(metadata); } - + metadata.notAfter = new Date(System.currentTimeMillis() + 10*TimeUtils.ONEYEAR); X509Utils.prepareX509Infrastructure(metadata, baseFolder, new X509Log() { @Override @@ -260,9 +260,9 @@ } }); - if (serverKeyStore.exists()) { + if (serverKeyStore.exists()) { Connector secureConnector = createSSLConnector(params.alias, serverKeyStore, serverTrustStore, params.storePassword, - caRevocationList, params.useNIO, params.securePort, params.requireClientCertificates); + caRevocationList, params.useNIO, params.securePort, settings.getInteger(Keys.server.threadPoolSize, 50), params.requireClientCertificates); String bindInterface = settings.getString(Keys.server.httpsBindInterface, null); if (!StringUtils.isEmpty(bindInterface)) { logger.warn(MessageFormat.format( @@ -297,7 +297,7 @@ // tempDir is where the embedded Gitblit web application is expanded and // where Jetty creates any necessary temporary files - File tempDir = com.gitblit.utils.FileUtils.resolveParameter(Constants.baseFolder$, baseFolder, params.temp); + File tempDir = com.gitblit.utils.FileUtils.resolveParameter(Constants.baseFolder$, baseFolder, params.temp); if (tempDir.exists()) { try { FileUtils.delete(tempDir, FileUtils.RECURSIVE | FileUtils.RETRY); @@ -342,7 +342,8 @@ // Override settings from the command-line settings.overrideSetting(Keys.realm.userService, params.userService); settings.overrideSetting(Keys.git.repositoriesFolder, params.repositoriesFolder); - + settings.overrideSetting(Keys.git.daemonPort, params.gitPort); + // Start up an in-memory LDAP server, if configured try { if (StringUtils.isEmpty(params.ldapLdifFile) == false) { @@ -353,21 +354,21 @@ String rootDN = firstLine.substring(4); String bindUserName = settings.getString(Keys.realm.ldap.username, ""); String bindPassword = settings.getString(Keys.realm.ldap.password, ""); - + // Get the port int port = ldapUrl.getPort(); if (port == -1) port = 389; - + InMemoryDirectoryServerConfig config = new InMemoryDirectoryServerConfig(rootDN); config.addAdditionalBindCredentials(bindUserName, bindPassword); config.setListenerConfigs(InMemoryListenerConfig.createLDAPConfig("default", port)); config.setSchema(null); - + InMemoryDirectoryServer ds = new InMemoryDirectoryServer(config); ds.importFromLDIF(true, new LDIFReader(ldifFile)); ds.startListening(); - + logger.info("LDAP Server started at ldap://localhost:" + port); } } @@ -399,29 +400,35 @@ System.exit(100); } } - + protected GitBlit getGitBlitInstance() { return GitBlit.self(); } /** * Creates an http connector. - * + * * @param useNIO * @param port + * @param threadPoolSize * @return an http connector */ - private Connector createConnector(boolean useNIO, int port) { + private Connector createConnector(boolean useNIO, int port, int threadPoolSize) { Connector connector; if (useNIO) { logger.info("Setting up NIO SelectChannelConnector on port " + port); SelectChannelConnector nioconn = new SelectChannelConnector(); nioconn.setSoLingerTime(-1); - nioconn.setThreadPool(new QueuedThreadPool(20)); + if (threadPoolSize > 0) { + nioconn.setThreadPool(new QueuedThreadPool(threadPoolSize)); + } connector = nioconn; } else { logger.info("Setting up SocketConnector on port " + port); SocketConnector sockconn = new SocketConnector(); + if (threadPoolSize > 0) { + sockconn.setThreadPool(new QueuedThreadPool(threadPoolSize)); + } connector = sockconn; } @@ -432,10 +439,10 @@ /** * Creates an https connector. - * + * * SSL renegotiation will be enabled if the JVM is 1.6.0_22 or later. * oracle.com/technetwork/java/javase/documentation/tlsreadme2-176330.html - * + * * @param certAlias * @param keyStore * @param clientTrustStore @@ -443,11 +450,12 @@ * @param caRevocationList * @param useNIO * @param port + * @param threadPoolSize * @param requireClientCertificates * @return an https connector */ private Connector createSSLConnector(String certAlias, File keyStore, File clientTrustStore, - String storePassword, File caRevocationList, boolean useNIO, int port, + String storePassword, File caRevocationList, boolean useNIO, int port, int threadPoolSize, boolean requireClientCertificates) { GitblitSslContextFactory factory = new GitblitSslContextFactory(certAlias, keyStore, clientTrustStore, storePassword, caRevocationList); @@ -461,11 +469,16 @@ } else { factory.setWantClientAuth(true); } - ssl.setThreadPool(new QueuedThreadPool(20)); + if (threadPoolSize > 0) { + ssl.setThreadPool(new QueuedThreadPool(threadPoolSize)); + } connector = ssl; } else { logger.info("Setting up NIO SslSocketConnector on port " + port); SslSocketConnector ssl = new SslSocketConnector(factory); + if (threadPoolSize > 0) { + ssl.setThreadPool(new QueuedThreadPool(threadPoolSize)); + } connector = ssl; } connector.setPort(port); @@ -473,10 +486,10 @@ return connector; } - + /** * Creates an ajp connector. - * + * * @param port * @return an ajp connector */ @@ -492,7 +505,7 @@ /** * Tests to see if the operating system is Windows. - * + * * @return true if this is a windows machine */ private boolean isWindows() { @@ -503,9 +516,9 @@ * The ShutdownMonitorThread opens a socket on a specified port and waits * for an incoming connection. When that connection is accepted a shutdown * message is issued to the running Jetty server. - * + * * @author James Moger - * + * */ private static class ShutdownMonitorThread extends Thread { @@ -601,6 +614,9 @@ @Parameter(names = "--ajpPort", description = "AJP port to serve. (port <= 0 will disable this connector)") public Integer ajpPort = FILESETTINGS.getInteger(Keys.server.ajpPort, 0); + @Parameter(names = "--gitPort", description = "Git Daemon port to serve. (port <= 0 will disable this connector)") + public Integer gitPort = FILESETTINGS.getInteger(Keys.git.daemonPort, 9418); + @Parameter(names = "--alias", description = "Alias of SSL certificate in keystore for serving https.") public String alias = FILESETTINGS.getString(Keys.server.certificateAlias, ""); @@ -618,7 +634,7 @@ */ @Parameter(names = { "--settings" }, description = "Path to alternative settings") public String settingsfile; - + @Parameter(names = { "--ldapLdifFile" }, description = "Path to LDIF file. This will cause an in-memory LDAP server to be started according to gitblit settings") public String ldapLdifFile; -- Gitblit v1.9.1