bricolage-streamingload 0.14.1 → 0.16.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
-SHA1:
-  metadata.gz: ba544bbf6c3a133aeb0882e5193296dc734a25ec
-  data.tar.gz: 8706c5cd050e1083075d2890e422e3834680c2ea
+SHA256:
+  metadata.gz: e799760fb58dd3ec1dcd3eeaf74b068a0336a21a32488f04dc0353d44514c0cd
+  data.tar.gz: 1ec9bfe0cadd6537a352e99cf7b61d53122ae17a67dd31c733e4e2e0c7890849
 SHA512:
-  metadata.gz: d03f41030b3c7bc7175c1abf7b2f6e8e5c3a40c3cff5754e678c76338190b9c0293792dc4a8cc34fbf262419ac054467503c823ceafc2a55a6e59d9a2762f937
-  data.tar.gz: 8bd7c0cc4b826383d552ed196581b00a8d33010e24ada2cb9f3eb76ed381178b35e706585a2da449d019bf8bd4e43233afc84902695382df152925261d3fb848
+  metadata.gz: 700157b7be5a53753056d7a830d5502c117cbcae474a97be17a72cb18b58a8aa48ea927aaf333040e6674c098413e91298a6bfe112b1fcd58293d32c4b207068
+  data.tar.gz: bb63e1c5e7f646003ed07c0f12758cb79bd316227033d43ae2cb0165a203d3706bfdf6ea2eae8ef681e8d9c02285fcca5a98ab4f8d7b0c3a78ab3a11b853fe31

data/.gitignore

@@ -0,0 +1,5 @@
+*.gem
+.bundle
+
+/config/development/
+/config/test/

data/Gemfile

@@ -0,0 +1,2 @@
+source "https://rubygems.org"
+gemspec

data/Gemfile.lock

@@ -0,0 +1,54 @@
+PATH
+  remote: .
+  specs:
+    bricolage-streamingload (0.15.2)
+      aws-sdk-s3 (~> 1.8)
+      aws-sdk-sqs (~> 1.3)
+      bricolage (>= 5.29.2)
+      pg (~> 0.18.0)
+
+GEM
+  remote: https://rubygems.org/
+  specs:
+    aws-eventstream (1.0.1)
+    aws-partitions (1.102.0)
+    aws-sdk-core (3.25.0)
+      aws-eventstream (~> 1.0)
+      aws-partitions (~> 1.0)
+      aws-sigv4 (~> 1.0)
+      jmespath (~> 1.0)
+    aws-sdk-kms (1.7.0)
+      aws-sdk-core (~> 3)
+      aws-sigv4 (~> 1.0)
+    aws-sdk-s3 (1.17.1)
+      aws-sdk-core (~> 3, >= 3.21.2)
+      aws-sdk-kms (~> 1)
+      aws-sigv4 (~> 1.0)
+    aws-sdk-sns (1.3.0)
+      aws-sdk-core (~> 3)
+      aws-sigv4 (~> 1.0)
+    aws-sdk-sqs (1.4.0)
+      aws-sdk-core (~> 3)
+      aws-sigv4 (~> 1.0)
+    aws-sigv4 (1.0.3)
+    bricolage (5.29.2)
+      aws-sdk-s3 (~> 1)
+      aws-sdk-sns (~> 1)
+      pg (~> 0.18.0)
+    jmespath (1.4.0)
+    pg (0.18.4)
+    power_assert (1.1.3)
+    rake (12.3.1)
+    test-unit (3.2.8)
+      power_assert
+
+PLATFORMS
+  ruby
+
+DEPENDENCIES
+  bricolage-streamingload!
+  rake
+  test-unit
+
+BUNDLED WITH
+   1.16.1

data/LICENSES

@@ -0,0 +1,21 @@
+The MIT License
+
+Copyright (c) 2016 Minero Aoki, Shimpei Kodama
+
+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/RELEASE.md

