dynamosaurus 0.0.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 7c83dcc84e2df48c376bcee1bbf2b75888a2ea63
+  data.tar.gz: a3481c7cd31a0ccce8beee8217539f0b0d993660
+SHA512:
+  metadata.gz: 1156a59ba6601629bd3e773285e4dd5b6e6741633a61364e201b01fae67bb433e95ceba19e69d8c5a471a004d0b9c5742dcf78fc6358d7d87dac60283c7b2d53
+  data.tar.gz: 6c6a0ea222c90b658d50a67b160fb6177be10946ad043b71f677661f3be55c2ada79199a87c067d67bb021f6c97e45e79bc7a6270299f7401b5c63116f00e2cb

data/.gitignore

@@ -0,0 +1,17 @@
+*.gem
+*.rbc
+.bundle
+.config
+.yardoc
+Gemfile.lock
+InstalledFiles
+_yardoc
+coverage
+doc/
+lib/bundler/man
+pkg
+rdoc
+spec/reports
+test/tmp
+test/version_tmp
+tmp

data/.rspec

@@ -0,0 +1,2 @@
+--format documentation
+--color

data/.travis.yml

@@ -0,0 +1,3 @@
+language: ruby
+rvm:
+  - 2.0.0

data/Gemfile

@@ -0,0 +1,4 @@
+source 'https://rubygems.org'
+
+# Specify your gem's dependencies in dynamosaurus.gemspec
+gemspec

data/LICENSE.txt

@@ -0,0 +1,22 @@
+Copyright (c) 2015 Isamu Arimoto
+
+MIT License
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.md

@@ -0,0 +1,55 @@
+# Dynamosaurus
+
+## Installation
+
+Add this line to your application's Gemfile:
+
+    gem 'dynamosaurus'
+
+And then execute:
+
+    $ bundle
+
+Or install it yourself as:
+
+    $ gem install dynamosaurus
+
+## Usage
+
+    require "dynamosaurus"
+    
+    class SimpleKVS < Dynamosaurus::DynamoBase
+      key :simple_key, :string
+    end
+    
+    aws_config = {
+      :access_key_id => 'aws_dynamodb_access_key_id',
+      :secret_access_key => 'aws_dynamodb_secret_access_key',
+      :region => 'us-west-1',
+    }
+    Aws.config = aws_config
+    
+    # create
+    SimpleKVS.put({:simple_key => "key"}, {:num => 1})
+    
+    # read
+    kvs = SimpleKVS.get("key")
+    
+    # update
+    kvs.update({:test => 1})
+    
+    # update
+    kvs.num = 100
+    kvs.save
+    
+    # delete
+    kvs.delete
+
+
+## Contributing
+
+1. Fork it
+2. Create your feature branch (`git checkout -b my-new-feature`)
+3. Commit your changes (`git commit -am 'Add some feature'`)
+4. Push to the branch (`git push origin my-new-feature`)
+5. Create new Pull Request

data/Rakefile

@@ -0,0 +1,6 @@
+require "bundler/gem_tasks"
+require "rspec/core/rake_task"
+
+RSpec::Core::RakeTask.new(:spec)
+
+task :default => :spec

data/dynamosaurus.gemspec

@@ -0,0 +1,27 @@
+# coding: utf-8
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'dynamosaurus/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = "dynamosaurus"
+  spec.version       = Dynamosaurus::VERSION
+  spec.authors       = ["Isamu Arimoto"]
+  spec.email         = ["isamu.a@gmail.com"]
+  spec.description   = %q{Dynamodb simple ORM}
+  spec.summary       = %q{Dynamodb simple ORM}
+  spec.homepage      = ""
+  spec.license       = "MIT"
+
+  spec.files         = `git ls-files`.split($/)
+  spec.executables   = spec.files.grep(%r{^bin/}) { |f| File.basename(f) }
+  spec.test_files    = spec.files.grep(%r{^(test|spec|features)/})
+  spec.require_paths = ["lib"]
+
+  spec.add_development_dependency "bundler", "~> 1.3"
+  spec.add_development_dependency "rake"
+  spec.add_development_dependency "rspec"
+
+  spec.add_runtime_dependency 'aws-sdk-core', ">= 2.0.20"
+
+end

