nginx_utils 0.0.2 → 0.0.3

data/lib/nginx_utils/logrotate.rb

@@ -0,0 +1,187 @@
+# coding: utf-8
+
+module NginxUtils
+  class Logrotate
+    attr_accessor :logger, :rename_logs, :delete_logs
+
+    def initialize(options={})
+      # Debug
+      #   debug: false => Not execute. Only output of logs to STDOUT.
+      #   debug: true => Execute rotate logs Nginx. (default)
+      if options[:debug]
+        debug
+      else
+        @execute = true
+
+        # Script log file
+        #   not specified => /tmp/nginx_rotate.log (default)
+        #   script_log: false => Not output of logs.
+        #   script_log: "/path/to/nginx_rotate.log" => /path/to/nginx_rotate.log
+        #   script_log: STDOUT => STDOUT
+        set_logger options[:script_log]
+
+        # Script log level
+        #   not specified => Logger::DEBUG (default)
+        #   log_level: [:fatal|:error|:info|:warn]
+        set_log_level options[:log_level]
+      end
+
+      # Target logs
+      # Log of rename target is "#{root_dir}/**/#{target_logs}"
+      # Log of delete target is "#{root_dir}/**/#{target_logs}.*"
+      # Default parameters are as follows:
+      #   - root_dir => /usr/local/nginx
+      #   - target_logs => *.log
+      @root_dir = options[:root_dir] || "/usr/local/nginx"
+      @target_logs = options[:target_logs] || "*.log"
+      @rename_logs = Dir.glob(File.join(@root_dir, "**", @target_logs))
+      @delete_logs = Dir.glob(File.join(@root_dir, "**", "#{@target_logs}.*"))
+
+      # Rename prefix
+      # Log of rename target to add the prefix.
+      # File name of the renamed after: "#{filename}.#{prefix}"
+      # Current time default. (YYYYmmddHHMMSS)
+      @prefix = options[:prefix] || Time.now.strftime("%Y%m%d%H%M%S")
+
+      # Retention period
+      # Delete log last modification time has passed the retention period.
+      # Specified unit of day.
+      dates = options[:retention] || 90
+      @retention = Time.now - (dates.to_i * 3600 * 24)
+
+      # PID file of Nginx
+      # The default is "#{root_dir}/logs/nginx.pid".
+      @pid_file = options[:pid_file] || File.join(@root_dir, "logs", "nginx.pid")
+    end
+
+    def config(options={})
+      # Debug
+      unless options[:debug].nil?
+        if options[:debug]
+          debug
+        else
+          debug false
+        end
+      end
+
+      # Script log file
+      unless options[:script_log].nil?
+        set_logger options[:script_log]
+      end
+
+      # Script log level
+      unless options[:log_level].nil?
+        set_log_level options[:log_level]
+      end
+
+      # Target logs
+      reglog = false
+      unless options[:root_dir].nil?
+        @root_dir = options[:root_dir]
+        reglob = true
+      end
+
+      unless options[:target_logs].nil?
+        @target_logs = options[:target_logs]
+        reglob = true
+      end
+
+      if reglob
+        @rename_logs = Dir.glob(File.join(@root_dir, "**", @target_logs))
+        @delete_logs = Dir.glob(File.join(@root_dir, "**", "#{@target_logs}.*"))
+      end
+
+      # Rename prefix
+      unless options[:prefix].nil?
+        @prefix = options[:prefix]
+      end
+
+      # Retention period
+      unless options[:retention].nil?
+        @retention = Time.now - (options[:retention].to_i * 3600 * 24)
+      end
+
+      # PID file of Nginx
+      unless options[:pid_file].nil?
+        @pid_file = options[:pid_file]
+      end
+    end
+
+    def rename
+      @rename_logs.each do |log|
+        after = "#{log}.#{@prefix}"
+        if File.exists? after
+          @logger.warn "File already exists: #{after}" if @logger
+        else
+          File.rename(log, after) if @execute
+          @logger.debug "Rename log file: #{log} to #{after}" if @logger
+        end
+      end
+    end
+
+    def delete
+      @delete_logs.each do |log|
+        if File.stat(log).mtime < @retention
+          File.unlink(log) if @execute
+          @logger.debug "Delete log file: #{log}" if @logger
+        end
+      end
+    end
+
+    def restart
+      if File.exists? @pid_file
+        cmd = "kill -USR1 `cat #{@pid_file}`"
+        @logger.debug "Nginx restart command: #{cmd}" if @logger
+        if @execute
+          if system(cmd)
+            @logger.info "Nginx restart is successfully!" if @logger
+          else
+            @logger.error "Nginx restart failed!" if @logger
+            raise "Nginx restart failed!" if @logger == false
+          end
+        end
+      else
+        @logger.warn "Pid file is not found. not restart nginx. (#{@pid_file})" if @logger
+      end
+    end
+
+    def execute
+      @logger.info "Nginx logrotate is started!" if @logger
+      rename
+      delete
+      @logger.info "Nginx logrotate is successfully!" if @logger
+      restart
+    end
+
+    private
+    def debug(set=true)
+      if set
+        @execute = false
+        @logger = Logger.new(STDOUT)
+      else
+        @execute = true
+      end
+    end
+
+    def set_logger(log)
+      case log
+      when nil then @logger = Logger.new("/tmp/nginx_rotate.log")
+      when false then @logger = false
+      else @logger = Logger.new(log)
+      end
+    end
+
+    def set_log_level(level)
+      if @logger
+        case level
+        when :fatal then @logger.level = Logger::FATAL
+        when :error then @logger.level = Logger::ERROR
+        when :warn then @logger.level = Logger::WARN
+        when :info then @logger.level = Logger::INFO
+        when :debug then @logger.level = Logger::DEBUG
+        else @logger.level = Logger::DEBUG
+        end
+      end
+    end
+  end
+end

