naminori 0.1.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: fc8b2a9badbab70707427fa9729edd14998a5079
+  data.tar.gz: d2f6668fd2bb6b4490963bc5b1e9aa785a2d8585
+SHA512:
+  metadata.gz: dfbda96472f946e34904dcdf2579826fbdd594d87e8029116f3ef032e4384f574ac707b67d3d19c574dcfe6192da113971ded3682b69fb9b00477f6a5112030d
+  data.tar.gz: 237e0c47068e0b86d9d9017ad9cd6366196d60777148f95e200e92ea14a010ec43b5132c7586d8a4e795cb53a82af6e55f0c97461d01cafe7f55c1e9db5e0043

data/.gitignore

@@ -0,0 +1,9 @@
+/.bundle/
+/.yardoc
+/Gemfile.lock
+/_yardoc/
+/coverage/
+/doc/
+/pkg/
+/spec/reports/
+/tmp/

data/.travis.yml

@@ -0,0 +1,8 @@
+language: ruby
+rvm:
+- 2.0.0
+- 2.1.6
+- 2.2.2
+addons:
+  code_climate:
+    repo_token: 47135cdaac4078905036e9ec2506567cd244bd711794ed292b8f36fee6fa1f22

data/Gemfile

@@ -0,0 +1,5 @@
+source 'https://rubygems.org'
+
+# Specify your gem's dependencies in naminori.gemspec
+gemspec
+gem "codeclimate-test-reporter", group: :test, require: nil

data/LICENSE.txt

@@ -0,0 +1,21 @@
+The MIT License (MIT)
+
+Copyright (c) 2015 kazuhiko yamahsita
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in
+all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+THE SOFTWARE.

data/README.md

