bson 1.12.5 → 2.0.0.alpha

data/lib/bson/byte_buffer.rb

@@ -1,241 +0,0 @@
-# Copyright (C) 2009-2013 MongoDB, Inc.
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#   http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-
-require 'bson/grow'
-
-module BSON
-  class ByteBuffer
-    include BSON::Grow
-
-    attr_reader :order, :max_size
-
-    INT32_PACK = 'l<'.freeze
-    INT64_PACK = 'q<'.freeze
-    DOUBLE_PACK = 'E'.freeze
-
-    def initialize(initial_data="", max_size=DEFAULT_MAX_BSON_SIZE)
-      @str = case initial_data
-        when String then
-          if initial_data.respond_to?(:force_encoding)
-            initial_data.force_encoding('binary')
-          else
-            initial_data
-          end
-        when BSON::ByteBuffer then
-          initial_data.to_a.pack('C*')
-        else
-          initial_data.pack('C*')
-      end
-
-      @cursor = @str.length
-      @max_size = max_size
-    end
-
-    def rewind
-      @cursor = 0
-    end
-
-    def position
-      @cursor
-    end
-
-    def position=(val)
-      @cursor = val
-    end
-
-    def clear
-      @str = ""
-      @str.force_encoding('binary') if @str.respond_to?(:force_encoding)
-      rewind
-    end
-
-    def size
-      @str.size
-    end
-    alias_method :length, :size
-
-    # Appends a second ByteBuffer object, +buffer+, to the current buffer.
-    def append!(buffer)
-      @str << buffer.to_s
-      self
-    end
-
-    # Prepends a second ByteBuffer object, +buffer+, to the current buffer.
-    def prepend!(buffer)
-      @str = buffer.to_s + @str
-      self
-    end
-
-    def put(byte, offset=nil)
-      @cursor = offset if offset
-      if more?
-        @str[@cursor] = chr(byte)
-      else
-        ensure_length(@cursor)
-        @str << chr(byte)
-      end
-      @cursor += 1
-    end
-
-    def put_binary(data, offset=nil)
-      @cursor = offset if offset
-      if defined?(BINARY_ENCODING)
-        data = data.dup.force_encoding(BINARY_ENCODING)
-      end
-      if more?
-        @str[@cursor, data.length] = data
-      else
-        ensure_length(@cursor)
-        @str << data
-      end
-      @cursor += data.length
-    end
-
-    def put_array(array, offset=nil)
-      @cursor = offset if offset
-      if more?
-        @str[@cursor, array.length] = array.pack("C*")
-      else
-        ensure_length(@cursor)
-        @str << array.pack("C*")
-      end
-      @cursor += array.length
-    end
-
-    def put_num(i, offset, bytes)
-      pack_type = bytes == 4 ? INT32_PACK : INT64_PACK
-      @cursor = offset if offset
-      if more?
-        @str[@cursor, bytes] = [i].pack(pack_type)
-      else
-        ensure_length(@cursor)
-        @str << [i].pack(pack_type)
-      end
-      @cursor += bytes
-    end
-
-    def put_int(i, offset=nil)
-      put_num(i, offset, 4)
-    end
-
-    def put_long(i, offset=nil)
-      put_num(i, offset, 8)
-    end
-
-    def put_double(d, offset=nil)
-      a = []
-      [d].pack(DOUBLE_PACK).each_byte { |b| a << b }
-      put_array(a, offset)
-    end
-
-    # If +size+ == nil, returns one byte. Else returns array of bytes of length
-    # # +size+.
-    if "x"[0].is_a?(Integer)
-      def get(len=nil)
-        one_byte = len.nil?
-        len ||= 1
-        check_read_length(len)
-        start = @cursor
-        @cursor += len
-        if one_byte
-          @str[start]
-        else
-          @str[start, len].unpack("C*")
-        end
-      end
-    else
-      def get(len=nil)
-        one_byte = len.nil?
-        len ||= 1
-        check_read_length(len)
-        start = @cursor
-        @cursor += len
-        if one_byte
-          @str[start, 1].ord
-        else
-          @str[start, len].unpack("C*")
-        end
-      end
-    end
-
-    def get_int
-      check_read_length(4)
-      vals = @str[@cursor..@cursor+3]
-      @cursor += 4
-      vals.unpack(INT32_PACK)[0]
-    end
-
-    def get_long
-      check_read_length(8)
-      vals = @str[@cursor..@cursor+7]
-      @cursor += 8
-      vals.unpack(INT64_PACK)[0]
-    end
-
-    def get_double
-      check_read_length(8)
-      vals = @str[@cursor..@cursor+7]
-      @cursor += 8
-      vals.unpack(DOUBLE_PACK)[0]
-    end
-
-    def more?
-      @cursor < @str.size
-    end
-
-    def ==(other)
-      other.respond_to?(:to_s) && @str == other.to_s
-    end
-
-    def to_a(format="C*")
-      @str.unpack(format)
-    end
-
-    def unpack(format="C*")
-      to_a(format)
-    end
-
-    def to_s
-      @str
-    end
-
-    def dump
-      @str.each_byte do |c, i|
-        $stderr.puts "#{'%04d' % i}: #{'%02x' % c} #{'%03o' % c} #{'%s' % c.chr} #{'%3d' % c}"
-        i += 1
-      end
-    end
-
-    private
-
-    def ensure_length(length)
-      if @str.size < length
-        @str << NULL_BYTE * (length - @str.size)
-      end
-    end
-
-    def chr(byte)
-      if byte < 0
-        [byte].pack('c')
-      else
-        byte.chr
-      end
-    end
-
-    def check_read_length(len)
-      raise "attempt to read past end of buffer" if @cursor + len > @str.length
-    end
-
-  end
-end