data/lib/nginx_utils/status.rb

@@ -0,0 +1,28 @@
+# coding: utf-8
+
+module NginxUtils
+  module Status
+    class << self
+      def get(options={host: "localhost", port: 80, path: "/nginx_status"})
+        begin
+          req = Net::HTTP::Get.new(options[:path])
+          res = Net::HTTP.start(options[:host], options[:port]){|http| http.request(req)}
+          status = res.body.split("\n")
+          server = status[2].split.map{|i| i.to_i}
+          rww = status[3].split.select{|i| /^[0-9]+$/ =~ i}.map{|i| i.to_i}
+          {
+            active_connection: status[0].split(":").last.gsub(/\s/, "").to_i,
+            accepts: server[0],
+            handled: server[1],
+            requests: server[2],
+            reading: rww[0],
+            writing: rww[1],
+            waiting: rww[2]
+          }
+        rescue
+          raise "Nginx status get failed"
+        end
+      end
+    end
+  end
+end

data/lib/nginx_utils/version.rb

@@ -1,3 +1,3 @@
 module NginxUtils
-  VERSION = "0.0.2"
+  VERSION = "0.0.3"
 end

data/lib/nginx_utils.rb

@@ -1,190 +1,8 @@
 # coding: utf-8
 
-require "nginx_utils/version"
 require "logger"
+require "net/http"
 
