repctl 0.0.6 → 0.0.7

data/lib/repctl/client.rb

@@ -0,0 +1,131 @@
+#
+# This module is intended to be used on the client side to connect to
+# the repctl webserver.  It is not part of repctl itself, as it is not
+# 'required' by any file in the repctl source code.
+#
+require 'net/http'
+
+module Repctl
+  module Client
+
+    USER = 'admin'
+    PASSWORD = 'secret'
+
+    def get_status(host, opts = {})
+      components = { 
+        :host => host,
+        :path => "/status",
+        :port => opts[:port] || 9393, # 7250
+      }
+      uri = URI::HTTP.build(components)
+      do_get(uri)
+    end
+
+    def switch_master(host, master, slaves, opts = {})
+      master = master.to_s
+      slaves = slaves.reduce("") { |list, slave| list + " " + slave.to_s}
+      body = "switch[master]=#{master}&switch[slaves]=#{slaves}"
+      components = { 
+        :host => host,
+        :path => "/switch_master",
+        :port => opts[:port] || 9393, # 7250
+      }
+      uri = URI::HTTP.build(components)
+      do_post(uri, body)
+    end
+
+    def add_slave(host, master, slave, opts = {})
+      master = master.to_s
+      slave = slave.to_s
+      body = "add_slave[master]=#{master}&add_slave[slaves]=#{slave}"
+      if opts[:sync]
+        body += "&add_slave[sync]=sync"
+      end
+      components = { 
+        :host => host,
+        :path => "/add_slave",
+        :port => opts[:port] || 9393, # 7250
+      }
+      uri = URI::HTTP.build(components)
+      do_post(uri, body)
+    end
+
+    def remove_slave(host, slave, opts = {})
+      slave = slave.to_s
+      body = "remove_slave[slave]=#{slave}"
+      components = { 
+        :host => host,
+        :path => "/remove_slave",
+        :port => opts[:port] || 9393, # 7250
+      }
+      uri = URI::HTTP.build(components)
+      do_post(uri, body)
+    end
+
+    def repl_trio(host, master, slaves, opts = {})
+      master = master.to_s
+      slaves = slaves.reduce("") { |list, slave| list + " " + slave.to_s}
+      body = "repl_trio[master]=#{master}&repl_trio[slaves]=#{slaves}"
+      components = { 
+        :host => host,
+        :path => "/repl_trio",
+        :port => opts[:port] || 9393, # 7250
+      }
+      uri = URI::HTTP.build(components)
+      do_post(uri, body)
+    end
+
+    private
+
+    def do_get(uri)
+      http = Net::HTTP.new(uri.host, uri.port)
+#     http.use_ssl = true
+#     http.verify_mode = OpenSSL::SSL::VERIFY_NONE
+
+      req = Net::HTTP::Get.new(uri.request_uri)
+      req.basic_auth(USER, PASSWORD)
+      req['Accept'] = 'text/plain'
+      response = http.request(req)
+      if response.is_a?(Net::HTTPOK)
+        response.body
+      else
+        response.message
+      end
+    end
+
+    def do_post(uri, body)
+      http = Net::HTTP.new(uri.host, uri.port)
+      #http.use_ssl = true
+      #http.verify_mode = OpenSSL::SSL::VERIFY_NONE
+      req = Net::HTTP::Post.new(uri.request_uri)
+      req.basic_auth(USER, PASSWORD)
+      req['Accept'] = 'text/plain'
+      req['Content-Type'] = "application/x-www-form-urlencoded"
+      req.body = body
+      response = http.request(req)
+      if response.is_a?(Net::HTTPOK)
+        response.body
+      else
+        response.message
+      end
+    end
+  end
+end
+
+include Repctl::Client
+
+def self_test
+  host = "deimos"
+  puts repl_trio(host, 2, [3, 4])
+  puts get_status(host)
+  puts add_slave(host, 2, 1)
+  puts get_status(host)
+  puts switch_master(host, 3, [1, 2, 4])
+  puts get_status(host)
+  puts remove_slave(host, 1)
+  puts get_status(host)
+  puts remove_slave(host, 4)
+  puts get_status(host)
+end
+
+self_test

data/lib/repctl/version.rb

@@ -1,3 +1,3 @@
 module Repctl