@@ -0,0 +1,91 @@
+# Naminori
+[![Build Status](https://travis-ci.org/pyama86/Naminori.svg)](https://travis-ci.org/pyama86/Naminori)
+[![Code Climate](https://codeclimate.com/github/pyama86/Naminori/badges/gpa.svg)](https://codeclimate.com/github/pyama86/Naminori)
+[![Test Coverage](https://codeclimate.com/github/pyama86/Naminori/badges/coverage.svg)](https://codeclimate.com/github/pyama86/Naminori/coverage)
+
+Library to manage the load balancer using the Serf
+
+## Installation
+
+```ruby
+gem 'naminori'
+```
+
+Or install it yourself as:
+
+    $ gem install naminori
+
+## Usage
+
+### exemple
+
+####  envent_example.rb
+```ruby
+#! /usr/bin/env ruby
+require 'rubygems'
+require 'naminori'
+
+notifier_options = {
+  webhook_url: "https://hooks.slack.com/services/XXXXXX",
+  channel: "#pyama"
+}
+
+service_options = {
+  vip:"192.168.77.9",
+  role: "dns",
+  notifier: Naminori::Notifier.get_notifier("slack" ,notifier_options)
+}
+
+case
+when Naminori::Serf.role?("dns")
+  Naminori::Service.event("dns", "lvs", service_options)
+when Naminori::Serf.role?("lb")
+  Naminori::Lb.health_check("dns", "lvs",service_options)
+end
+
+```
+
+#### health_check_example.rb
+```ruby
+#! /usr/bin/env ruby
+require 'rubygems'
+require 'naminori'
+
+service_options = { vip:"192.168.77.9", role: "dns" }
+
+Naminori::Lb.health_check("dns", "lvs", service_options)
+```
+```zsh
+# crontab -l
+# Chef Name: health_check
+PATH=/usr/local/sbin:/usr/local/bin:/sbin:/bin:/usr/sbin:/usr/bin:/root/bin
+* * * * * for i in `seq 0 10 59`;do (sleep ${i}; /opt/serf/event_handlers/health_check_example.rb)& done;
+```
+
+
+
+
+#### Service Event
+```
+Naminori::Service.event(service_name, lb_name, options)
+```
+* service_name:
+  dns or http
+* lb_name:
+  lvs
+* options(dns_default)
+```
+        {
+          role: "dns",              # role name
+          port: "53",               # service port
+          protocol: ["udp", "tcp"], # protocol(array)
+          vip: "192.168.77.9",      # service vip
+          method: "gateway",        # lvs_method gateway/ip/nat
+          query: "pepabo.com",      # health_check_query
+          retry: 3,                 # health_check_retry_count
+          timeout: 3,               # health_check_time_out
+          notifier: nil             # notifier
+        }
+```
+## Author
+* pyama

data/Rakefile

@@ -0,0 +1,6 @@
+require "bundler/gem_tasks"
+require "rspec/core/rake_task"
+ENV['CODECLIMATE_REPO_TOKEN'] = "47135cdaac4078905036e9ec2506567cd244bd711794ed292b8f36fee6fa1f22"
+ENV['PEC_TEST'] = "test"
+RSpec::Core::RakeTask.new("spec")
+task :default => :spec

data/lib/naminori.rb

@@ -0,0 +1,20 @@
+require "naminori/version"
+require 'naminori/serf'
+require 'naminori/lb'
+require 'naminori/lb/base'
+require 'naminori/lb/lvs'
+require 'naminori/service'
+require 'naminori/service/base'
+require 'naminori/service/configure'
+require 'naminori/service/dns'
+require 'naminori/notifier'
+require 'naminori/notifier/base'
+require 'naminori/notifier/configure'
+require 'naminori/notifier/slack'
+require 'resolv'
+require 'slack-notifier'
+
+
+module Naminori
+  # Your code goes here...
+end

data/lib/naminori/lb.rb

@@ -0,0 +1,31 @@
+#! /usr/bin/env ruby
+module Naminori
+  class Lb
+    class << self
+      def health_check(service_name, lb_name, options={})
+        service = Naminori::Service.get_service(service_name).new(options)
+        members = Naminori::Serf.get_alive_member_by_role(service.config.role)
+
+        members.each do |member|
+          ip = member_ip(member)
+          if service.healty?(ip)
+            get_lb(lb_name).add_member(ip, service)
+          elsif service.config.retry.times.all? { sleep 1; !service.healty?(ip) }
+            get_lb(lb_name).delete_member(ip, service)
+          end
+        end if members
+      end
+
+      def get_lb(lb_name)
+        case lb_name
+        when "lvs"
+          Naminori::Lb::Lvs
+        end
+      end
+
+      def member_ip(member)
+        member[:ip]
+      end
+    end
+  end
+end

data/lib/naminori/lb/base.rb

@@ -0,0 +1,15 @@
+#! /usr/bin/env ruby
+module Naminori
+  class Lb
+    class Base
+      class << self
+        def add_member(rip, service)
+          raise "Called abstract method: add_member"
+        end
+        def delete_member(rip, service)
+          raise "Called abstract method: add_member"
+        end
+      end
+    end
+  end
+end

data/lib/naminori/lb/lvs.rb

@@ -0,0 +1,79 @@
+#! /usr/bin/env ruby
+module Naminori
+  class Lb
+    class Lvs < Naminori::Lb::Base
+      class << self
+        def add_member(rip, service)
+          transaction("add", lvs_option(rip, service)) if service.healty?(rip)
+        end
+
+        def delete_member(rip, service)
+          transaction("delete", lvs_option(rip, service))
+        end
+
+        def transaction(type, ops)
+          ops[:protocols].collect do |protocol|
+            merge_option =  ops.merge({ protocol: protocol })
+            if self.send("#{type}?", merge_option) && system("ipvsadm #{command_option(type, merge_option)}")
+              notifier(type, merge_option)
+              true
+            end
+          end.all? {|res| res }
+        end
+
+        def add?(ops)
+          vip_exists?(ops) && !delete?(ops)
+        end
+
+        def delete?(ops)
+          fetch_service(ops).each do |line|
+            return true if line.match(/#{ops[:rip]}:/)
+          end
+          false
+        end
+
+        def vip_exists?(ops)
+          vips = fetch_service(ops)
+          vips.each do |line|
+            return true if line.match(/#{ops[:vip]}:/)
+          end if vips
+          false
+        end
+
+        def fetch_service(ops)
+          `ipvsadm -Ln --#{ops[:protocol]}-service #{ops[:vip]}:#{ops[:port]}`.split("\n")
+        end
+
+        def lvs_option(rip, service)
+          { service: service, vip: service.config.vip, rip: rip, protocols: service.config.protocol, port: service.config.port, method: service.config.method }
+        end
+
+        def command_option(type, ops)
+          case
+          when type == "add"
+            "--#{type}-server --#{ops[:protocol]}-service #{ops[:vip]}:#{ops[:port]} -r #{ops[:rip]}:#{ops[:port]} #{method_option(ops[:method])}"
+          when type == "delete"
+            "--#{type}-server --#{ops[:protocol]}-service #{ops[:vip]}:#{ops[:port]} -r #{ops[:rip]}:#{ops[:port]}"
+          end
+        end
+
+        def method_option(method)
+          case
+          when method == "gateway"
+            "-g"
+          when method == "nat"
+            "-m"
+          when method == "ip"
+            "-i"
+          end
+        end
+
+        def notifier(type, options)
+          message = "#{type} member ip:#{options[:rip]} protocol:#{options[:protocol]} in vip:#{options[:vip]}"
+          puts message
+          options[:service].config.notifier.send(type, message) if options[:service].config.notifier
+        end
+      end
+    end
+  end
+end

data/lib/naminori/naminori.rb

@@ -0,0 +1,11 @@
+#! /usr/bin/env ruby
+require File.dirname(__FILE__) + '/serf'
+require File.dirname(__FILE__) + '/lb'
+require File.dirname(__FILE__) + '/lb/base'
+require File.dirname(__FILE__) + '/lb/lvs'
+require File.dirname(__FILE__) + '/service'
+require File.dirname(__FILE__) + '/service/base'
+require File.dirname(__FILE__) + '/service/configure'
+require File.dirname(__FILE__) + '/service/dns'
+require 'resolv'
+

data/lib/naminori/notifier.rb

@@ -0,0 +1,13 @@
+#! /usr/bin/env ruby
+module Naminori
+  class Notifier
+    class << self
+      def get_notifier(notifier, options={})
+        case notifier
+        when "slack"
+          Naminori::Notifier::Slack.new(options)
+        end
+      end
+    end
+  end
+end

data/lib/naminori/notifier/base.rb

@@ -0,0 +1,22 @@
+#! /usr/bin/env ruby
+module Naminori
+  class Notifier
+    class Base
+      attr_reader :config
+      def initialize(options={})
+        @config = Naminori::Notifier::Configure.new(
+          default_config.merge(options)
+        )
+      end
+
+      def send(type, message)
+        raise "Called abstract method: add_server"
+      end
+
+      def default_config
+        raise "Called abstract method: default_config"
+      end
+
+    end
+  end
+end

data/lib/naminori/notifier/configure.rb

@@ -0,0 +1,13 @@
+#! /usr/bin/env ruby
+module Naminori
+  class Notifier
+    class Configure
+      attr_reader :webhook_url, :channel, :user
+      def initialize(options)
+        @webhook_url = options[:webhook_url]
+        @channel     = options[:channel]
+        @user        = options[:user]
+      end
+    end
+  end
+end

data/lib/naminori/notifier/slack.rb

@@ -0,0 +1,22 @@
+#! /usr/bin/env ruby
+module Naminori
+  class Notifier
+    class Slack < Base
+      def send(type, message)
+        icon type == "add" ? ":white_check_mark:" : ":no_entry_sign:"
+        notifier = ::Slack::Notifier.new(config.webhook_url, { channel: config.channel, username: config.user})
+        notifier.ping  icon + message, icon_emoji: ":sparkle:"
+      rescue => e
+        p e
+      end
+
+      def default_config
+        {
+          webhook_url: "",
+          channel: "general",
+          user: "naminori-notifier"
+        }
+      end
+    end
+  end
+end

data/lib/naminori/serf.rb

@@ -0,0 +1,54 @@
+#! /usr/bin/env ruby
+module Naminori
+  class Serf
+    @@event = nil
+    class << self
+      def gets
+        @@event ||= STDIN.gets.chomp.match(/(?<node>.+?)\t(?<ip>.+?)\t(?<role>.+?)\t/)
+      end
+
+      def event_message
+        puts "event:#{ENV['SERF_EVENT']} value:#{get}"
+      end
+
+      def join?
+        ENV['SERF_EVENT'] == 'member-join'
+      end
+
+      def leave?
+        ENV['SERF_EVENT'] == 'member-leave'
+      end
+
+      def failed?
+        ENV['SERF_EVENT'] == 'member-failed'
+      end
+
+      def role?(role)
+        gets[:role] == role
+      end
+
+      def get_alive_member_by_role(role)
+        members.map do |line|
+          member = parse_member(line)
+          target_role?(member, role) && alive?(member) ? member : nil
+        end.compact if members
+      end
+
+      def target_role?(member, role)
+        member[:role] == role
+      end
+
+      def alive?(member)
+        member[:status] == "alive"
+      end
+
+      def members
+        `serf members`.split("\n")
+      end
+
+      def parse_member(member)
+        member.match(/(?<node>.+?)\s+(?<ip>.+?):[0-9]+\s+(?<status>\w+)+\s+role=(?<role>\w+)/)
+      end
+    end
+  end
+end

data/lib/naminori/service.rb

@@ -0,0 +1,17 @@
+#! /usr/bin/env ruby
+module Naminori
+  class Service
+    class << self
+      def event(service_name, lb_name, options={})
+        get_service(service_name).event(lb_name, options)
+      end
+
+      def get_service(service_name)
+        case service_name
+        when "dns"
+          Naminori::Service::Dns
+        end
+      end
+    end
+  end
+end

data/lib/naminori/service/base.rb

@@ -0,0 +1,36 @@
+#! /usr/bin/env ruby
+module Naminori
+  class Service
+    class Base
+      attr_reader :config
+      class << self
+        def event(lb_name, options)
+          case
+          when Naminori::Serf.join?
+            Naminori::Lb.get_lb(lb_name).add_member(event_ip, self.new(options))
+          when Naminori::Serf.leave? || Naminori::Serf.failed?
+            Naminori::Lb.get_lb(lb_name).delete_member(event_ip, self.new(options))
+          end
+        end
+
+        def event_ip
+          Naminori::Serf.gets[:ip]
+        end
+      end
+
+      def initialize(options={})
+        @config = Naminori::Service::Configure.new(
+          default_config.merge(options)
+        )
+      end
+
+      def healty?(ip)
+        raise "Called abstract method: healty?"
+      end
+
+      def default_config
+        raise "Called abstract method: default_config"
+      end
+    end
+  end
+end

data/lib/naminori/service/configure.rb

@@ -0,0 +1,19 @@
+#! /usr/bin/env ruby
+module Naminori
+  class Service
+    class Configure
+      attr_reader :role, :port, :protocol, :vip, :method, :query, :retry, :timeout, :notifier
+      def initialize(options)
+        @role     = options[:role]
+        @port     = options[:port]
+        @protocol = options[:protocol]
+        @vip      = options[:vip]
+        @method   = options[:method]
+        @query    = options[:query]
+        @retry    = options[:retry]
+        @timeout  = options[:timeout]
+        @notifier = options[:notifier]
+      end
+    end
+  end
+end

data/lib/naminori/service/dns.rb

@@ -0,0 +1,31 @@
+#! /usr/bin/env ruby
+module Naminori
+  class Service
+    class Dns < Naminori::Service::Base
+
+      def healty?(ip)
+        dns = Resolv::DNS.new(:nameserver => ip )
+        dns.timeouts = config.timeout
+        begin
+          dns.getaddress(@config.query)
+        rescue => e
+          false
+        end
+      end
+
+      def default_config
+        {
+          role: "dns",
+          port: "53",
+          protocol: ["udp", "tcp"],
+          vip: "192.168.77.9",
+          method: "nat",
+          query: "pepabo.com",
+          retry: 3,
+          timeout: 3,
+          notifier: nil
+        }
+      end
+    end
+  end
+end

data/lib/naminori/version.rb

@@ -0,0 +1,3 @@
+module Naminori
+  VERSION = "0.1.1"
+end

data/naminori.gemspec

@@ -0,0 +1,24 @@
+# coding: utf-8
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'naminori/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = "naminori"
+  spec.version       = Naminori::VERSION
+  spec.authors       = ["kazuhiko yamahsita"]
+  spec.email         = ["pyama@pepabo.com"]
+
+  spec.summary       = %q{LoadBarancer library on Serf.}
+  spec.description   = %q{Loadbarancer library on Serf.}
+  spec.homepage      = "http://ten-snapon.com"
+  spec.license       = "MIT"
+
+  spec.files         = `git ls-files -z`.split("\x0").reject { |f| f.match(%r{^(test|spec|features)/}) }
+  spec.require_paths = ["lib"]
+
+  spec.add_dependency 'slack-notifier', "~> 1.2"
+  spec.add_development_dependency "bundler", "~> 1.9"
+  spec.add_development_dependency "rake", "~> 10.0"
+  spec.add_development_dependency "rspec", "~> 3.2"
+end

metadata

@@ -0,0 +1,122 @@
+--- !ruby/object:Gem::Specification
+name: naminori
+version: !ruby/object:Gem::Version
+  version: 0.1.1
+platform: ruby
+authors:
+- kazuhiko yamahsita
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2015-06-21 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: slack-notifier
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.2'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.2'
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.9'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.9'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.2'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.2'
+description: Loadbarancer library on Serf.
+email:
+- pyama@pepabo.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- ".travis.yml"
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- lib/naminori.rb
+- lib/naminori/lb.rb
+- lib/naminori/lb/base.rb
+- lib/naminori/lb/lvs.rb
+- lib/naminori/naminori.rb
+- lib/naminori/notifier.rb
+- lib/naminori/notifier/base.rb
+- lib/naminori/notifier/configure.rb
+- lib/naminori/notifier/slack.rb
+- lib/naminori/serf.rb
+- lib/naminori/service.rb
+- lib/naminori/service/base.rb
+- lib/naminori/service/configure.rb
+- lib/naminori/service/dns.rb
+- lib/naminori/version.rb
+- naminori.gemspec
+homepage: http://ten-snapon.com
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.4.6
+signing_key: 
+specification_version: 4
+summary: LoadBarancer library on Serf.
+test_files: []