murakumo 0.1.0

data/lib/srv/murakumo_health_checker.rb

@@ -0,0 +1,134 @@
+require 'timeout'
+require 'resolv-replace'
+
+require 'srv/murakumo_health_checker_context'
+require 'misc/murakumo_const'
+
+module Murakumo
+
+  class HealthChecker
+
+    def initialize(name, cloud, logger, options)
+      # 名前は小文字に変換
+      @name = name.downcase
+      @cloud = cloud
+      @logger = logger
+
+      # 各種変数の設定
+      {
+        'interval'  => [30, 1, 300],
+        'timeout'   => [ 5, 1, 300],
+        'healthy'   => [10, 2,  10],
+        'unhealthy' => [ 2, 2,  10],
+      }.each {|key, vals|
+        defval, min, max = vals
+        value = (options[key] || defval).to_i
+        value = min if value < min
+        value = max if value > max
+        instance_variable_set("@#{key}", value)
+      }
+
+      # スクリプトの読み込み
+      @script = options['script']
+      raise "health check script of #{@name} is not found" unless @script
+      @script = File.read(script) if File.exists?(@script)
+    end
+
+    def good
+      if @normal_health
+        @unhealthy_count = 0
+      elsif (@healthy_count += 1) >= @healthy
+        toggle_health
+      end
+    end
+
+    def bad
+      if not @normal_health
+        @healthy_count = 0
+      elsif (@unhealthy_count += 1) >= @unhealthy
+        toggle_health
+      end
+    end
+
+    def toggle_health
+      @normal_health = !@normal_health
+      activity = (@normal_health ? ACTIVE : INACTIVE)
+
+      @cloud.gossip.transaction do
+        @cloud.gossip.data.each do |i|
+          # 名前の一致するデータを更新
+          i[3] = activity if i[0] == @name
+        end
+      end
+
+      @cloud.db.execute(<<-EOS, activity, @cloud.address, @name)
+        UPDATE records SET activity = ?
+        WHERE ip_address = ? AND name = ?
+      EOS
+
+      @healthy_count = 0
+      @unhealthy_count = 0
+
+      status = @normal_health ? 'healthy' : 'unhealthy'
+      @logger.info("health condition changed: #{@name}: #{status}")
+    end
+
+    def start
+      # 各種変数は初期状態にする
+      @alive = true
+      @normal_health = true
+      @healthy_count = 0
+      @unhealthy_count = 0
+
+      # 既存のスレッドは破棄
+      if @thread and @thread.alive?
+        begin
+          @thread.kill
+        rescue ThreadError
+        end
+      end
+
+      @thread = Thread.start {
+        healthy = 0
+        unhealthy = 0
+
+        begin
+          while @alive
+            retval = nil
+
+            begin
+              retval = timeout(@timeout) {
+                HealthCheckerContext.new.instance_eval(@script)
+              }
+            rescue Timeout::Error
+              retval = false
+            end
+
+            status = retval == true ? 'good' : retval == false ? 'bad' : '-'
+            @logger.debug("result of a health check: #{@name}: #{status}")
+
+            if retval == true
+              good
+            elsif retval == false
+              bad
+            end
+
+            sleep @interval
+          end # while
+        rescue Exception => e
+          message = (["#{e.class}: #{e.message}"] + (e.backtrace || [])).join("\n\tfrom ")
+          @logger.error("#{@name}: #{message}")
+        end # begin
+      }
+    end
+
+    def stop
+      @alive = false
+    end
+
+    def alive?
+      @alive
+    end
+  end # HealthChecker
+
+end # Murakumo

data/lib/srv/murakumo_health_checker_context.rb

@@ -0,0 +1,33 @@
+require 'net/http'
+require 'socket'
+
+require 'misc/murakumo_const'
+
+module Murakumo
+
+  # ヘルスチェックのコンテキスト
+  class HealthCheckerContext
+
+    # TCPチェッカー
+    def tcp_check(port, host = '127.0.0.1')
+      s = TCPSocket.new(host, port)
+      s.close
+      return true
+    rescue Exception
+      return false
+    end
+
+    # HTTPチェッカー
+    def http_get(path, statuses = [200], host = '127.0.0.1', port = 80)
+      res = Net::HTTP.start('127.0.0.1', 80) do |http|
+        http.get(path)
+      end
+
+      statuses.include?(res.code.to_i)
+    rescue Exception
+      return false
+    end
+
+  end # HealthCheckerContext
+
+end # Murakumo

data/lib/srv/murakumo_server.rb