-  VERSION = "0.0.6"
+  VERSION = "0.0.7"
 end

data/webserver/webserver.rb

@@ -1,6 +1,11 @@
 require 'sinatra'
 require 'repctl'
 
+# Add middleware for HTTP Basic Authentication.
+use Rack::Auth::Basic do |username, password|
+  username == 'admin' && password == 'secret'
+end
+
 include Repctl::Config
 include Repctl::Commands
 include Repctl::Servers
@@ -58,7 +63,7 @@ post '/switch_master' do
     secs = time do
       do_switch_master(master, slaves)
     end
-    @message = "Switch master processed in #{secs} secs."
+    @message = "Switch to master #{master.to_s} processed in #{secs} secs."
     @success = true
   end
   if request.accept == ['text/plain']
@@ -135,7 +140,7 @@ post '/repl_trio' do
       do_repl_trio(master, slaves[0], slaves[1])
     end
     @message = "Create replication trio with master #{master} and " +
-      "slaves #{slaves[0] and slaves[1]} in #{secs} secs."
+      "slaves #{slaves[0]} and #{slaves[1]} in #{secs} secs."
     @success = true
   end
   if request.accept == ['text/plain']

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: repctl
 version: !ruby/object:Gem::Version
-  version: 0.0.6
+  version: 0.0.7
   prerelease: 
 platform: ruby
 authors:
@@ -9,11 +9,11 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2012-03-09 00:00:00.000000000Z
+date: 2012-03-13 00:00:00.000000000Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rspec
-  requirement: &195841920 !ruby/object:Gem::Requirement
+  requirement: &185309620 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -21,10 +21,10 @@ dependencies:
         version: '0'
   type: :development
   prerelease: false
-  version_requirements: *195841920
+  version_requirements: *185309620
 - !ruby/object:Gem::Dependency
   name: shotgun
-  requirement: &195841500 !ruby/object:Gem::Requirement
+  requirement: &185309200 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -32,10 +32,10 @@ dependencies:
         version: '0'
   type: :development
   prerelease: false
-  version_requirements: *195841500
+  version_requirements: *185309200
 - !ruby/object:Gem::Dependency
   name: compass
-  requirement: &195841080 !ruby/object:Gem::Requirement
+  requirement: &185308780 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -43,10 +43,10 @@ dependencies:
         version: '0'
   type: :development
   prerelease: false
-  version_requirements: *195841080
+  version_requirements: *185308780
 - !ruby/object:Gem::Dependency
   name: thor
-  requirement: &195840660 !ruby/object:Gem::Requirement
+  requirement: &185308360 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -54,10 +54,10 @@ dependencies:
         version: '0'
   type: :runtime
   prerelease: false
-  version_requirements: *195840660
+  version_requirements: *185308360
 - !ruby/object:Gem::Dependency
   name: mysql2
-  requirement: &195840240 !ruby/object:Gem::Requirement
+  requirement: &185307940 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -65,10 +65,10 @@ dependencies:
         version: '0'
   type: :runtime
   prerelease: false
-  version_requirements: *195840240
+  version_requirements: *185307940
 - !ruby/object:Gem::Dependency
   name: sinatra
-  requirement: &195839820 !ruby/object:Gem::Requirement
+  requirement: &185307520 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -76,10 +76,10 @@ dependencies:
         version: '0'
   type: :runtime
   prerelease: false
-  version_requirements: *195839820
+  version_requirements: *185307520
 - !ruby/object:Gem::Dependency
   name: thin
-  requirement: &195839400 !ruby/object:Gem::Requirement
+  requirement: &185307100 !ruby/object:Gem::Requirement
     none: false
     requirements:
     - - ! '>='
@@ -87,7 +87,7 @@ dependencies:
         version: '0'
   type: :runtime
   prerelease: false
-  version_requirements: *195839400
+  version_requirements: *185307100
 description: Ruby gem with Thor script to manage MySQL and PostgreSQL replication
 email:
 - lydianblues@gmail.com
@@ -107,6 +107,7 @@ files:
 - config/mysql.cnf
 - config/servers.yml
 - lib/repctl.rb
+- lib/repctl/client.rb
 - lib/repctl/color.rb
 - lib/repctl/helpers.rb
 - lib/repctl/mysql_admin.rb