vagrant-openstack 0.0.4

data/tmp/librarian/cache/source/chef/site/877777683730772c36b1e6a3fc3aa2c3/mongodb/version-uri/ddb10eb171f8093fd208b02b177f6214/package/files/default/mongodb.init

@@ -0,0 +1,216 @@
+#!/bin/sh
+#
+# init.d script with LSB support.
+#
+# Copyright (c) 2007 Javier Fernandez-Sanguino <jfs@debian.org>
+# Copyright (c) 2011 edelight GmbH
+#       Author: Markus Korn <markus.korn@edelight.de>
+#
+# This is free software; you may redistribute it and/or modify
+# it under the terms of the GNU General Public License as
+# published by the Free Software Foundation; either version 2,
+# or (at your option) any later version.
+#
+# This is distributed in the hope that it will be useful, but
+# WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License with
+# the Debian operating system, in /usr/share/common-licenses/GPL;  if
+# not, write to the Free Software Foundation, Inc., 59 Temple Place,
+# Suite 330, Boston, MA 02111-1307 USA
+#
+
+PATH=/usr/local/sbin:/usr/local/bin:/sbin:/bin:/usr/sbin:/usr/bin
+DESC=database
+DAEMON=/usr/bin/mongod
+DAEMON_OPTS="--config /etc/mongodb.conf"
+
+ENABLE_MONGODB=yes
+
+NAME=`basename $0`
+
+# Include mongodb defaults if available
+if [ -f /etc/default/$NAME ] ; then
+	. /etc/default/$NAME
+fi
+
+PIDFILE=/var/run/$NAME.pid
+
+if test ! -x $DAEMON; then
+    echo "Could not find $DAEMON"
+    exit 0
+fi
+
+if test "x$ENABLE_MONGODB" != "xyes"; then
+    exit 0
+fi
+
+. /lib/lsb/init-functions
+
+STARTTIME=1
+DIETIME=10                  # Time to wait for the server to die, in seconds
+                            # If this value is set too low you might not
+                            # let some servers to die gracefully and
+                            # 'restart' will not work
+
+DAEMONUSER=${DAEMONUSER:-mongodb}
+
+# debugging
+echo "** Running $NAME ($DAEMON $DAEMON_OPTS)"
+
+set -e
+
+
+running_pid() {
+# Check if a given process pid's cmdline matches a given name
+    pid=$1
+    name=$2
+    [ -z "$pid" ] && return 1
+    [ ! -d /proc/$pid ] &&  return 1
+    cmd=`cat /proc/$pid/cmdline | tr "\000" "\n"|head -n 1 |cut -d : -f 1`
+    # Is this the expected server
+    [ "$cmd" != "$name" ] &&  return 1
+    return 0
+}
+
+running() {
+# Check if the process is running looking at /proc
+# (works for all users)
+
+    # No pidfile, probably no daemon present
+    [ ! -f "$PIDFILE" ] && return 1
+    pid=`cat $PIDFILE`
+    running_pid $pid $DAEMON || return 1
+    return 0
+}
+
+start_server() {
+# Start the process using the wrapper
+            start-stop-daemon --background --start --quiet --pidfile $PIDFILE \
+                        --make-pidfile --chuid $DAEMONUSER \
+                        --exec $DAEMON -- $DAEMON_OPTS
+            errcode=$?
+	return $errcode
+}
+
+stop_server() {
+# Stop the process using the wrapper
+            start-stop-daemon --stop --quiet --pidfile $PIDFILE \
+                        --retry 300 \
+                        --user $DAEMONUSER \
+                        --exec $DAEMON
+            errcode=$?
+	return $errcode
+}
+
+force_stop() {
+# Force the process to die killing it manually
+	[ ! -e "$PIDFILE" ] && return
+	if running ; then
+		kill -15 $pid
+	# Is it really dead?
+		sleep "$DIETIME"s
+		if running ; then
+			kill -9 $pid
+			sleep "$DIETIME"s
+			if running ; then
+				echo "Cannot kill $NAME (pid=$pid)!"
+				exit 1
+			fi
+		fi
+	fi
+	rm -f $PIDFILE
+}
+
+
+case "$1" in
+  start)
+	log_daemon_msg "Starting $DESC" "$NAME"
+        # Check if it's running first
+        if running ;  then
+            log_progress_msg "apparently already running"
+            log_end_msg 0
+            exit 0
+        fi
+        if start_server ; then
+            # NOTE: Some servers might die some time after they start,
+            # this code will detect this issue if STARTTIME is set
+            # to a reasonable value
+            [ -n "$STARTTIME" ] && sleep $STARTTIME # Wait some time 
+            if  running ;  then
+                # It's ok, the server started and is running
+                log_end_msg 0
+            else
+                # It is not running after we did start
+                log_end_msg 1
+            fi
+        else
+            # Either we could not start it
+            log_end_msg 1
+        fi
+	;;
+  stop)
+        log_daemon_msg "Stopping $DESC" "$NAME"
+        if running ; then
+            # Only stop the server if we see it running
+			errcode=0
+            stop_server || errcode=$?
+            log_end_msg $errcode
+        else
+            # If it's not running don't do anything
+            log_progress_msg "apparently not running"
+            log_end_msg 0
+            exit 0
+        fi
+        ;;
+  force-stop)
+        # First try to stop gracefully the program
+        $0 stop
+        if running; then
+            # If it's still running try to kill it more forcefully
+            log_daemon_msg "Stopping (force) $DESC" "$NAME"
+			errcode=0
+            force_stop || errcode=$?
+            log_end_msg $errcode
+        fi
+	;;
+  restart|force-reload)
+        log_daemon_msg "Restarting $DESC" "$NAME"
+		errcode=0
+        stop_server || errcode=$?
+        # Wait some sensible amount, some server need this
+        [ -n "$DIETIME" ] && sleep $DIETIME
+        start_server || errcode=$?
+        [ -n "$STARTTIME" ] && sleep $STARTTIME
+        running || errcode=$?
+        log_end_msg $errcode
+	;;
+  status)
+
+        log_daemon_msg "Checking status of $DESC" "$NAME"
+        if running ;  then
+            log_progress_msg "running"
+            log_end_msg 0
+        else
+            log_progress_msg "apparently not running"
+            log_end_msg 1
+            exit 1
+        fi
+        ;;
+  # MongoDB can't reload its configuration.
+  reload)
+        log_warning_msg "Reloading $NAME daemon: not implemented, as the daemon"
+        log_warning_msg "cannot re-read the config file (use restart)."
+        ;;
+
+  *)
+	N=/etc/init.d/$NAME
+	echo "Usage: $N {start|stop|force-stop|restart|force-reload|status}" >&2
+	exit 1
+	;;
+esac
+
+exit 0
+