data/lib/dynamosaurus.rb

@@ -0,0 +1,163 @@
+require "dynamosaurus/version"
+require "dynamosaurus/dynamo_base"
+require "dynamosaurus/dynamo_class"
+
+module Dynamosaurus
+  class << self
+    attr_accessor :logger
+  end
+
+  class Logger
+
+    attr_accessor :level
+    attr_accessor :delimiter
+    attr_accessor :auto_flush
+    attr_reader   :buffer
+    attr_reader   :log
+    attr_reader   :init_args
+
+    Levels =
+    {
+      :fatal => 7,
+      :error => 6,
+      :warn  => 4,
+      :info  => 3,
+      :debug => 0
+    }
+
+    private
+
+    # Readies a log for writing.
+    #
+    # ==== Parameters
+    # log<IO, String>:: Either an IO object or a name of a logfile.
+    def initialize_log(log)
+      close if @log # be sure that we don't leave open files laying around.
+
+      if log.respond_to?(:write)
+        @log = log
+      elsif File.exist?(log)
+        @log = open(log, (File::WRONLY | File::APPEND))
+        @log.sync = true
+      else
+        FileUtils.mkdir_p(File.dirname(log)) unless File.directory?(File.dirname(log))
+        @log = open(log, (File::WRONLY | File::APPEND | File::CREAT))
+        @log.sync = true
+        @log.write("#{Time.now.httpdate} #{delimiter} info #{delimiter} Logfile created\n")
+      end
+    end
+
+    public
+
+    # To initialize the logger you create a new object, proxies to set_log.
+    #
+    # ==== Parameters
+    # *args:: Arguments to create the log from. See set_logs for specifics.
+    def initialize(*args)
+      @init_args = args
+      set_log(*args)
+      self.auto_flush = true
+      Dynamosaurus.logger = self
+    end
+
+    # Replaces an existing logger with a new one.
+    #
+    # ==== Parameters
+    # log<IO, String>:: Either an IO object or a name of a logfile.
+    # log_level<~to_sym>::
+    #   The log level from, e.g. :fatal or :info. Defaults to :error in the
+    #   production environment and :debug otherwise.
+    # delimiter<String>::
+    #   Delimiter to use between message sections. Defaults to " ~ ".
+    # auto_flush<Boolean>::
+    #   Whether the log should automatically flush after new messages are
+    #   added. Defaults to false.
+    def set_log(log, log_level = nil, delimiter = " ~ ", auto_flush = false)
+      if log_level && Levels[log_level.to_sym]
+        @level = Levels[log_level.to_sym]
+      else
+        @level = Levels[:debug]
+      end
+      @buffer     = []
+      @delimiter  = delimiter
+      @auto_flush = auto_flush
+
+      initialize_log(log)
+    end
+
+    # Flush the entire buffer to the log object.
+    def flush
+      return unless @buffer.size > 0
+      @log.write(@buffer.slice!(0..-1).join)
+    end
+
+    # Close and remove the current log object.
+    def close
+      flush
+      @log.close if @log.respond_to?(:close) && !@log.tty?
+      @log = nil
+    end
+
+    # Appends a message to the log. The methods yield to an optional block and
+    # the output of this block will be appended to the message.
+    #
+    # ==== Parameters
+    # string<String>:: The message to be logged. Defaults to nil.
+    #
+    # ==== Returns
+    # String:: The resulting message added to the log file.
+    def <<(string = nil)
+      message = ""
+      message << delimiter
+      message << string if string
+      message << "\n" unless message[-1] == ?\n
+      @buffer << message
+      flush if @auto_flush
+
+      message
+    end
+    alias_method :push, :<<
+
+    # Generate the logging methods for DataMapper.logger for each log level.
+    Levels.each_pair do |name, number|
+      class_eval <<-LEVELMETHODS, __FILE__, __LINE__
+
+      # Appends a message to the log if the log level is at least as high as
+      # the log level of the logger.
+      #
+      # ==== Parameters
+      # string<String>:: The message to be logged. Defaults to nil.
+      #
+      # ==== Returns
+      # self:: The logger object for chaining.
+      def #{name}(message = nil)
+        self << message if #{number} >= level
+        self
+      end
+
+      # Appends a message to the log if the log level is at least as high as
+      # the log level of the logger. The bang! version of the method also auto
+      # flushes the log buffer to disk.
+      #
+      # ==== Parameters
+      # string<String>:: The message to be logged. Defaults to nil.
+      #
+      # ==== Returns
+      # self:: The logger object for chaining.
+      def #{name}!(message = nil)
+        self << message if #{number} >= level
+        flush if #{number} >= level
+        self
+      end
+
+      # ==== Returns
+      # Boolean:: True if this level will be logged by this logger.
+      def #{name}?
+        #{number} >= level
+      end
+      LEVELMETHODS
+    end
+    
+  end
+
+end

