jvmargs 0.0.2

data/lib/jvmargs.rb

@@ -0,0 +1,12 @@
+
+require 'jvmargs/args'
+require 'jvmargs/directive'
+require 'jvmargs/standard'
+require 'jvmargs/nonstandard'
+require 'jvmargs/unstable'
+require 'jvmargs/rules'
+require 'jvmargs/util'
+
+module JVMArgs
+  Types = [:directive, :nonstandard, :standard, :unstable ]
+end

data/lib/jvmargs/args.rb

@@ -0,0 +1,117 @@
+module JVMArgs
+  class Args
+    
+    def initialize(*initial_args,&block)
+      @args = Hash.new
+      @rules = JVMArgs::Rules.new
+      Types.each {|type| @args[type] = Hash.new }
+      server_arg = JVMArgs::Standard.new("-server")
+      @args[:standard][server_arg.key] = server_arg
+      set_default_heap_size
+      # in case user passed in an array
+      initial_args.flatten!  
+      parse_args(initial_args) unless initial_args.empty?
+      self.instance_exec &block if block
+    end
+
+    def [](key)
+      @args[key]
+    end
+
+    def add(*args)
+      args.flatten!
+      parse_args(args)
+    end
+    
+    def process_rules
+      @rules.rules.each do |rule|
+       @rules.send(rule, @args)
+      end
+    end
+    
+    def parse_args(args)
+      args.each do |arg|
+        type = nil
+        jvm_arg = case arg
+                  when /^-?XX.*/
+                    type = :unstable
+                    JVMArgs::Unstable.new(arg)
+                  when /^-?X.*/
+                    type = :nonstandard
+                    JVMArgs::NonStandard.new(arg)
+                  when /^-?D.*/
+                    type = :directive
+                    JVMArgs::Directive.new(arg)
+                  else
+                    type = :standard
+                    JVMArgs::Standard.new(arg)
+                  end
+        @args[type][jvm_arg.key] = jvm_arg
+      end
+      process_rules
+    end
+
+    def set_default_heap_size
+      if defined? node
+        total_ram = node['memory']['total'].sub(/kB/, '')
+      else
+        total_ram = (JVMArgs::Util.get_system_ram_m.sub(/M/,'').to_i * 0.4).to_i
+      end
+      @args[:nonstandard]['Xmx'] = JVMArgs::NonStandard.new("Xmx#{total_ram}M")
+    end
+    
+    def to_s
+      process_rules
+      args_str = ""
+      Types.each do |type|
+        type_str = @args[type].map {|k,v| v.to_s }
+        args_str << " " + type_str.join(' ') + " "
+      end
+      args_str
+    end
+
+    def parse_named_args(named_args)
+      named_args.each do |k,v|
+        case k.to_sym
+        when :jmx
+          add_default_jmx
+        end
+      end
+    end
+
+    def jmx(boolean)
+      if boolean
+        [
+         "-Djava.rmi.server.hostname=127.0.0.1",
+         "-Dcom.sun.management.jmxremote",
+         "-Dcom.sun.management.jmxremote",
+         "-Dcom.sun.management.jmxremote.port=9000",
+         "-Dcom.sun.management.jmxremote.authenticate=false",
+         "-Dcom.sun.management.jmxremote.ssl=false"
+        ].each do |arg|
+          directive = JVMArgs::Directive.new arg
+          @args[:directive][directive.key] = directive
+        end
+      end
+    end
+
+    def heap_size(percentage)
+      percentage =~ /([0-9]+)%/
+      percent = $1.to_i * 0.01
+      if percent > 1
+        raise ArgumentError, "heap_size percentage must be less than 100%, you provided #{$1}%"
+      elsif percent < 0.01
+        raise ArgumentError, "heap_size percentage must be between 100% - 1%, you provided #{$1}%"
+      end
+      percent_int = JVMArgs::Util.get_system_ram_m.sub(/M/,'').to_i
+      percentage_ram = (percent_int * percent).to_i
+      @args[:nonstandard]["Xmx"] = JVMArgs::NonStandard.new("-Xmx#{percentage_ram}M")
+      @args[:nonstandard]["Xms"] = JVMArgs::NonStandard.new("-Xms#{percentage_ram}M")
+    end
+
+    def add_rule(rule_name, &block)
+      @rules.add(rule_name, block)
+    end
+
+  end
+end

