multi_ec2_kiq 0.0.2 → 0.0.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: aa8f523926f920f85e754e199b9904a3eacc70d6
-  data.tar.gz: 7d0a24414b0e95dd0de15ed6ca7c82e811b10479
+  metadata.gz: 5d54c1b4879b8e479d310c7ec683133f36d37a7b
+  data.tar.gz: fdbf905399aeec95547880cce90a9121ecb17037
 SHA512:
-  metadata.gz: 6968fbbb2c98b5395afb1aa3b5280b18d6a74f3525c9208b43ecfa675f48cd670b149a066a5bacf2664b873c670007a8aea88ca2c8055fa4d4409cfa6b81adb5
-  data.tar.gz: 29812526672914724243477ea0b3b27e3caa3fa2799fe4f45a85f20cfd84c073423721b638b86c5047740c26f6ae23bda40bd0f02d483df4726e526ffc0eca18
+  metadata.gz: da6234e855b50d9152ac1ebf5d8657a30602931b4193f65a38bacfed640725424e50803fd8a8b7d9ab67f3131926f1cbfe1c49505d5da51b80d7e2451c55c041
+  data.tar.gz: 4d148db6c7b87ed4f7aa4ee5dab855a65e6536d3bcffe1ed1951f744da077104e779667e89866ae9c2360861e88c7dd4469bcb212786eac8bbcd6d5363749605

data/README.md

@@ -33,11 +33,17 @@ Or install it yourself as:
 
 ## Usage
 
-1. You have to set aws profile in your ~/.aws/credentials.
-
-2. Write configuration yml file, use "lib/settings.yml" as a reference.
-
-3. Then, call methods.
+You have to set aws profile in your ~/.aws/credentials.  
+  
+Write configuration yml file, use "lib/settings.yml" as a reference.  
+  
+To create ec2-status-table on DynamoDB, you have to call Dynamodb::create_status_table method from rails console, ruby source, e.t.c.
+```ruby
+    dynamodb = Dynamodb.new
+    dynamodb.create_status_table
+```
+  
+Then, call methods.
 ```ruby
     require "multi_ec2_kiq"
 
@@ -46,7 +52,7 @@ Or install it yourself as:
     MultiEc2Kiq.config_path = File.expand_path("your configuration file path", __FILE__)
     MultiEc2Kiq.start
 ```
-
+  
 ```ruby
     require "multi_ec2_kiq"
 

data/lib/multi_ec2_kiq/version.rb

@@ -1,3 +1,3 @@
 module MultiEc2Kiq
-  VERSION = "0.0.2"
+  VERSION = "0.0.3"
 end

data/lib/threads/dynamodb.rb

@@ -9,7 +9,6 @@ module MultiEc2Kiq
     end
 
     def started(instance_id)
-      create_table_if_not_exists
       put_item(instance_id, @STATUS.started)
       true
     end
@@ -30,31 +29,7 @@ module MultiEc2Kiq
       true
     end
 
-    private
-
-    def raise_not_exist
-      raise "instance_id doesn't exist on status table."
-    end
-
-    def get_item(instance_id)
-      dynamodb.get_item(
-        table_name: Settings.aws.dynamodb.status_table_name,
-        key: {instance_id: instance_id}
-      )
-    end
-
-    def put_item(instance_id, status)
-      dynamodb.put_item(
-        table_name: Settings.aws.dynamodb.status_table_name,
-        item: {instance_id: instance_id, status: status}
-      )
-    end
-
-    def create_table_if_not_exists
-      dynamodb.list_tables.table_names.each {|table|
-        return if table == Settings.aws.dynamodb.status_table_name
-      }
-
+    def create_status_table
       options = {
         table_name: Settings.aws.dynamodb.status_table_name,
         key_schema: [
@@ -75,14 +50,40 @@ module MultiEc2Kiq
         }
       }
 
+      create_table(options, Settings.aws.dynamodb.status_table_name)
+
+      true
+    end
+
+    private
+
+    def create_table(options, table_name)
       dynamodb.create_table(options)
 
-      dynamodb.wait_until(:table_exists, {table_name: Settings.aws.dynamodb.status_table_name}) do |w|
+      dynamodb.wait_until(:table_exists, {table_name: table_name}) do |w|
         w.max_attempts = 5
         w.delay = 5
       end
     end
 
+    def raise_not_exist
+      raise "instance_id doesn't exist on status table."
+    end
+
+    def get_item(instance_id)
+      dynamodb.get_item(
+        table_name: Settings.aws.dynamodb.status_table_name,
+        key: {instance_id: instance_id}
+      )
+    end
+
+    def put_item(instance_id, status)
+      dynamodb.put_item(
+        table_name: Settings.aws.dynamodb.status_table_name,
+        item: {instance_id: instance_id, status: status}
+      )
+    end
+
     def dynamodb
       @dynamodb ||= Aws::DynamoDB::Client.new(
         region: Settings.aws.dynamodb.region,

data/spec/threads/dynamodb_spec.rb

@@ -10,8 +10,13 @@ describe Dynamodb do
     @dynamodb = Dynamodb.new
   end
 
+  it 'should create statud table.' do
+    expect(@dynamodb).to receive(:create_table).with(anything, anything)
+    result = @dynamodb.create_status_table
+    expect(result).to eq(true)
+  end
+
   it 'should write started.' do
-    expect(@dynamodb).to receive(:create_table_if_not_exists)
     expect(@dynamodb).to receive(:put_item).with(anything, anything)
     result = @dynamodb.started("instance_test")
     expect(result).to eq(true)

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: multi_ec2_kiq
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.3
 platform: ruby
 authors:
 - SrcHndWng