data/lib/dynamosaurus/dynamo_base.rb

@@ -0,0 +1,144 @@
+require "aws-sdk-core"
+
+module Dynamosaurus
+  class DynamoBase
+    TYPES = {
+      :string => :s,
+      :number => :n
+    }
+   
+    def dynamo_db
+      self.class.dynamo_db
+    end
+  
+    def initialize params
+      data = params[:data] if params[:data]
+      @data = Dynamosaurus::DynamoBase.res2hash(data)
+    end
+
+    def table_name
+      self.class.table_name
+    end
+    
+    def [] key
+      @data[key]
+    end
+
+    def []=(key,value)
+      @data[key] = value
+    end
+    
+    def data
+      @data
+    end
+
+    def exist?
+      ! @data.empty?
+    end
+    
+    def empty?
+      @data.empty?
+    end
+
+    def try name
+      @data[name.to_s]
+    end
+    
+    def method_missing(name, *params)
+      name = name.to_s[0..-2] if name.to_s[-1] == "="
+      if exist? and @data.has_key?(name.to_s)
+        if params.empty?
+          @data[name.to_s]
+        else
+          @data[name.to_s] = params[0]
+        end
+      else
+        super
+      end
+    end
+    
+    def key
+      return @key if @key
+      @key = self.class.get_key
+    end
+
+    def keys
+      item_key = {}
+      item_key[key[0].to_sym] = @data[key[0].to_s]
+      item_key[key[2].to_sym] = @data[key[2].to_s] if key.size == 4
+      item_key
+    end
+
+    def update attributes={}, attribute_nums={}, options={}
+      Dynamosaurus.logger << "update"
+
+      attribute_updates = {}
+      attributes.each do |k, v|
+        @data[k.to_s] = v
+        attribute_updates[k.to_s] = {
+          :value => v,
+          :action => "PUT"
+        }
+      end
+
+      attribute_nums.each do |k, v|
+        @data[k.to_s] = v.to_i
+        attribute_updates[k.to_s] = {
+          :value => v.to_i,
+          :action => "PUT"
+        }
+      end
+      
+      query = {
+        :table_name => self.class.table_name,
+        :key => keys,
+        :attribute_updates => attribute_updates
+      }.merge(options)
+      res = dynamo_db.update_item(query)
+    end
+
+    def save
+      attribute_updates = {}
+      @data.each do |k, v|
+        if key.index(k.to_sym).nil?
+          attribute_updates[k.to_s] = {
+            :value => v,
+            :action => "PUT"
+          }
+        end
+      end
+
+      query = {
+        :table_name => self.class.table_name,
+        :key => keys,
+        :attribute_updates => attribute_updates
+      }
+      res = dynamo_db.update_item(query)
+    end
+
+    def attr_delete attributes=[]
+      Dynamosaurus.logger << "delete"
+
+      attribute_updates = {}
+      attributes.each do |k|
+        attribute_updates[k.to_s] = {
+          :action => "DELETE"
+        }
+        @data.delete(k.to_s)
+      end
+      
+      res = dynamo_db.update_item(
+        :table_name => self.class.table_name,
+        :key => keys,
+        :attribute_updates => attribute_updates
+      )
+    end
+
+    def delete
+      dynamo_db.delete_item(
+        :table_name => self.class.table_name,
+        :key => keys
+      )
+    end
+  end
+end