data/lib/jvmargs/directive.rb

@@ -0,0 +1,29 @@
+module JVMArgs
+
+  class Directive
+
+    attr_accessor :key, :value
+
+    def initialize(arg)
+      arg.sub!(/^-?D?/, '')
+    
+      if arg =~ /(.*?)=(.*)/
+        @key = $1
+        @value = $2
+      else
+        @key = arg
+        @value = true
+      end
+    end
+
+    def to_s
+      if @value == true
+        "-D#{@key}"
+      else
+        "-D#{@key}=#{@value}"
+      end
+    end
+   
+  end
+
+end

data/lib/jvmargs/heaprules.rb

@@ -0,0 +1,32 @@
+module JVMArgs
+  module Rules
+
+    def self.heap_too_big(key="Xmx",args)
+      total_ram = JVMArgs::Util.get_raw_num(JVMArgs::Util.get_system_ram_m)
+      new_ram = JVMArgs::Util.get_raw_num(args[:nonstandard][key].value)
+      if new_ram > total_ram
+        raise ArgumentError, "You can't set #{key} to larger than the available system RAM of #{total_ram}"
+      end
+    end
+
+    def self.max_smaller_than_min(key="Xmx",args)
+      if args[:nonstandard]["Xms"].nil?
+        return
+      else
+        max_heap = JVMArgs::Util.get_raw_num(args[:nonstandard]["Xmx"].value)
+        min_heap = JVMArgs::Util.get_raw_num(args[:nonstandard]["Xms"].value)
+        if max_heap < min_heap
+          raise ArgumentError, "Max heap size #{max_heap}M cannot be smaller than Minimum heap size #{min_heap}M"
+        end
+      end
+    end
+
+  end
+end
+
+  # HeapRules = [
+  #              method(:heap_too_big).to_proc,
+  #              method(:max_smaller_than_min).to_proc
+  #             ]
+
+  

data/lib/jvmargs/nonstandard.rb

@@ -0,0 +1,24 @@
+module JVMArgs
+  class NonStandard
+    attr_accessor :key, :value
+    
+    def initialize(arg)
+      arg.sub!(/^-/, '')
+      arg =~ /(X[a-z]+)([0-9]+[a-zA-Z])?/
+      @key = $1
+      if $2.nil?
+        @value =  true
+      else
+        @value = JVMArgs::Util.convert_to_m($2)
+      end
+    end
+    
+    def to_s
+      if @value == true
+        "-#{@key}"
+      else
+        "-#{@key}#{@value}"
+      end
+    end
+  end
+end

data/lib/jvmargs/rules.rb

@@ -0,0 +1,51 @@
+module JVMArgs
+  class Rules 
+
+    def initialize(*args)
+      @rules = []
+    end
+
+    def each
+      @rules.each do |rule|
+        yield(rule)
+      end
+    end
+    
+    def add(rule_name, block)
+      new_rule_name = "rule_#{rule_name.to_s}".to_sym
+      JVMArgs::Rules.define_method(new_rule_name) {|args| block.call(args) }
+    end
+
+    def rules
+      JVMArgs::Rules.public_instance_methods(false).select {|rule| rule.to_s =~ /^rule_.*/  }
+    end
+    
+    def [](index)
+      @rules[index]
+    end
+
+    
+    def rule_heap_too_big(key="Xmx",args)
+      total_ram = JVMArgs::Util.get_raw_num(JVMArgs::Util.get_system_ram_m)
+      new_ram = JVMArgs::Util.get_raw_num(args[:nonstandard][key].value)
+      if new_ram > total_ram
+        raise ArgumentError, "You can't set #{key} to larger than the available system RAM of #{total_ram}"
+      end
+    end
+
+    def rule_max_smaller_than_min(key="Xmx",args)
+      if args[:nonstandard]["Xms"].nil?
+        return
+      else
+        max_heap = JVMArgs::Util.get_raw_num(args[:nonstandard]["Xmx"].value)
+        min_heap = JVMArgs::Util.get_raw_num(args[:nonstandard]["Xms"].value)
+        if max_heap < min_heap
+          raise ArgumentError, "Max heap size #{max_heap}M cannot be smaller than Minimum heap size #{min_heap}M"
+        end
+      end
+    end
+
+  end
+end
+
+  