@@ -0,0 +1,127 @@
+# Bricolage Streaming Load Release Note
+
+## version 0.16.0
+
+- [new] Retry for also error tasks, not only failure tasks.
+  Error tasks are assumed as "non-retriable" e.g. DB login error, but some of them are really able to be resolved
+  by retrying on ECS environment. Retrying is not expensive (and human operation is relatively expensive),
+  we'll retry on all failures and errors.
+
+## version 0.15.2
+
+- [fix] send-data-event: Send objectSize=0 to indicate this message is fake.
+
+## version 0.15.1
+
+- [new] Use the schema name in the parameters table by default (no more "no such variable: XXXX_schema" error).
+
+## version 0.15.0
+
+- [CHANGE] Upgrade AWS-SDK to v3.  This version requires at least Bricolage 5.25.
+
+## version 0.14.2
+
+- Reduces loading retry count
+
+## version 0.14.1
+
+- [fix] task logging did not work
+
+## version 0.14.0
+
+- [new] Logs (task - object) relashonships to S3 on dispatch, as a DB backup.
+
+## version 0.13.0
+
+- grand refactoring
+
+## version 0.12.0
+
+- [CHANGE] Adds task_id column to the log table (strload_load_logs).
+
+## version 0.11.0
+
+- Loosen dependent Bricolage version
+
+## version 0.10.2
+
+- [new] New parameter dispatch-interval.
+
+## version 0.10.1
+
+- [fix] Fixes simple variable ref bug.
+
+## version 0.10.0
+
+- [new] Automatically complement strload_jobs status with Redshift-side log table.
+
+## version 0.9.0
+
+- [new] Introduces Redshift-side load log table (strload_load_logs) for load duplication checking.
+
+## version 0.8.1
+
+- [fix] tmp: Do not retry on data connection failure.
+
+## version 0.8.0
+
+- [CHANGE] Loader retries failed load tasks automatically.  Streaming loader does NOT delete a task from the queue on job failures.
+- [enhancement] Rewrites loader to get better error handling (ensure to write log record in the wider range of errornous situations).
+
+## version 0.7.1
+
+- fix utilities
+
+## version 0.7.0
+
+- [CHANGE] SQS data source requires "region" attribute.
+
+## version 0.6.2
+
+- [new] AWS access key id and secret key are now optional for SQS data sources (to allow using EC2 instance or ECS task attached IAM roles).
+- [new] New utility commands send-data-event, send-shutdown, send-checkpoint, send-load-task.
+- Adds sample config files.
+
+## version 0.6.1
+
+- [fix] dispatcher: Default ctl data source was wrong.
+- [fix] dispatcher: Detects S3 events by "s3" attribute instead of "eventSource" attribute, to allow fake S3 events (from non-S3 system).
+- [fix] dispatcher: SNS alert is now optional.
+- [fix] dispatcher: Correctly deletes unknown format messages.
+- [enhancement] Adds more logging messages.
+
+## version 0.6.0
+
+- [CHANGE] Adds loaded column to strload_objects table to record if the object is really loaded or not.
+- [CHANGE] Now strload_objects' object_url is unique.  Duplicated objects are stored in another table, strload_dup_objects.
+- [CHANGE] Now strload_table has table_id column, which is the primary key.
+- [new] Loader daemon supports new command line option --working-dir, to support symbolic linked path, such as Capistrano deploy target (current/).
+- [new] Keeps Redshift manifest file for later inspection.
+- [enhancement] Reduces the number of Redshift writer transactions (1 transaction for 1 loading).
+- [enhancement] Delay dispatching tasks until current event batch is processed, to avoid unexpected visibility timeout.
+- [enhancement] Adds more logging messages.
+
+## version 0.5.1
+
+- [fix] Fixes slow query
+
+## version 0.5.0
+
+- [new] Introduces FLUSHTABLE dispatcher event
+
+## version 0.4.0
+
+- [new] Introduces CHECKPOINT dispatcher event
+
+## version 0.3.0
+
+- [new] Supoprts SNS notification
+
+## version 0.2.0
+
+- not released
+- [fix] Fixes async delete timing
+
+## version 0.1.0
+
+- 2016-07-13 works 1 month

data/Rakefile

@@ -0,0 +1,3 @@
+task :test do
+  load "#{__dir__}/test/all.rb"
+end

data/bin/send-data-event

@@ -28,7 +28,7 @@ ARGF.each do |line|
       },
       object: {
         key: key,
-        size: 1
+        size: 0
       }
     },
     noDispatch: no_dispatch