data/lib/dynamosaurus/dynamo_class.rb

@@ -0,0 +1,359 @@
+module Dynamosaurus
+  class DynamoBase
+    class << self
+      def all_models
+        subclasses = []
+        ObjectSpace.each_object(Dynamosaurus::DynamoBase.singleton_class) do |k|
+          subclasses << k if k.superclass == Dynamosaurus::DynamoBase
+        end
+        subclasses
+      end
+
+      def create_table
+        tables = dynamo_db.list_tables.table_names
+
+        Dynamosaurus::DynamoBase.all_models.each do |model_class|
+          if tables.index(model_class.table_name).nil?
+            table = dynamo_db.create_table(
+              model_class.get_table_schema
+            )
+          end
+        end
+      end
+
+      def tables
+        Dynamosaurus::DynamoBase.all_models.map do |model_class|
+          model_class.table_name
+        end
+      end
+
+      def table_name
+        name.downcase + (ENV['DYNAMODB_SUFFIX'] || "_local")
+      end
+      
+      def dynamo_db
+        return @dynamo_db if @dynamo_db
+        unless ENV['DYNAMODB_SUFFIX']
+          @dynamo_db = Aws::DynamoDB::Client.new(
+            :endpoint => "http://localhost:8000",
+            :region => "us-west-1"
+          )
+        else
+          @dynamo_db = Aws::DynamoDB::Client.new
+        end
+      end
+
+      def key k, type, range_key=nil, range_key_type=nil
+        @key = [k, Dynamosaurus::DynamoBase::TYPES[type]]
+        @key << range_key << Dynamosaurus::DynamoBase::TYPES[range_key_type] if range_key
+      end
+      
+      def get_key
+       @key
+      end
+
+      def table_schema schema = {}
+        @schema = schema
+      end
+
+      def get_table_schema
+        @schema[:table_name] = table_name
+        @schema
+      end
+      
+      def global_index index_name, key, key_type, range_key=nil, range_key_type=nil
+        @global_index = {} if @global_index.nil?
+        @global_index[index_name] = [key, Dynamosaurus::DynamoBase::TYPES[key_type]]
+        @global_index[index_name] << range_key <<  Dynamosaurus::DynamoBase::TYPES[range_key_type] if range_key
+      end
+      
+      def get_global_indexes
+        (@global_index) ? @global_index : {}
+
+      end
+      def get_global_index name
+        @global_index[name]
+      end
+
+      def secondary_index index_name, range_key=nil, range_key_type=nil
+        @secondary_index = {} if @secondary_index.nil?
+        @secondary_index[index_name.to_sym] = [@key[0], @key[1], range_key, Dynamosaurus::DynamoBase::TYPES[range_key_type]] if range_key
+      end
+
+      def get_secondary_indexes
+        @secondary_index ? @secondary_index : {}
+      end
+      def get_secondary_index name
+        @secondary_index[name]
+      end
+
+      def get_index hash
+        get_secondary_indexes.merge(get_global_indexes).each{|key, value|
+          if hash.size == 1 && hash.keys.first == value.first
+            return {
+              :index_name => key,
+              :keys => [value[0]]
+            }
+          else hash.size == 2 && hash.keys.sort == [value[0], value[2]].sort
+            return {
+              :index_name => key,
+              :keys => [value[0], value[2]]
+            }
+          end
+        }
+        nil
+      end
+      
+      def key_list
+        keys = get_key + get_secondary_indexes.values.flatten
+        list = {}
+        until keys.empty?
+          convi = keys.shift(2)
+          list[convi[0]] = convi[1]
+        end
+        list
+      end
+
+      def res2hash hash
+        new_hash = {}
+        return new_hash if hash.nil?
+        hash
+      end
+      
+
+      def query_without_index value, option
+        keys = {}
+
+        value.each_with_index{|(k,v), i|
+          keys[k] = {
+            :comparison_operator => "EQ",
+            :attribute_value_list => [v.to_s]
+          }
+        }
+        Dynamosaurus.logger << "query index #{table_name} #{keys}"
+        query keys, nil, option
+      end
+
+      def get_item_key value
+        if value.is_a? Array
+          {
+            get_key[0] => value[0].to_s,
+            get_key[2] => value[1].to_s,
+          }
+        else
+          {
+            get_key[0] => value.to_s,
+          }
+        end
+      end
+
+      def get_from_key value, option={}
+        return nil if value.nil?
+
+        item_key = get_item_key(value)
+        Dynamosaurus.logger << "get_item #{table_name} #{item_key}"
+      
+        res = dynamo_db.get_item(
+          :table_name => table_name,
+          :key => item_key
+        )
+        if res.item
+          new :data => res.item
+        else
+          nil  
+        end
+      end
+      
+      def get_from_index hash, option={}
+        if index = get_index(hash)
+          keys = {}
+          
+          index[:keys].each do |key|
+            keys[key] = {
+              :comparison_operator => "EQ",
+              :attribute_value_list => [hash[key]]
+            }
+          end
+          Dynamosaurus.logger << "query index #{table_name} #{keys}"
+          query keys, index[:index_name], option
+        end
+      end
+
+      def get_from_local_index hash, option={}
+        if index = hash.delete(:index)
+          keys = {}
+          hash.each do |k, v|
+            keys[k] = {
+              :comparison_operator => "EQ",
+              :attribute_value_list => [ v.to_s ]
+            }
+          end 
+          Dynamosaurus.logger << "query local_index #{table_name} #{keys}"
+          query keys, index, option
+        end
+      end
+
+      def query keys, index=nil, option={}
+        query = {
+          :table_name => table_name,
+          :key_conditions => keys,
+        }
+        if index
+          query[:index_name] = index
+        end
+
+        query.merge!(option)
+        res = dynamo_db.query(query)
+        if query[:select] == "COUNT"
+          res[:count]
+        else
+          if res.items
+            return res.items.map{|item|
+             new :data => item
+            }
+          end
+        end
+      end
+
+      # public method
+      def get value, option={}
+        if value.is_a? Hash
+          if value[:index]
+            get_from_local_index value, option
+          elsif option[:noindex]
+            option.delete(:noindex)
+            query_without_index value, option
+          else
+            get_from_index value, option
+          end
+        else
+          get_from_key value, option
+        end
+      end
+
+      def batch_get_item keys
+        return nil if keys.nil? or keys.empty?
+        Dynamosaurus.logger << "batch_get_item #{table_name} #{keys}"
+        if get_key.size == 2
+          my_keys = keys.map{|_key| {get_key[0].to_s => _key.to_s } }
+        else
+          my_keys = []
+          keys[get_key[0]].each do |key1|
+            keys[get_key[2]].each do |key2|
+              my_keys << { 
+                get_key[0].to_s => key1,
+                get_key[2].to_s => key2,
+              }
+            end
+          end
+        end
+
+        res = dynamo_db.batch_get_item(
+          :request_items => {
+            table_name => {
+              :keys => my_keys
+            }
+          })
+        
+        if res.responses[table_name]
+          return res.responses[table_name].map{|item|
+            new :data => item
+          }
+        end
+        nil
+      end
+
+      def first
+        Dynamosaurus.logger << "first #{table_name}"
+        res = dynamo_db.scan({
+          :table_name => table_name,
+          :limit => 1
+        })
+        if res.items && res.count > 0
+          new :data => res.items[0]
+        else
+          nil
+        end
+      end
+
+      def all
+        Dynamosaurus.logger << "all #{table_name}"
+        res = dynamo_db.scan({
+          :table_name => table_name,
+        })
+        if res.items
+          res.items.map{|item|
+            new :data => item
+          }
+        end
+      end
+      
+      def put hash, num_hash={}, return_values=nil
+        
+        new_hash = {}
+        hash.each{|key, value|
+          new_hash[key] = value.to_s unless value.nil?
+        }
+        num_hash.merge({:updated_at => Time.now.to_i}).each{|key, value|
+          new_hash[key] = value.to_i unless value.nil?
+        } if num_hash
+        
+        res = dynamo_db.put_item(
+          :table_name => table_name,
+          :item => new_hash,
+          :return_values => return_values || "NONE"
+        )
+      end
+
+      def save hash, num_hash={}, return_values=nil
+        put(hash, num_hash, return_values)
+        my_keys = get_key
+        
+        if my_keys.size == 4
+          get([hash[my_keys[0]], hash[my_keys[2]]])
+        else
+          get(hash[my_keys[0]])
+        end
+      end
+      
+      def add key=[], attribute_nums={}, options={}
+        Dynamosaurus.logger << "update"
+        
+        attribute_updates = {}
+        attribute_nums.each do |k, v|
+          attribute_updates[k.to_s] = {
+            :value => v,
+            :action => "ADD"
+          }
+        end
+        
+        class_key = get_key
+        keys = {
+          class_key[0].to_sym => key.is_a?(Array) ? key[0] : key 
+        }
+        keys[class_key[2].to_sym] = key[1]  if class_key.size > 2
+
+        query ={
+          :table_name => table_name,
+          :key => keys,
+          :attribute_updates => attribute_updates
+        }
+        query = query.merge(options)
+        res = dynamo_db.update_item(query)
+        
+      end
+
+      def delete_item value
+        return nil if value.nil?
+
+        old_item = dynamo_db.delete_item(
+          :table_name => table_name,
+          :key => get_item_key(value),
+          :return_values => "ALL_OLD"
+        )
+        new :data => old_item.attributes
+      end
+    end
+    # end of self class
+  end
+end

