configuration for number of open mongodb connections

nixon · nixon · commit dfc96c5869bc · 2012-02-16T18:31:09.000-06:00
Allow users to change the default nofile ulimit value when installing
mongodb.  The mongodb class has been changed to a parameterized class
that accepts a `ulimit_nofile` parameter as well as a `replSet`
parameter.

This is a backwards incompatible change for users of the old
`replica_set` definition.
diff --git a/README b/README
@@ -19,9 +19,12 @@ Update it:
 Sample Usage :
 	node 'mongodb-1.domain.com', 'mongodb-2.domain.com', 'mongodb-3.domain.com' {
 		# Install MongoDB
-		include mongodb
-		# Nodes are in a replica set
-		mongodb::replica_set { "example_set_name": }
+		class { mongodb:
+			# Nodes are in a replica set
+			replSet => "example_set_name",
+			# Increase number of available mongodb connections
+			ulimit_nofile => 21000,
+		}
 	}
 
 ## CONTRIBUTORS ##
diff --git a/manifests/init.pp b/manifests/init.pp
@@ -12,11 +12,18 @@
 #  - MongoDB can be part of a replica set
 #
 # Sample Usage:
-#  include mongodb
+#  class { mongodb:
+#    replSet => "myReplicaSet",
+#    ulimit_nofile => 21000,
+#  }
 #
-class mongodb {
-	include mongodb::params
-	
+class mongodb(
+  $replSet = $mongodb::params::replSet,
+  $ulimit_nofile = $mongodb::params::ulimit_nofile,
+  $repository = $mongodb::params::repository,
+  $package = $mongodb::params::package
+) inherits mongodb::params {
+
 	package { "python-software-properties":
 		ensure => installed,
 	}
@@ -46,19 +53,15 @@
 		ensure => installed,
 		require => Exec["update-apt"],
 	}
-	
+
 	service { "mongodb":
 		enable => true,
 		ensure => running,
-		require => Package[$mongodb::params::package],
-	}
-	
-	define replica_set {
-		file { "/etc/init/mongodb.conf":
-			content => template("mongodb/mongodb.conf.erb"),
-			mode => "0644",
-			notify => Service["mongodb"],
-			require => Package[$mongodb::params::package],
-		}
 	}
+
+  file { "/etc/init/mongodb.conf":
+    content => template("mongodb/mongodb.conf.erb"),
+    mode => "0644",
+    notify => Service["mongodb"],
+  }
 }
diff --git a/manifests/params.pp b/manifests/params.pp
@@ -5,11 +5,22 @@
 # Parameters:
 # - The 10gen Ubuntu $repository to use
 # - The 10gen Ubuntu $package to use
+# - A replica set to join
+# - A nofile ulimit
 #
 # Sample Usage:
 #  include mongodb::params
 #
 class mongodb::params {
 	$repository="deb http://downloads-distro.mongodb.org/repo/ubuntu-upstart dist 10gen"
 	$package="mongodb-10gen"
+
+  # name of replica set (if any) to join
+  $replSet = ""
+
+  # number of open files ulimit can be changed if mongodb.log reports
+  # "too many open files" or "too many open connections" messages.
+  # mongodb has an upper limit of 20k.
+  # http://www.mongodb.org/display/DOCS/Too+Many+Open+Files
+  $ulimit_nofile = 1024
 }
diff --git a/templates/mongodb.conf.erb b/templates/mongodb.conf.erb
@@ -9,8 +9,12 @@ end script
 start on runlevel [2345]
 stop on runlevel [06]
 
+# http://www.mongodb.org/display/DOCS/Too+Many+Open+Files
+# mongodb has a limit of 20k connections
+limit nofile <%= ulimit_nofile %> <%= ulimit_nofile %>
+
 script
   ENABLE_MONGODB="yes"
   if [ -f /etc/default/mongodb ]; then . /etc/default/mongodb; fi
-  if [ "x$ENABLE_MONGODB" = "xyes" ]; then exec start-stop-daemon --start --quiet --chuid mongodb --exec  /usr/bin/mongod -- --config /etc/mongodb.conf --replSet <%= name %>; fi
+  if [ "x$ENABLE_MONGODB" = "xyes" ]; then exec start-stop-daemon --start --quiet --chuid mongodb --exec  /usr/bin/mongod -- --config /etc/mongodb.conf <%= replSet ? "--replSet #{replSet}" : "" %>; fi
 end script