-module NginxUtils
-  class Logrotate
-    attr_accessor :logger, :rename_logs, :delete_logs
-
-    def initialize(options={})
-      # Debug
-      #   debug: false => Not execute. Only output of logs to STDOUT.
-      #   debug: true => Execute rotate logs Nginx. (default)
-      if options[:debug]
-        debug
-      else
-        @execute = true
-
-        # Script log file
-        #   not specified => /tmp/nginx_rotate.log (default)
-        #   script_log: false => Not output of logs.
-        #   script_log: "/path/to/nginx_rotate.log" => /path/to/nginx_rotate.log
-        #   script_log: STDOUT => STDOUT
-        set_logger options[:script_log]
-
-        # Script log level
-        #   not specified => Logger::DEBUG (default)
-        #   log_level: [:fatal|:error|:info|:warn]
-        set_log_level options[:log_level]
-      end
-
-      # Target logs
-      # Log of rename target is "#{root_dir}/**/#{target_logs}"
-      # Log of delete target is "#{root_dir}/**/#{target_logs}.*"
-      # Default parameters are as follows:
-      #   - root_dir => /usr/local/nginx
-      #   - target_logs => *.log
-      @root_dir = options[:root_dir] || "/usr/local/nginx"
-      @target_logs = options[:target_logs] || "*.log"
-      @rename_logs = Dir.glob(File.join(@root_dir, "**", @target_logs))
-      @delete_logs = Dir.glob(File.join(@root_dir, "**", "#{@target_logs}.*"))
-
-      # Rename prefix
-      # Log of rename target to add the prefix.
-      # File name of the renamed after: "#{filename}.#{prefix}"
-      # Current time default. (YYYYmmddHHMMSS)
-      @prefix = options[:prefix] || Time.now.strftime("%Y%m%d%H%M%S")
-
-      # Retention period
-      # Delete log last modification time has passed the retention period.
-      # Specified unit of day.
-      dates = options[:retention] || 90
-      @retention = Time.now - (dates.to_i * 3600 * 24)
-
-      # PID file of Nginx
-      # The default is "#{root_dir}/logs/nginx.pid".
-      @pid_file = options[:pid_file] || File.join(@root_dir, "logs", "nginx.pid")
-    end
-
-    def config(options={})
-      # Debug
-      unless options[:debug].nil?
-        if options[:debug]
-          debug
-        else
-          debug false
-        end
-      end
-
-      # Script log file
-      unless options[:script_log].nil?
-        set_logger options[:script_log]
-      end
-
-      # Script log level
-      unless options[:log_level].nil?
-        set_log_level options[:log_level]
-      end
-
-      # Target logs
-      reglog = false
-      unless options[:root_dir].nil?
-        @root_dir = options[:root_dir]
-        reglob = true
-      end
-
-      unless options[:target_logs].nil?
-        @target_logs = options[:target_logs]
-        reglob = true
-      end
-
-      if reglob
-        @rename_logs = Dir.glob(File.join(@root_dir, "**", @target_logs))
-        @delete_logs = Dir.glob(File.join(@root_dir, "**", "#{@target_logs}.*"))
-      end
-
-      # Rename prefix
-      unless options[:prefix].nil?
-        @prefix = options[:prefix]
-      end
-
-      # Retention period
-      unless options[:retention].nil?
-        @retention = Time.now - (options[:retention].to_i * 3600 * 24)
-      end
-
-      # PID file of Nginx
-      unless options[:pid_file].nil?
-        @pid_file = options[:pid_file]
-      end
-    end
-
-    def rename
-      @rename_logs.each do |log|
-        after = "#{log}.#{@prefix}"
-        if File.exists? after
-          @logger.warn "File already exists: #{after}" if @logger
-        else
-          File.rename(log, after) if @execute
-          @logger.debug "Rename log file: #{log} to #{after}" if @logger
-        end
-      end
-    end
-
-    def delete
-      @delete_logs.each do |log|
-        if File.stat(log).mtime < @retention
-          File.unlink(log) if @execute
-          @logger.debug "Delete log file: #{log}" if @logger
-        end
-      end
-    end
-
-    def restart
-      if File.exists? @pid_file
-        cmd = "kill -USR1 `cat #{@pid_file}`"
-        @logger.debug "Nginx restart command: #{cmd}" if @logger
-        if @execute
-          if system(cmd)
-            @logger.info "Nginx restart is successfully!" if @logger
-          else
-            @logger.error "Nginx restart failed!" if @logger
-            raise "Nginx restart failed!" if @logger == false
-          end
-        end
-      else
-        @logger.warn "Pid file is not found. not restart nginx. (#{@pid_file})" if @logger
-      end
-    end
-
-    def execute
-      @logger.info "Nginx logrotate is started!" if @logger
-      rename
-      delete
-      @logger.info "Nginx logrotate is successfully!" if @logger
-      restart
-    end
-
-    private
-    def debug(set=true)
-      if set
-        @execute = false
-        @logger = Logger.new(STDOUT)
-      else
-        @execute = true
-      end
-    end
-
-    def set_logger(log)
-      case log
-      when nil then @logger = Logger.new("/tmp/nginx_rotate.log")
-      when false then @logger = false
-      else @logger = Logger.new(log)
-      end
-    end
-
-    def set_log_level(level)
-      if @logger
-        case level
-        when :fatal then @logger.level = Logger::FATAL
-        when :error then @logger.level = Logger::ERROR
-        when :warn then @logger.level = Logger::WARN
-        when :info then @logger.level = Logger::INFO
-        when :debug then @logger.level = Logger::DEBUG
-        else @logger.level = Logger::DEBUG
-        end
-      end
-    end
-  end
-end
+require "nginx_utils/version"
+require "nginx_utils/logrotate"
+require "nginx_utils/status"