data/bricolage-streamingload.gemspec

@@ -0,0 +1,26 @@
+require_relative 'lib/bricolage/streamingload/version'
+
+Gem::Specification.new do |s|
+  s.platform = Gem::Platform::RUBY
+  s.name = 'bricolage-streamingload'
+  s.version = Bricolage::StreamingLoad::VERSION
+  s.summary = 'Bricolage Streaming Load Daemon'
+  s.description = 'Bricolage Streaming Load Daemon loads S3 data files to Redshift continuously.'
+  s.license = 'MIT'
+
+  s.author = ['Minero Aoki', 'Shimpei Kodama']
+  s.email = ['aamine@loveruby.net']
+  s.homepage = 'https://github.com/aamine/bricolage-streamingload'
+
+  s.files = `git ls-files -z`.split("\x0").reject {|f| f.match(%r{^(test|spec|features)/}) }
+  s.executables = s.files.grep(%r{bin/}).map {|path| File.basename(path) }
+  s.require_path = 'lib'
+
+  s.required_ruby_version = '>= 2.3.0'
+  s.add_dependency 'bricolage', '>= 5.29.2'
+  s.add_dependency 'pg', '~> 0.18.0'
+  s.add_dependency 'aws-sdk-s3', '~> 1.8'
+  s.add_dependency 'aws-sdk-sqs', '~> 1.3'
+  s.add_development_dependency 'rake'
+  s.add_development_dependency 'test-unit'
+end

data/config/production/database.yml

@@ -0,0 +1,66 @@
+### Databases
+
+db_ctl:
+  type: psql
+  host: localhost
+  port: 5432
+  database: bricolage
+  username: bricolage
+  # Get password from password.yml
+  password: <%= password 'postgres_bricolage_password' %>
+  encoding: utf8
+
+db_data: &db_data
+  type: psql
+  host: redshift.host
+  port: 5439
+  database: production
+  username: bricolage
+  # Get password from password.yml
+  password: <%= password 'redshift_bricolage_password' %>
+  encoding: utf8
+
+sql:
+  <<: *db_data
+
+### SQS
+
+sqs_event:
+  type: sqs
+  region: ap-northeast-1
+  url: https://sqs.ap-northeast-1.amazonaws.com/111111111111/bricolage-events
+  max_number_of_messages: 10
+  visibility_timeout: 600
+  wait_time_seconds: 10
+  # Enable following lines if you use access key explicitly.
+  # Otherwise Bricolage uses EC2 instance or ECS task attached IAM role.
+  #access_key_id: "<%%= password 'aws_access_key_id' %>"
+  #secret_access_key: "<%%= password 'aws_secret_access_key' %>"
+
+sqs_task:
+  type: sqs
+  region: ap-northeast-1
+  url: https://sqs.ap-northeast-1.amazonaws.com/111111111111/bricolage-tasks
+  max_number_of_messages: 1
+  visibility_timeout: 1800
+  wait_time_seconds: 10
+  # Enable following lines if you use access key explicitly.
+  # Otherwise Bricolage uses EC2 instance or ECS task attached IAM role.
+  #access_key_id: "<%%= password 'aws_access_key_id' %>"
+  #secret_access_key: "<%%= password 'aws_secret_access_key' %>"
+
+### S3
+
+s3_ctl: &s3_ctl
+  type: s3
+  region: ap-northeast-1
+  endpoint: s3-ap-northeast-1.amazonaws.com
+  bucket: bricolagectl.ap-northeast-1
+  prefix: development/strload
+  # Enable following lines if you use access key explicitly.
+  # Otherwise Bricolage uses EC2 instance or ECS task attached IAM role.
+  #access_key_id: "<%%= password 'aws_access_key_id' %>"
+  #secret_access_key: "<%%= password 'aws_secret_access_key' %>"
+
+s3:
+  <<: *s3_ctl

data/config/production/password.yml

@@ -0,0 +1,5 @@
+# Never commit this file
+redshift_bricolage_password: xxxxxxxxxxx
+postgres_bricolage_password: xxxxxxxxxxx
+aws_access_key_id: "AKIAAAAAAAAAAAAAAAAA"
+aws_secret_access_key: "xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx"

