File: fast_struct.rb

package info (click to toggle)
ruby-beaneater 1.1.3-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 296 kB
  • sloc: ruby: 1,628; sh: 4; makefile: 2
file content (96 lines) | stat: -rw-r--r-- 3,381 bytes parent folder | download | duplicates (5)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
class Beaneater
  #
  # Borrowed from:
  # https://github.com/dolzenko/faster_open_struct/blob/master/lib/faster_open_struct.rb
  #
  # Up to 40 (!) times more memory efficient version of OpenStruct
  #
  # Differences from Ruby MRI OpenStruct:
  #
  # 1. Doesn't `dup` passed initialization hash (NOTE: only reference to hash is stored)
  #
  # 2. Doesn't convert hash keys to symbols (by default string keys are used,
  #    with fallback to symbol keys)
  #
  # 3. Creates methods on the fly on `OpenStruct` class, instead of singleton class.
  #    Uses `module_eval` with string to avoid holding scope references for every method.
  #
  # 4. Refactored, crud clean, spec covered :)
  #
  # @private
  class FasterOpenStruct
    # Undefine particularly nasty interfering methods on Ruby 1.8
    undef :type if method_defined?(:type)
    undef :id if method_defined?(:id)

    def initialize(hash = nil)
      @hash = hash || {}
      @initialized_empty = hash == nil
    end

    def method_missing(method_name_sym, *args)
      if method_name_sym.to_s[-1] == ?=
        if args.size != 1
          raise ArgumentError, "wrong number of arguments (#{args.size} for 1)", caller(1)
        end

        if self.frozen?
          raise TypeError, "can't modify frozen #{self.class}", caller(1)
        end

        __new_ostruct_member__(method_name_sym.to_s.chomp("="))
        send(method_name_sym, args[0])
      elsif args.size == 0
        __new_ostruct_member__(method_name_sym)
        send(method_name_sym)
      else
        raise NoMethodError, "undefined method `#{method_name_sym}' for #{self}", caller(1)
      end
    end

    def __new_ostruct_member__(method_name_sym)
      self.class.module_eval <<-END_EVAL, __FILE__, __LINE__ + 1
      def #{ method_name_sym }
        @hash.fetch("#{ method_name_sym }", @hash[:#{ method_name_sym }]) # read by default from string key, then try symbol
                                                                          # if string key doesn't exist
      end
      END_EVAL

      unless method_name_sym.to_s[-1] == ?? # can't define writer for predicate method
        self.class.module_eval <<-END_EVAL, __FILE__, __LINE__ + 1
        def #{ method_name_sym }=(val)
          if @hash.key?("#{ method_name_sym }") || @initialized_empty       # write by default to string key (when it is present
                                                                            # in initialization hash or initialization hash
                                                                            # wasn't provided)
            @hash["#{ method_name_sym }"] = val                             # if it doesn't exist - write to symbol key
          else
            @hash[:#{ method_name_sym }] = val
          end
        end
        END_EVAL
      end
    end

    def empty?
      @hash.empty?
    end

    #
    # Compare this object and +other+ for equality.
    #
    def ==(other)
      return false unless other.is_a?(self.class)
      @hash == other.instance_variable_get(:@hash)
    end

    #
    # Returns a string containing a detailed summary of the keys and values.
    #
    def inspect
      str = "#<#{ self.class }"
      str << " #{ @hash.map { |k, v| "#{ k }=#{ v.inspect }" }.join(", ") }" unless @hash.empty?
      str << ">"
    end
    alias :to_s :inspect
  end # FasterOpenStruct
end # Beaneater