data/tmp/librarian/cache/source/chef/site/877777683730772c36b1e6a3fc3aa2c3/mongodb/version-uri/ddb10eb171f8093fd208b02b177f6214/package/libraries/mongodb.rb

@@ -0,0 +1,262 @@
+#
+# Cookbook Name:: mongodb
+# Definition:: mongodb
+#
+# Copyright 2011, edelight GmbH
+# Authors:
+#       Markus Korn <markus.korn@edelight.de>
+#
+# 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.
+#
+
+require 'json'
+
+class Chef::ResourceDefinitionList::MongoDB
+
+  def self.configure_replicaset(node, name, members)
+    # lazy require, to move loading this modules to runtime of the cookbook
+    require 'rubygems'
+    require 'mongo'
+    
+    if members.length == 0
+      abort("cannot configure replicaset '#{name}', no member nodes found")
+    end
+    
+    begin
+      connection = Mongo::Connection.new('localhost', node['mongodb']['port'], :op_timeout => 5, :slave_ok => true)
+    rescue
+      Chef::Log.warn("Could not connect to database: 'localhost:#{node['mongodb']['port']}'")
+      return
+    end
+    
+    members.sort!{ |x,y| x['name'] <=> y['name'] }
+    rs_members = []
+    members.each_index do |n|
+      port = members[n]['mongodb']['port']
+      rs_members << {"_id" => n, "host" => "#{members[n]['fqdn']}:#{port}"}
+    end
+    
+    Chef::Log.info(
+      "Configuring replicaset with members #{members.collect{ |n| n['hostname'] }.join(', ')}"
+    )
+    
+    rs_member_ips = []
+    members.each_index do |n|
+      port = members[n]['mongodb']['port']
+      rs_member_ips << {"_id" => n, "host" => "#{members[n]['ipaddress']}:#{port}"}
+    end
+    
+    admin = connection['admin']
+    cmd = BSON::OrderedHash.new
+    cmd['replSetInitiate'] = {
+        "_id" => name,
+        "members" => rs_members
+    }
+    
+    begin
+      result = admin.command(cmd, :check_response => false)
+    rescue Mongo::OperationTimeout
+      Chef::Log.info("Started configuring the replicaset, this will take some time, another run should run smoothly")
+      return
+    end
+    if result.fetch("ok", nil) == 1
+      # everything is fine, do nothing
+    elsif result.fetch("errmsg", nil) == "already initialized"
+      # check if both configs are the same
+      config = connection['local']['system']['replset'].find_one({"_id" => name})
+      if config['_id'] == name and config['members'] == rs_members
+        # config is up-to-date, do nothing
+        Chef::Log.info("Replicaset '#{name}' already configured")
+      elsif config['_id'] == name and config['members'] == rs_member_ips
+        # config is up-to-date, but ips are used instead of hostnames, change config to hostnames
+        Chef::Log.info("Need to convert ips to hostnames for replicaset '#{name}'")
+        old_members = config['members'].collect{ |m| m['host'] }
+        mapping = {}
+        rs_member_ips.each do |mem_h|
+          members.each do |n|
+            ip, prt = mem_h['host'].split(":")
+            if ip == n['ipaddress']
+              mapping["#{ip}:#{prt}"] = "#{n['fqdn']}:#{prt}"
+            end
+          end
+        end
+        config['members'].collect!{ |m| {"_id" => m["_id"], "host" => mapping[m["host"]]} }
+        config['version'] += 1
+        
+        rs_connection = Mongo::ReplSetConnection.new( *old_members.collect{ |m| m.split(":") })
+        admin = rs_connection['admin']
+        cmd = BSON::OrderedHash.new
+        cmd['replSetReconfig'] = config
+        result = nil
+        begin
+          result = admin.command(cmd, :check_response => false)
+        rescue Mongo::ConnectionFailure
+          # reconfiguring destroys exisiting connections, reconnect
+          Mongo::Connection.new('localhost', node['mongodb']['port'], :op_timeout => 5, :slave_ok => true)
+          config = connection['local']['system']['replset'].find_one({"_id" => name})
+          Chef::Log.info("New config successfully applied: #{config.inspect}")
+        end
+        if !result.nil?
+          Chef::Log.error("configuring replicaset returned: #{result.inspect}")
+        end
+      else
+        # remove removed members from the replicaset and add the new ones
+        max_id = config['members'].collect{ |member| member['_id']}.max
+        rs_members.collect!{ |member| member['host'] }
+        config['version'] += 1
+        old_members = config['members'].collect{ |member| member['host'] }
+        members_delete = old_members - rs_members        
+        config['members'] = config['members'].delete_if{ |m| members_delete.include?(m['host']) }
+        members_add = rs_members - old_members
+        members_add.each do |m|
+          max_id += 1
+          config['members'] << {"_id" => max_id, "host" => m}
+        end
+        
+        rs_connection = Mongo::ReplSetConnection.new( *old_members.collect{ |m| m.split(":") })
+        admin = rs_connection['admin']
+        
+        cmd = BSON::OrderedHash.new
+        cmd['replSetReconfig'] = config
+        
+        result = nil
+        begin
+          result = admin.command(cmd, :check_response => false)
+        rescue Mongo::ConnectionFailure
+          # reconfiguring destroys exisiting connections, reconnect
+          Mongo::Connection.new('localhost', node['mongodb']['port'], :op_timeout => 5, :slave_ok => true)
+          config = connection['local']['system']['replset'].find_one({"_id" => name})
+          Chef::Log.info("New config successfully applied: #{config.inspect}")
+        end
+        if !result.nil?
+          Chef::Log.error("configuring replicaset returned: #{result.inspect}")
+        end
+      end
+    elsif !result.fetch("errmsg", nil).nil?
+      Chef::Log.error("Failed to configure replicaset, reason: #{result.inspect}")
+    end
+  end
+  
+  def self.configure_shards(node, shard_nodes)
+    # lazy require, to move loading this modules to runtime of the cookbook
+    require 'rubygems'
+    require 'mongo'
+    
+    shard_groups = Hash.new{|h,k| h[k] = []}
+    
+    shard_nodes.each do |n|
+      if n['recipes'].include?('mongodb::replicaset')
+        key = "rs_#{n['mongodb']['shard_name']}"
+      else
+        key = '_single'
+      end
+      shard_groups[key] << "#{n['fqdn']}:#{n['mongodb']['port']}"
+    end
+    Chef::Log.info(shard_groups.inspect)
+    
+    shard_members = []
+    shard_groups.each do |name, members|
+      if name == "_single"
+        shard_members += members
+      else
+        shard_members << "#{name}/#{members.join(',')}"
+      end
+    end
+    Chef::Log.info(shard_members.inspect)
+    
+    begin
+      connection = Mongo::Connection.new('localhost', node['mongodb']['port'], :op_timeout => 5)
+    rescue Exception => e
+      Chef::Log.warn("Could not connect to database: 'localhost:#{node['mongodb']['port']}', reason #{e}")
+      return
+    end
+    
+    admin = connection['admin']
+    
+    shard_members.each do |shard|
+      cmd = BSON::OrderedHash.new
+      cmd['addShard'] = shard
+      begin
+        result = admin.command(cmd, :check_response => false)
+      rescue Mongo::OperationTimeout
+        result = "Adding shard '#{shard}' timed out, run the recipe again to check the result"
+      end
+      Chef::Log.info(result.inspect)
+    end
+  end
+  
+  def self.configure_sharded_collections(node, sharded_collections)
+    # lazy require, to move loading this modules to runtime of the cookbook
+    require 'rubygems'
+    require 'mongo'
+    
+    begin
+      connection = Mongo::Connection.new('localhost', node['mongodb']['port'], :op_timeout => 5)
+    rescue Exception => e
+      Chef::Log.warn("Could not connect to database: 'localhost:#{node['mongodb']['port']}', reason #{e}")
+      return
+    end
+    
+    admin = connection['admin']
+    
+    databases = sharded_collections.keys.collect{ |x| x.split(".").first}.uniq
+    Chef::Log.info("enable sharding for these databases: '#{databases.inspect}'")
+    
+    databases.each do |db_name|
+      cmd = BSON::OrderedHash.new
+      cmd['enablesharding'] = db_name
+      begin
+        result = admin.command(cmd, :check_response => false)
+      rescue Mongo::OperationTimeout
+        result = "enable sharding for '#{db_name}' timed out, run the recipe again to check the result"
+      end
+      if result['ok'] == 0
+        # some error
+        errmsg = result.fetch("errmsg")
+        if errmsg == "already enabled"
+          Chef::Log.info("Sharding is already enabled for database '#{db_name}', doing nothing")
+        else
+          Chef::Log.error("Failed to enable sharding for database #{db_name}, result was: #{result.inspect}")
+        end
+      else
+        # success
+        Chef::Log.info("Enabled sharding for database '#{db_name}'")
+      end
+    end
+    
+    sharded_collections.each do |name, key|
+      cmd = BSON::OrderedHash.new
+      cmd['shardcollection'] = name
+      cmd['key'] = {key => 1}
+      begin
+        result = admin.command(cmd, :check_response => false)
+      rescue Mongo::OperationTimeout
+        result = "sharding '#{name}' on key '#{key}' timed out, run the recipe again to check the result"
+      end
+      if result['ok'] == 0
+        # some error
+        errmsg = result.fetch("errmsg")
+        if errmsg == "already sharded"
+          Chef::Log.info("Sharding is already configured for collection '#{name}', doing nothing")
+        else
+          Chef::Log.error("Failed to shard collection #{name}, result was: #{result.inspect}")
+        end
+      else
+        # success
+        Chef::Log.info("Sharding for collection '#{result['collectionsharded']}' enabled")
+      end
+    end
+  
+  end
+  
+end