data/config/production/streamingload.yml

@@ -0,0 +1,20 @@
+#event-queue-ds: sqs_event
+#task-queue-ds: sqs_task
+
+#ctl-postgres-ds: db_ctl
+#ctl-s3-ds: s3_ctl
+
+#redshift-ds: db_data
+#log-table: strload_load_logs
+
+#dispatch-interval: 60
+
+# To Enable SNS notification
+#sns-ds: sns
+#alert-level: warn
+
+url_patterns:
+  -
+    url: "s3://some-log-bucket/\\w{4}\\.\\w+?\\.\\w+?\\.(?<schema>\\w+)\\.(?<table>\\w+)/\\d{4}/\\d{2}/\\d{2}/.*\\.gz"
+    schema: "%schema"
+    table: "%table"

data/config/production/variable.yml

@@ -0,0 +1,5 @@
+defaults:
+    redshift-ds: db_data
+    ctl-ds: s3_ctl
+
+dwh_schema: dwh

data/lib/bricolage/sqsdatasource.rb

@@ -1,6 +1,6 @@
 require 'bricolage/datasource'
 require 'securerandom'
-require 'aws-sdk'
+require 'aws-sdk-sqs'
 require 'json'
 require 'time'
 

data/lib/bricolage/streamingload/dispatcher.rb

@@ -9,7 +9,6 @@ require 'bricolage/streamingload/chunkrouter'
 require 'bricolage/streamingload/chunkbuffer'
 require 'bricolage/streamingload/loadtasklogger'
 require 'bricolage/streamingload/alertinglogger'
-require 'aws-sdk'
 require 'yaml'
 require 'optparse'
 require 'fileutils'

data/lib/bricolage/streamingload/job.rb

@@ -63,13 +63,13 @@ module Bricolage
         return false
       rescue JobError => ex
         @logger.error ex.message
-        return true
+        return false
       rescue Exception => ex
         @logger.exception ex
         return true
       end
 
-      MAX_RETRY = 5
+      MAX_RETRY = 2
 
       def execute_task
         @process_id = "#{Socket.gethostname}-#{$$}"
@@ -241,6 +241,12 @@ module Bricolage
               ;
           EndSQL
           @logger.info "load succeeded: #{manifest.url}"
+        rescue JobFailure => ex
+          if /stl_load_errors/ =~ ex.message
+            # We cannot resolve this load error by retry, give up now.
+            raise JobError, ex.message
+          end
+          raise
         end
 
         def write_load_log(log)

data/lib/bricolage/streamingload/jobparams.rb

@@ -35,7 +35,7 @@ module Bricolage
       end
 
       def JobParams.resolve_schema(ctx, schema)
-        ctx.global_variables["#{schema}_schema"] || schema
+        ctx.global_variables.get_force("#{schema}_schema") || schema
       end
       private_class_method :resolve_schema
 

data/lib/bricolage/streamingload/manifest.rb

@@ -1,3 +1,5 @@
+require 'aws-sdk-s3'
+
 module Bricolage
 
   module StreamingLoad

data/lib/bricolage/streamingload/taskhandler.rb

@@ -149,12 +149,15 @@ module Bricolage
 
     class NoopJob
 
-      def initialize(context:, ctl_ds:, task_id:, force: false, logger:)
+      def initialize(context:, ctl_ds:, data_ds:, log_table:, task_id:, force: false, logger:)
         @ctx = context
         @ctl_ds = ctl_ds
+        @data_ds = data_ds
+        @log_table = log_table
         @task_id = task_id
         @force = force
         @logger = logger
+        @working_dir = Dir.getwd
       end
 
       def execute(fail_fast: false)
@@ -164,7 +167,7 @@ module Bricolage
       end
 
       def execute_task
-        @logger.info "execute_task: task_id=#{@task_id} force=#{@force} ctx=#{@ctx.home_path} ctl_ds=#{@ctl_ds.name} dir=#{@working_dir}"
+        @logger.info "execute_task: task_id=#{@task_id} force=#{@force} ctx=#{@ctx.home_path} ctl_ds=#{@ctl_ds.name} data_ds=#{@data_ds.name} dir=#{@working_dir}"
       end
 
     end