@@ -0,0 +1,102 @@
+require 'drb/drb'
+require 'fileutils'
+require 'rexec'
+require 'rexec/daemon'
+require 'rubydns'
+require 'socket'
+
+require 'srv/murakumo_cloud'
+
+BasicSocket.do_not_reverse_lookup = true
+
+module Murakumo
+
+  # RExecに依存しているのでこんな設計に…
+  class Server < RExec::Daemon::Base
+
+    # クラスメソッドを定義
+    class << self
+
+      def init(options)
+        # クラスインスタンス変数を使わないこと
+        @@options = options
+        @@cloud = Cloud.new(options)
+      end
+
+      def pid_directory=(path)
+        @@pid_directory = path
+      end
+
+      def run
+        RubyDNS.run_server(:listen => [[:udp, @@options[:dns_address], @@options[:dns_port]]]) do
+          # RubyDNS::Serverのコンテキスト
+          @logger = @@options[:logger] if @@options[:logger]
+
+          on(:start) do
+            if @@options[:socket]
+              # ServerクラスをDRuby化
+              DRb.start_service("drbunix:#{@@options[:socket]}", @@cloud)
+              at_exit { FileUtils.rm_f(@@options[:socket]) }
+            end
+
+            # HUPでログをローテート
+            if @@options[:logger]
+              trap(:HUP) do
+                if logger = @@options[:logger]
+                  logdev = logger.instance_variable_get(:@logdev)
+
+                  if (dev = logdev.dev).kind_of?(File)
+                    path = dev.path
+                    mutex = logdev.instance_variable_get(:@mutex)
+
+                    mutex.synchronize do
+                      dev.reopen(path, 'a')
+                      dev.sync = true
+                    end
+                  end
+                end
+              end
+            end
+
+            # ゴシッププロトコルを開始
+            @@cloud.start
+          end
+
+          # look up A record
+          match(@@cloud.method(:address_exist?), :A) do |transaction|
+            records = @@cloud.lookup_addresses(transaction.name)
+
+            # 先頭のAレコードを決定
+            max_ip_num = [records.length, @@options[:max_ip_num]].min
+            first_index = rand(max_ip_num);
+
+            # Aレコードを返す
+            (records + records).slice(first_index, max_ip_num).each do |r|
+              address, ttl = r
+              transaction.respond!(address, :ttl => ttl)
+            end
+          end # match
+
+          # look up PTR record
+          match(@@cloud.method(:name_exist?), :PTR) do |transaction|
+            name, ttl = @@cloud.lookup_name(transaction.name)
+            transaction.respond!(Resolv::DNS::Name.create("#{name}."), :ttl => ttl)
+          end
+
+          if @@options[:resolver]
+            otherwise do |transaction|
+              transaction.passthrough!(@@options[:resolver])
+            end
+          end
+        end # RubyDNS.run_server
+      end # run
+
+      def shutdown
+        @@cloud.stop
+      end
+
+    end # class << self
+
+  end # Server
+
+end # Murakumo

metadata

@@ -0,0 +1,142 @@
+--- !ruby/object:Gem::Specification 
+name: murakumo
+version: !ruby/object:Gem::Version 
+  hash: 27
+  prerelease: 
+  segments: 
+  - 0
+  - 1
+  - 0
+  version: 0.1.0
+platform: ruby
+authors: 
+- winebarrel
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2011-11-16 00:00:00 Z
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: rubydns
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ~>
+      - !ruby/object:Gem::Version 
+        hash: 21
+        segments: 
+        - 0
+        - 3
+        - 3
+        version: 0.3.3
+  type: :runtime
+  version_requirements: *id001
+- !ruby/object:Gem::Dependency 
+  name: rgossip2
+  prerelease: false
+  requirement: &id002 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ~>
+      - !ruby/object:Gem::Version 
+        hash: 17
+        segments: 
+        - 0
+        - 1
+        - 5
+        version: 0.1.5
+  type: :runtime
+  version_requirements: *id002
+- !ruby/object:Gem::Dependency 
+  name: optopus
+  prerelease: false
+  requirement: &id003 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ~>
+      - !ruby/object:Gem::Version 
+        hash: 9
+        segments: 
+        - 0
+        - 1
+        - 9
+        version: 0.1.9
+  type: :runtime
+  version_requirements: *id003
+- !ruby/object:Gem::Dependency 
+  name: sqlite3-ruby
+  prerelease: false
+  requirement: &id004 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ~>
+      - !ruby/object:Gem::Version 
+        hash: 21
+        segments: 
+        - 1
+        - 2
+        - 5
+        version: 1.2.5
+  type: :runtime
+  version_requirements: *id004
+description: 
+email: sgwr_dts@yahoo.co.jp
+executables: 
+- murakumo
+- mrkmctl
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- README
+- bin/mrkmctl
+- bin/murakumo
+- lib/cli/murakumo_options.rb
+- lib/cli/mrkmctl.rb
+- lib/cli/mrkmctl_options.rb
+- lib/cli/murakumo.rb
+- lib/srv/murakumo_health_checker_context.rb
+- lib/srv/murakumo_server.rb
+- lib/srv/murakumo_health_checker.rb
+- lib/srv/murakumo_cloud.rb
+- lib/misc/murakumo_const.rb
+- etc/murakumo.yml.sample
+- etc/murakumo.server
+homepage: https://bitbucket.org/winebarrel/murakumo
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.8.1
+signing_key: 
+specification_version: 3
+summary: Murakumo is the internal DNS server which manages name information using a gossip protocol.
+test_files: []
+