data/lib/dynamosaurus/version.rb

@@ -0,0 +1,3 @@
+module Dynamosaurus
+  VERSION = "0.0.1"
+end

data/spec/dynamosaurus_spec.rb

@@ -0,0 +1,149 @@
+require 'spec_helper'
+
+describe Dynamosaurus do
+  before(:all) do  
+    ENV['DYNAMODB_SUFFIX'] = "_local"
+
+    Dynamosaurus::Logger.new('log/vm.log', :debug)
+    Aws.config = {
+      :endpoint => "http://localhost:8000",
+      :region => 'local_test',
+    }
+
+    Dynamosaurus::DynamoBase.create_table
+  end
+
+  after(:all) do  
+    connect = Dynamosaurus::DynamoBase.dynamo_db
+    Dynamosaurus::DynamoBase.tables.each do |table_name|
+      connect.delete_table(:table_name => table_name)
+    end
+  end
+
+  it 'should have a version number' do
+    expect(Dynamosaurus::VERSION).not_to be_nil
+  end
+
+  it 'simple ordered kvs test' do
+    expect(SimpleOrderedKVS.first).to be_nil
+
+    SimpleOrderedKVS.put({:simple_key => "key", :simple_id => "1"})
+
+    kvs = SimpleOrderedKVS.first
+    expect(kvs.simple_key).to eq "key"
+    expect(kvs.simple_id).to eq "1"
+
+    SimpleOrderedKVS.get(["key", "1"])
+    expect(kvs.simple_id).to eq "1"
+
+    sleep 1
+    SimpleOrderedKVS.put({:simple_key => "key", :simple_id => "3"})
+    sleep 1
+    SimpleOrderedKVS.put({:simple_key => "key", :simple_id => "2"})
+    
+    orderd_items = SimpleOrderedKVS.get({
+      :index => "updated_at_index",
+      :simple_key => "key"
+    },{
+      :scan_index_forward => false,
+      :limit => 50,
+      })
+    expect(orderd_items[0].simple_id).to eq "2"
+    expect(orderd_items[1].simple_id).to eq "3"
+    expect(orderd_items[2].simple_id).to eq "1"
+
+    batch_items = SimpleOrderedKVS.batch_get_item({:simple_key => ["key"], :simple_id => ["1", "2", "3"]})
+    expect(orderd_items.size).to eq 3
+
+    kvs = SimpleOrderedKVS.first
+    kvs2 = SimpleOrderedKVS.get({:simple_key => kvs.simple_key, :updated_at => kvs.updated_at})
+    expect(kvs.simple_id).to eq kvs2[0].simple_id
+
+  end
+
+  it 'simple kvs test' do
+    expect(SimpleKVS.first).to be_nil
+    expect(SimpleKVS.get("key")).to be_nil
+
+    SimpleKVS.put({:simple_key => "key"}, {:num => 1})
+    expect(SimpleKVS.first.num).to eq 1
+
+    kvs = SimpleKVS.get("key")
+    expect(kvs.num).to eq 1
+
+    kvs = SimpleKVS.get("key2")
+    expect(kvs).to be_nil
+
+    kvs = SimpleKVS.first
+    expect(kvs.num).to eq 1
+
+    SimpleKVS.add("key", {:num => 1})
+    expect(SimpleKVS.first.num).to eq 2
+
+    SimpleKVS.put({:simple_key => "key"}, {:num => 1})
+    expect(SimpleKVS.first.num).to eq 1
+
+    kvs = SimpleKVS.first
+    kvs.update({}, {:test => 1})
+    expect(SimpleKVS.first.test).to eq 1
+
+    kvs = SimpleKVS.first
+    kvs.update({:test => "1"})
+    expect(SimpleKVS.first.test).to eq "1"
+
+    kvs.attr_delete(["test"])
+    expect(kvs["test"]).to be_nil
+
+    kvs = SimpleKVS.first
+    expect(kvs["test"]).to be_nil
+
+    kvs.num = 100
+    kvs.save
+
+    kvs = SimpleKVS.first
+    expect(kvs.num).to eq 100
+
+    SimpleKVS.put({:simple_key => "key1"}, {:num => 1})
+    SimpleKVS.put({:simple_key => "key2"}, {:num => 2})
+    SimpleKVS.put({:simple_key => "key3"}, {:num => 3})
+
+    expect(SimpleKVS.all.size).to eq 4
+
+    expect(SimpleKVS.batch_get_item(["key1"]).size).to eq 1
+    expect(SimpleKVS.batch_get_item(["key1", "key2"]).size).to eq 2
+    expect(SimpleKVS.batch_get_item(["key1", "key10"]).size).to eq 1
+
+    old_kvs = SimpleKVS.delete_item("key2")
+    expect(old_kvs.simple_key).to eq "key2"
+    expect(SimpleKVS.batch_get_item(["key1", "key2"]).size).to eq 1
+
+    kvs = SimpleKVS.get("key3")
+    kvs.delete
+    kvs = SimpleKVS.get("key3")
+    expect(kvs).to be_nil
+    
+  end
+
+  it 'comment test' do
+    expect(Comment.first).to be_nil
+    expect(Comment.get(["1", "1"])).to be_nil
+
+    Comment.put({:content_id => "1", :message_id => "1", :user_id => "abc"})
+    expect(Comment.first.content_id).to eq "1"
+
+    kvs = Comment.get(["1", "1"])
+    expect(kvs.content_id).to eq "1"
+
+    kvs = Comment.get(["2", "2"])
+    expect(kvs).to be_nil
+
+    Comment.put({:content_id => "1", :message_id => "2", :user_id => "abc"})
+    Comment.put({:content_id => "1", :message_id => "3", :user_id => "xyz"})
+    
+    # global secondary index
+    comments = Comment.get({:user_id => "abc"})
+    expect(comments.size).to eq 2
+
+  end
+
+end

