mogdbsetup

   1 #!/usr/bin/perl
   2 use strict;
   3 use Getopt::Long;
   4 use lib 'lib';
   5 use MogileFS::Store;
   6 use MogileFS::Config;
   7
   8 # Rename binary in process list to make init scripts saner
   9 $0 = $_ = $0;
  10
  11 my %args = (
  12             dbhost => "localhost",
  13             dbport => undef,
  14             dbname => "mogilefs",
  15             dbrootuser => undef,
  16             dbrootpass => "",
  17             dbuser => "mogile",
  18             dbpass => "",
  19             );
  20
  21 my $opt_help;
  22 my $opt_verbose = 0;
  23 my $opt_yes = 0;
  24 my $opt_noschemabump;
  25 my $dbtype = "MySQL";
  26 my $plugins;
  27
  28 usage()
  29     unless GetOptions(
  30                       "dbhost=s"         => \$args{dbhost},
  31                       "dbport=s"         => \$args{dbport},
  32                       "dbname=s"         => \$args{dbname},
  33                       "dbrootuser=s"     => \$args{dbrootuser},
  34                       "dbrootpassword:s" => \$args{dbrootpass},
  35                       "dbuser=s"         => \$args{dbuser},
  36                       "dbpassword:s"     => \$args{dbpass},
  37                       "help"             => \$opt_help,
  38                       "verbose"          => \$opt_verbose,
  39                       "yes"              => \$opt_yes,
  40                       "noschemabump"     => \$opt_noschemabump,
  41                       "type=s"           => \$dbtype,
  42                       "plugins=s"        => \$plugins,
  43                       );
  44
  45 usage() if $opt_help;
  46
  47 # Be nice about what the default admin user is called.
  48 if(!defined($args{dbrootuser})) {
  49     $args{dbrootuser} = 'root' if $dbtype =~ /MySQL/i;
  50     $args{dbrootuser} = 'postgres' if $dbtype =~ /Postgres/i;
  51 }
  52 # Saner port management.
  53 # This should default to the UNIX sockets on localhost
  54 if(!defined($args{dbport}) and $args{dbhost} != "localhost" and $args{dbhost} != "127.0.0.1") {
  55     $args{dbport} = '3306' if $dbtype =~ /MySQL/i;
  56     $args{dbport} = '5432' if $dbtype =~ /Postgres/i;
  57 }
  58
  59 sub usage {
  60     die <<USAGE;
  61 Usage: mogdbsetup [opts]
  62
  63 Options:
  64
  65                   Default      Description
  66                   ============ ===========================================
  67  --verbose        <off>        Be verbose about what\'s happening.
  68
  69  --dbhost=        localhost    hostname or IP to database server.
  70
  71  --dbport=        dbd default  port number to database server.
  72
  73  --dbname=        mogilefs     database name to create/upgrade.
  74
  75  --dbrootuser=    root         Database administrator username.  Only needed
  76                                for initial setup, not subsequent upgrades.
  77
  78  --dbrootpass=    <blank>      Database administrator password.  Only needed
  79                                for initial setup, not subsequent upgrades.
  80
  81  --dbuser=        mogile       Regular database user to create and/or use
  82                                for MogileFS database.  This is what the
  83                                mogilefsd trackers connect as.
  84
  85  --dbpass=        <blank>      You should change this, especially if your
  86                                database servers are accessible to other users
  87                                on the network.  But they shouldn't be
  88                                if you're running MogileFS, because MogileFS
  89                                assumes your network is closed.
  90
  91   --type=         MySQL        Which MogileFS::Store implementation to use.
  92                                Available: MySQL, Postgres
  93
  94   --yes                        Run without questions.
  95
  96 USAGE
  97 }
  98
  99 my $sclass = "MogileFS::Store::$dbtype";
 100 eval "use $sclass; 1;" or die "Failed to load $sclass: $@";
 101
 102 foreach my $plugin (split /\s*,\s*/, $plugins) {
 103     eval "use MogileFS::Plugin::$plugin; 1;" or die "Failed to load plugin $plugin: $@";
 104 }
 105
 106 confirm("This will attempt to setup or upgrade your MogileFS database.\nIt won't destroy existing data.\nRun with --help for more information.  Run with --yes to shut up these prompts.\n\nContinue?", 0);
 107
 108 $sclass->on_status(\&status);
 109 $sclass->on_confirm(\&confirm);
 110
 111 MogileFS::Config->load_config;
 112
 113 my $sto = $sclass->new_from_mogdbsetup(
 114                                        map { $_ => $args{$_} }
 115                                        qw(dbhost dbport dbname
 116                                           dbrootuser dbrootpass
 117                                           dbuser dbpass)
 118                                        );
 119 my $dbh = $sto->dbh;
 120
 121 $sto->setup_database
 122     or die "Database upgrade failed.\n";
 123
 124 my $latestver = MogileFS::Store->latest_schema_version;
 125 if ($opt_noschemabump) {
 126     warn "\n*\n* Per your request, NOT UPGRADING to $latestver.  I assume you understand why.\n*\n";
 127 } else {
 128     $sto->set_schema_vesion($latestver);
 129 }
 130
 131
 132 warn "Done.\n" if $opt_verbose;
 133 exit 0;
 134
 135 ############################################################################
 136
 137 sub confirm {
 138     my $q = shift;
 139     my $def = shift;
 140     $def = 1 unless defined $def;
 141
 142     return 1 if $opt_yes;
 143     my $deftext = $def ? "[Y/n]" : "[N/y]";
 144
 145     print "\n$q $deftext: ";
 146     my $ans = <STDIN>;
 147     if ($ans =~ /^\s*$/) {
 148         die "Stopped.\n" unless $def;
 149         return 0;
 150     }
 151     return 1 if $ans =~ /^y/i;
 152     die "Stopped.\n";
 153 }
 154
 155 sub status {
 156     warn "$_[0]\n" if $opt_verbose;
 157 }
 158
 159