data/spec/nginx_utils/status_spec.rb

@@ -0,0 +1,23 @@
+# coding: utf-8
+
+require "spec_helper"
+
+describe NginxUtils do
+  describe "Status" do
+    describe ".get" do
+      it "should get status" do
+        body = "Active connections: 1 \nserver accepts handled requests\n 4 5 51 \nReading: 1 Writing: 3 Waiting: 2 \n"
+        status = {active_connection: 1, accepts: 4, handled: 5, requests: 51, reading: 1, writing: 3, waiting: 2}
+        response = OpenStruct.new(body: body)
+        Net::HTTP.should_receive(:start).and_return(response)
+        expect(NginxUtils::Status.get).to eq(status)
+      end
+
+      it "should generate an exception if status get fails" do
+        res = OpenStruct.new(body: "invalid content")
+        Net::HTTP.should_receive(:start).and_return(res)
+        expect(proc{NginxUtils::Status.get}).to raise_error("Nginx status get failed")
+      end
+    end
+  end
+end

data/spec/spec_helper.rb

@@ -1,2 +1,3 @@
 require "fakefs/spec_helpers"
+require "ostruct"
 require "nginx_utils"

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: nginx_utils
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.3
   prerelease: 
 platform: ruby
 authors:
@@ -89,9 +89,12 @@ files:
 - README.md
 - Rakefile
 - lib/nginx_utils.rb
+- lib/nginx_utils/logrotate.rb
+- lib/nginx_utils/status.rb
 - lib/nginx_utils/version.rb
 - nginx_utils.gemspec
-- spec/lib/nginx_utils_spec.rb
+- spec/nginx_utils/logrotate_spec.rb
+- spec/nginx_utils/status_spec.rb
 - spec/spec_helper.rb
 homepage: https://github.com/i2bskn/nginx_utils
 licenses:
@@ -108,7 +111,7 @@ required_ruby_version: !ruby/object:Gem::Requirement
       version: '0'
       segments:
       - 0
-      hash: -2674447921800970947
+      hash: 17147575815869233
 required_rubygems_version: !ruby/object:Gem::Requirement
   none: false
   requirements:
@@ -117,7 +120,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
       segments:
       - 0
-      hash: -2674447921800970947
+      hash: 17147575815869233
 requirements: []
 rubyforge_project: 
 rubygems_version: 1.8.23
@@ -125,5 +128,6 @@ signing_key:
 specification_version: 3
 summary: Nginx utilities
 test_files:
-- spec/lib/nginx_utils_spec.rb
+- spec/nginx_utils/logrotate_spec.rb
+- spec/nginx_utils/status_spec.rb
 - spec/spec_helper.rb