data/spec/spec_helper.rb

@@ -0,0 +1,3 @@
+$LOAD_PATH.unshift File.expand_path('../../lib', __FILE__)
+require 'dynamosaurus'
+require 'testmodel'

data/spec/testmodel.rb

@@ -0,0 +1,103 @@
+# -*- coding: utf-8 -*-
+
+class SimpleKVS < Dynamosaurus::DynamoBase
+  key :simple_key, :string
+
+  table_schema   :key_schema => 
+    [
+    { :key_type => "HASH",
+      :attribute_name => "simple_key"
+    }
+    ],
+  :provisioned_throughput => {
+    :read_capacity_units => 100,
+    :write_capacity_units => 10
+  },
+  :attribute_definitions => [
+    {:attribute_name => "simple_key", :attribute_type => "S"},
+  ]
+end
+
+class SimpleOrderedKVS < Dynamosaurus::DynamoBase
+  key :simple_key, :string, :simple_id, :string
+  secondary_index :updated_at_index, :updated_at, :number
+
+  table_schema :key_schema => 
+    [
+    { :key_type => "HASH",
+      :attribute_name => "simple_key"
+    },
+    { :key_type => "RANGE",
+      :attribute_name => "simple_id"},
+    ],
+  :provisioned_throughput => {
+    :read_capacity_units => 20,
+    :write_capacity_units => 10
+  },
+  :attribute_definitions => [
+    {:attribute_name => "simple_key", :attribute_type => "S"},
+    {:attribute_name => "simple_id", :attribute_type => "S"},
+    {:attribute_name => "updated_at", :attribute_type => "N"},
+  ],
+  :local_secondary_indexes => [
+    {
+      :index_name => "updated_at_index",
+      :key_schema => [
+                      { :key_type => "HASH",
+                        :attribute_name => "simple_key"},
+        
+                      {
+                        :key_type => "RANGE",
+                        :attribute_name => "updated_at"
+                      }
+      ],
+      :projection => {
+        :projection_type => "ALL"
+      },
+    },
+  ]
+end
+
+class Comment < Dynamosaurus::DynamoBase
+  key :content_id, :string, :message_id, :string
+  global_index :user_index, :user_id, :string
+
+  table_schema :key_schema => 
+    [
+    { 
+      :key_type => "HASH",
+      :attribute_name => "content_id"
+        },
+        {
+          :key_type => "RANGE",
+          :attribute_name => "message_id"
+        }
+        ],
+      :provisioned_throughput => {
+        :read_capacity_units => 100,
+        :write_capacity_units => 10
+      },
+      :attribute_definitions => [
+        {:attribute_name => "content_id", :attribute_type => "S"},
+        {:attribute_name => "message_id", :attribute_type => "S"},
+        {:attribute_name => "user_id", :attribute_type => "S"},
+      ],
+      :global_secondary_indexes => [
+        {
+          :index_name => "user_index",
+          :key_schema => [
+            {
+              :key_type => "HASH",
+              :attribute_name => "user_id"
+            },
+          ],
+          :projection => {
+            :projection_type => "KEYS_ONLY",
+          },
+          :provisioned_throughput => {
+            :read_capacity_units => 50,
+            :write_capacity_units => 10
+          },
+        },
+      ]
+end

metadata

@@ -0,0 +1,118 @@
+--- !ruby/object:Gem::Specification
+name: dynamosaurus
+version: !ruby/object:Gem::Version
+  version: 0.0.1
+platform: ruby
+authors:
+- Isamu Arimoto
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2015-08-17 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.3'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.3'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: aws-sdk-core
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: 2.0.20
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '>='
+      - !ruby/object:Gem::Version
+        version: 2.0.20
+description: Dynamodb simple ORM
+email:
+- isamu.a@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- .rspec
+- .travis.yml
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- dynamosaurus.gemspec
+- lib/dynamosaurus.rb
+- lib/dynamosaurus/dynamo_base.rb
+- lib/dynamosaurus/dynamo_class.rb
+- lib/dynamosaurus/version.rb
+- spec/dynamosaurus_spec.rb
+- spec/spec_helper.rb
+- spec/testmodel.rb
+homepage: ''
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.4.6
+signing_key: 
+specification_version: 4
+summary: Dynamodb simple ORM
+test_files:
+- spec/dynamosaurus_spec.rb
+- spec/spec_helper.rb
+- spec/testmodel.rb