Contents

class Jets::Builders
  class CodeSize
    LAMBDA_SIZE_LIMIT = 250 # Total lambda limit is 250MB
    include Util

    def self.check!
      new.check
    end

    def check
      return if within_lambda_limit?
      say "Over the Lambda size limit of #{LAMBDA_SIZE_LIMIT}MB".colorize(:red)
      say "Please reduce the size of your code."
      display_sizes
      exit 1
    end

    def within_lambda_limit?
      total_size < LAMBDA_SIZE_LIMIT * 1024 # 120MB
    end

    def total_size
      code_size = compute_size("#{stage_area}/code")
      opt_size = compute_size("#{stage_area}/opt")
      opt_size + code_size # total_size
    end

    def display_sizes
      code_size = compute_size("#{stage_area}/code")
      opt_size = compute_size("#{stage_area}/opt")
      total_size = opt_size + code_size
      overlimit = (LAMBDA_SIZE_LIMIT * 1024 - total_size) * -1
      say "Sizes:"
      say "Code: #{megabytes(code_size)} - #{stage_area}/code"
      say "Gem Layer: #{megabytes(opt_size)} - #{stage_area}/opt"
      say "Total Package: #{megabytes(total_size)}"
      say "Over limit by: #{megabytes(overlimit)}"
      # sh "du -csh #{stage_area}/*" unless ENV['TEST'] # uncomment to debug
    end

    def compute_size(path)
      out = `du -s #{path}`
      out.split(' ').first.to_i # bytes
    end

    def megabytes(bytes)
      n = bytes / 1024.0
      sprintf('%.1f', n) + 'MB'
    end

    def say(message)
      puts message unless ENV['TEST']
    end
  end
end

Version data entries

10 entries across 10 versions & 1 rubygems

Version	Path
jets-1.3.9	lib/jets/builders/code_size.rb
jets-1.3.8	lib/jets/builders/code_size.rb
jets-1.3.7	lib/jets/builders/code_size.rb
jets-1.3.6	lib/jets/builders/code_size.rb
jets-1.3.5	lib/jets/builders/code_size.rb
jets-1.3.4	lib/jets/builders/code_size.rb
jets-1.3.3	lib/jets/builders/code_size.rb
jets-1.3.2	lib/jets/builders/code_size.rb
jets-1.3.1	lib/jets/builders/code_size.rb
jets-1.3.0	lib/jets/builders/code_size.rb