data/lib/bson/exceptions.rb

@@ -1,37 +0,0 @@
-# Copyright (C) 2009-2013 MongoDB, Inc.
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#   http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-
-module BSON
-  # Generic Mongo Ruby Driver exception class.
-  class MongoRubyError < StandardError; end
-
-  # Raised when MongoDB itself has returned an error.
-  class MongoDBError < RuntimeError; end
-
-  # This will replace MongoDBError.
-  class BSONError < MongoDBError; end
-
-  # Raised when given a string is not valid utf-8 (Ruby 1.8 only).
-  class InvalidStringEncoding < BSONError; end
-
-  # Raised when attempting to initialize an invalid ObjectId.
-  class InvalidObjectId < BSONError; end
-
-  # Raised when trying to insert a document that exceeds the 4MB limit or
-  # when the document contains objects that can't be serialized as BSON.
-  class InvalidDocument < BSONError; end
-
-  # Raised when an invalid name is used.
-  class InvalidKeyName < BSONError; end
-end

data/lib/bson/grow.rb

@@ -1,173 +0,0 @@
-# encoding: utf-8
-
-# Copyright (C) 2009-2013 MongoDB, Inc.
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#   http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-
-module BSON
-  module Grow
-    # module with methods to grow BSON docs/objects/arrays
-    # this module is intended for internal use and is subject to change
-    # proper usage is essential as minimal overhead is preferred over usage checks
-    # unfinish! returns unfinished BSON for faster growing with bang! methods
-    # bang! methods work on unfinished BSON with neither terminating nulls nor proper sizes
-    # finish! must be called to finish BSON after using bang! methods
-    # corresponding non-bang methods work on finished BSON
-    # object/array methods should be paired, ex., array!/b_end! and array/b_end
-    # push!/push and push_doc!/push_doc append to arrays with correct keys
-    # b_end needs a better name
-
-    def to_e # Extract bytes for elements from BSON
-      @str[4...-1]
-    end
-
-    def to_t # Extract type from (single-element) BSON
-      @str[4,1]
-    end
-
-    def to_v # Extract value from (single-element) BSON
-      @str[(@str.index(NULL_BYTE,5)+1)...-1]
-    end
-
-    def finish_one!(offset = 0) # Appends terminating null byte and sets size
-      put(0)
-      put_int(@str.size - offset, offset)
-      @cursor = @str.size
-      self
-    end
-
-    def unfinish! # Backup past terminating null bytes
-      @b_pos ||= [0]
-      @cursor = @str.size - @b_pos.size # BSON::BSON_CODER.serialize may not restore @cursor
-      self
-    end
-
-    def finish! # Append all terminating null bytes and set all sizes
-      @b_pos ||= [0]
-      (@b_pos.size-1).downto(0){|i| finish_one!(@b_pos[i])}
-      self
-    end
-
-    def grow!(bson) # Appends BSON elements unfinished
-      put_binary(bson.to_e)
-      self
-    end
-
-    def grow(bson) # Appends BSON elements finished
-      @b_pos ||= [0]
-      put_binary(bson.to_e, @str.size - @b_pos.size)
-      finish!
-    end
-
-    def push!(bson) # Appends BSON element value with correct key unfinished
-      @a_index ||= [0]
-      @b_pos ||= [0]
-      put_binary(bson.to_t)
-      put_binary(@a_index[-1].to_s)
-      put_binary(NULL_BYTE)
-      @a_index[-1] += 1
-      put_binary(bson.to_v)
-      self
-    end
-
-    def push(bson) # Appends BSON element value with correct key finished
-      @a_index ||= [0]
-      @b_pos ||= [0]
-      put_binary(bson.to_t, @str.size - @b_pos.size)
-      put_binary(@a_index[-1].to_s)
-      put_binary(NULL_BYTE)
-      @a_index[-1] += 1
-      put_binary(bson.to_v)
-      finish!
-    end
-
-    def push_doc!(bson) # Appends BSON doc with correct key unfinished
-      @a_index ||= [0]
-      @b_pos ||= [0]
-      put(BSON::BSON_RUBY::OBJECT)
-      put_binary(@a_index[-1].to_s)
-      put(0)
-      @a_index[-1] += 1
-      put_binary(bson.to_s)
-      self
-    end
-
-    def push_doc(bson) # Appends BSON doc with correct key finished
-      @a_index ||= [0]
-      @b_pos ||= [0]
-      put(BSON::BSON_RUBY::OBJECT, @str.size - @b_pos.size)
-      put_binary(@a_index[-1].to_s)
-      put(0)
-      @a_index[-1] += 1
-      put_binary(bson.to_s)
-      finish!
-    end
-
-    def b_do!(key, type = BSON::BSON_RUBY::OBJECT) # Append object/array element unfinished
-      put(type)
-      BSON::BSON_RUBY.serialize_cstr(self, key)
-      @b_pos ||= [0]
-      @a_index ||= [0]
-      @b_pos << @cursor # mark position of size
-      @a_index << 0
-      put_int(0)
-      self
-    end
-
-    def b_do(key, type = BSON::BSON_RUBY::OBJECT) # Append object/array element finished
-      @b_pos ||= [0]
-      @cursor = @str.size - @b_pos.size
-      b_do!(key, type)
-      finish!
-    end
-
-    def doc!(key) # Append object element unfinished
-      b_do!(key, BSON::BSON_RUBY::OBJECT)
-    end
-
-    def doc(key) # Append object element finished
-      b_do(key, BSON::BSON_RUBY::OBJECT)
-    end
-
-    def array!(key) # Append array element unfinished
-      b_do!(key, BSON::BSON_RUBY::ARRAY)
-    end
-
-    def array(key) # Append array element finished
-      b_do(key, BSON::BSON_RUBY::ARRAY)
-    end
-
-    def b_end! # End object/array unfinished - next operation will be up one level
-      @b_pos ||= [0]
-      finish_one!(@b_pos[-1])
-      @b_pos.pop
-      @a_index ||= [0]
-      @a_index.pop
-      self
-    end
-
-    def b_end # End object/array finished - next operation will be up one level
-      @b_pos ||= [0]
-      @b_pos.pop
-      @a_index ||= [0]
-      @a_index.pop
-      self
-    end
-
-    def clear! # Clear internal state for reuse
-      @b_pos = @a_index = nil
-      self
-    end
-
-  end
-end