data/lib/jvmargs/standard.rb

@@ -0,0 +1,22 @@
+module JVMArgs
+
+  class Standard
+    
+    attr_accessor :key, :value
+    
+    def initialize(arg)
+      arg.sub!(/^-/, '')
+      @key = arg
+      @value = true
+    end
+       
+    def to_s
+      if @value
+        "-#{@key}"
+      else
+        ""
+      end
+    end
+  end
+
+end

data/lib/jvmargs/unstable.rb

@@ -0,0 +1,30 @@
+module JVMArgs
+  class Unstable
+
+   attr_accessor :key, :value
+    
+    def initialize(arg)
+      arg.sub!(/^-?XX:/, '')
+      # check it if a boolean option
+      arg =~ /(\+|-)(.*)/
+      if !$1.nil?
+        @key = $2
+        @value = $1 == '+' ? true : false
+      else
+        arg =~ /(.*)=(.*)/
+        @key = $1
+        @value = $2
+      end
+    end
+    
+    def to_s
+      if @value.class == TrueClass or @value.class == FalseClass
+        symbol = @value ? '+' : '-'
+        "-XX:#{symbol}#{@key}"
+      else
+        "-XX:#{@key}=#{@value}"
+      end
+    end
+  end 
+    
+end

data/lib/jvmargs/util.rb

@@ -0,0 +1,43 @@
+module JVMArgs
+  module Util
+
+
+    def self.get_system_ram_m
+      if defined? node
+        total_ram = node['memory']['total']
+      else
+        require 'ohai'
+        ohai = Ohai::System.new
+        ohai.require_plugin "linux::memory"
+        total_ram = ohai["memory"]["total"]
+      end
+      self.convert_to_m(total_ram)
+    end
+
+    def self.get_raw_num(number)
+      number.sub(/M/, '').to_i
+    end
+    
+    def self.convert_to_m(number)
+      raw_num, unit = self.normalize_units(number)
+      case unit
+      when "K"
+        converted_num = (raw_num/1024).to_int
+      when "G"
+        converted_num = (raw_num * 1024).to_int
+      when "M"
+        converted_num = raw_num
+      end
+      "#{converted_num}M"
+    end
+      
+    def self.normalize_units(number)
+      number =~ /^([0-9]+)\s*([kKmMgG])[bB]?/
+      if $2.nil?
+        raise ArgumentError, "You must indicate whether number is KiloBytes(K), MegaBytes (M), or Gigabytes(G)"
+      end
+      [$1.to_i, $2.upcase]
+    end
+    
+  end
+end

data/lib/jvmargs/version.rb

@@ -0,0 +1,4 @@
+module JVMArgs
+  # The gem version
+  VERSION = '0.0.2'
+end

metadata

@@ -0,0 +1,120 @@
+--- !ruby/object:Gem::Specification
+name: jvmargs
+version: !ruby/object:Gem::Version
+  version: 0.0.2
+  prerelease: 
+platform: ruby
+authors:
+- Bryan W. Berry
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2013-03-25 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: chef
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '10.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '10.0'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 2.13.0
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 2.13.0
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 0.9.2.2
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 0.9.2.2
+- !ruby/object:Gem::Dependency
+  name: pry
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+description: Sanity check command-line arguments to the JVM
+email: 
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/jvmargs.rb
+- lib/jvmargs/args.rb
+- lib/jvmargs/nonstandard.rb
+- lib/jvmargs/version.rb
+- lib/jvmargs/rules.rb
+- lib/jvmargs/util.rb
+- lib/jvmargs/heaprules.rb
+- lib/jvmargs/directive.rb
+- lib/jvmargs/standard.rb
+- lib/jvmargs/unstable.rb
+homepage: https://github.com/bryanwb/jvmargs
+licenses:
+- Apache2
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.24
+signing_key: 
+specification_version: 3
+summary: jvmargs-0.0.2
+test_files: []
+has_rdoc: