masamune 0.11.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 4aa0472dd6fa4bb472d79caf9a6c17a4570baad2
+  data.tar.gz: 853b036996129473d7f9f8dee3129f198988face
+SHA512:
+  metadata.gz: b48a634347ae17d3a214be9ee13b021a7099c992b6e23382347ab44019b7d5209e6c3d73af0957067fff8abb9b3365c6a40fe880ef88cb56f67f4f7cfcab62cc
+  data.tar.gz: 861919413965988c50128a731f05e0ba9fd1bc9a5fc7a410038b23ce1e2e06fb4624889b11fb0142f321d0d2a8cc657e15a22f06a7efbc646674a4e99c228c16

data/LICENSE.txt

@@ -0,0 +1,21 @@
+The MIT License (MIT)
+
+Copyright (c) 2014-2015, VMware, Inc. All Rights Reserved.
+
+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,54 @@
+[![Build Status](https://secure.travis-ci.org/socialcast/masamune.png?branch=master)](http://travis-ci.org/socialcast/masamune)
+[![License](http://img.shields.io/license/MIT.png?color=green)](http://opensource.org/licenses/MIT)
+
+Masamune
+========
+A Dataflow Programming Library.
+
+Description
+------------
+Masamune provides a [dataflow programming](http://en.wikipedia.org/wiki/Dataflow_programming) framework on top of [Thor](http://whatisthor.com/). In the framework, dataflows are constructed as Thor tasks that transform source data into the target data. Source and target data descriptions are encoded as annotations associated with the Thor command. From these source and target annotations, Masamune constructs a data dependency tree that describes how to automatically construct a target data set.
+
+Usage
+----------
+
+Describe your dataflow as source, target data transformations:
+```ruby
+class ExampleThor  < Thor
+  # Mix in Masamune specific Data Flow Behavior
+  include Masamune::Thor
+  include Masamune::Actions::DataFlow
+
+  # Mix in Masamune Actions for Data Processing
+  include Masamune::Actions::Streaming
+  include Masamune::Actions::Hive
+
+  # Describe a Data Processing Job
+  desc 'extract_logs', 'Organize log files by YYYY-MM-DD'
+
+  target fs.path(:target_dir, '%Y-%m-%d', mkdir: true)
+  source fs.path(:source_dir, '%Y%m%d*.log')
+  def extract_logs
+    targets.missing.each do |target|
+      target.sources.each do |source|
+        # Transform source into target
+        fs.copy(source.path, target.path)
+      end
+    end
+  end
+end
+```
+
+Execute your dataflow with the goal of processing all data from the start of the year:
+
+```
+thor extract_logs --start '1 year ago'
+```
+
+Contributing
+---------------
+
+* Fork the project
+* Fix the issue
+* Add unit tests
+* Submit pull request on github

data/Rakefile

@@ -0,0 +1,15 @@
+#!/usr/bin/env rake
+
+begin
+  require 'bundler/setup'
+rescue LoadError
+  puts 'You must `gem install bundler` and `bundle install` to run rake tasks'
+end
+
+Bundler::GemHelper.install_tasks
+
+require 'rspec/core/rake_task'
+
+RSpec::Core::RakeTask.new(:spec)
+
+task :default => :spec

data/bin/masamune-elastic-mapreduce

@@ -0,0 +1,4 @@
+#!/usr/bin/env ruby
+$: << File.expand_path('../../lib/', __FILE__)
+require 'masamune/tasks/elastic_mapreduce_thor'
+Masamune::Tasks::ElasticMapreduceThor.start(ARGV)

data/bin/masamune-hive

@@ -0,0 +1,4 @@
+#!/usr/bin/env ruby
+$: << File.expand_path('../../lib/', __FILE__)
+require 'masamune/tasks/hive_thor'
+Masamune::Tasks::HiveThor.start(ARGV)

data/bin/masamune-psql

@@ -0,0 +1,4 @@
+#!/usr/bin/env ruby
+$: << File.expand_path('../../lib/', __FILE__)
+require 'masamune/tasks/postgres_thor'
+Masamune::Tasks::PostgresThor.start(ARGV)

data/bin/masamune-shell

@@ -0,0 +1,4 @@
+#!/usr/bin/env ruby
+$: << File.expand_path('../../lib/', __FILE__)
+require 'masamune/tasks/shell_thor'
+Masamune::Tasks::ShellThor.start(ARGV)

data/lib/masamune.rb

@@ -0,0 +1,56 @@
+#  The MIT License (MIT)
+#
+#  Copyright (c) 2014-2015, VMware, Inc. All Rights Reserved.
+#
+#  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.
+
+module Masamune
+  require 'masamune/environment'
+  require 'masamune/has_environment'
+  require 'masamune/io'
+  require 'masamune/template'
+  require 'masamune/commands'
+  require 'masamune/accumulate'
+  require 'masamune/last_element'
+  require 'masamune/actions'
+  require 'masamune/helpers'
+  require 'masamune/configuration'
+  require 'masamune/data_plan'
+  require 'masamune/thor'
+  require 'masamune/thor_loader'
+  require 'masamune/filesystem'
+  require 'masamune/cached_filesystem'
+  require 'masamune/method_logger'
+  require 'masamune/after_initialize_callbacks'
+  require 'masamune/schema'
+  require 'masamune/transform'
+  require 'masamune/topological_hash'
+
+  extend self
+  extend Masamune::HasEnvironment
+
+  def load(config_file, &block)
+    Masamune::Environment.new.tap do |env|
+      env.configure do |config|
+        config.load(config_file)
+      end
+      env.catalog.instance_eval &block
+    end
+  end
+end

data/lib/masamune/accumulate.rb

@@ -0,0 +1,60 @@
+#  The MIT License (MIT)
+#
+#  Copyright (c) 2014-2015, VMware, Inc. All Rights Reserved.
+#
+#  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.
+
+module Masamune
+  module Accumulate
+    def accumulate(method, accumulator, *args)
+      accumulator.call(self, *args).tap do |elems|
+        send(method, *args) do |elem|
+          elems << elem
+        end
+      end
+    end
+
+    module ClassMethods
+      def method_accumulate(method, accumulator = lambda { |_, *args| Array.new })
+        self.class_eval do
+          new_method = "#{method}_with_accumulate".to_sym
+          old_method = "#{method}_without_accumulate".to_sym
+          alias_method old_method, method
+
+          define_method(new_method) do |*args, &block|
+            if block
+              instance = accumulator.call(self, *args)
+              send(old_method, *args) do |elem|
+                next if instance.respond_to?(:add?) && !instance.add?(elem)
+                block.call elem
+              end
+            else
+              accumulate(old_method, accumulator, *args)
+            end
+          end
+          alias_method method, new_method
+        end
+      end
+    end
+
+    def self.included(base)
+      base.extend ClassMethods
+    end
+  end
+end

data/lib/masamune/actions.rb

@@ -0,0 +1,38 @@
+#  The MIT License (MIT)
+#
+#  Copyright (c) 2014-2015, VMware, Inc. All Rights Reserved.
+#
+#  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.
+
+module Masamune
+  module Actions
+    require 'masamune/actions/hive'
+    require 'masamune/actions/s3cmd'
+    require 'masamune/actions/hadoop_streaming'
+    require 'masamune/actions/hadoop_filesystem'
+    require 'masamune/actions/postgres'
+    require 'masamune/actions/postgres_admin'
+    require 'masamune/actions/filesystem'
+    require 'masamune/actions/date_parse'
+    require 'masamune/actions/data_flow'
+    require 'masamune/actions/elastic_mapreduce'
+    require 'masamune/actions/execute'
+    require 'masamune/actions/transform'
+  end
+end

data/lib/masamune/actions/data_flow.rb

@@ -0,0 +1,131 @@
+#  The MIT License (MIT)
+#
+#  Copyright (c) 2014-2015, VMware, Inc. All Rights Reserved.
+#
+#  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.
+
+require 'chronic'
+require 'active_support/concern'
+
+module Masamune::Actions
+  module DataFlow
+    extend ActiveSupport::Concern
+
+    include Masamune::Actions::DateParse
+
+    def engine
+      self.class.engine
+    end
+
+    def targets
+      engine.targets(current_command_name)
+    end
+
+    def sources
+      engine.sources(current_command_name)
+    end
+
+    def target
+      targets.first
+    end
+
+    def source
+      sources.first
+    end
+
+    # TODO sources from file or input array
+    def parse_file_type(key)
+      return Set.new unless key
+      value = options[key] or return Set.new
+      File.exists?(value) or raise Thor::MalformattedArgumentError, "Expected file value for '--#{key}'; got #{value}"
+      Set.new File.read(value).split(/\s+/)
+    end
+
+    private
+
+    included do |base|
+      base.extend ClassMethods
+      base.class_eval do
+        class_option :sources, :desc => 'File of data sources to process'
+        class_option :targets, :desc => 'File of data targets to process'
+        class_option :resolve, :type => :boolean, :desc => 'Recursively resolve data dependencies', :default => true
+      end
+
+      base.after_initialize(:final) do |thor, options|
+        # Only execute this block if DataPlan::Engine is not currently executing
+        next if thor.engine.executing?
+        thor.engine.environment = thor.environment
+        thor.engine.filesystem.environment = thor.environment
+
+        raise Thor::RequiredArgumentMissingError, "No value provided for required options '--start' or '--at'" unless options[:start] || options[:at] || options[:sources] || options[:targets]
+        raise Thor::MalformattedArgumentError, "Cannot specify both option '--sources' and option '--targets'" if options[:sources] && options[:targets]
+
+        desired_sources = Masamune::DataPlan::Set.new thor.current_command_name, thor.parse_file_type(:sources)
+        desired_targets = Masamune::DataPlan::Set.new thor.current_command_name, thor.parse_file_type(:targets)
+
+        if thor.start_time && thor.stop_time
+          desired_targets.merge thor.engine.targets_for_date_range(thor.current_command_name, thor.start_time, thor.stop_time)
+        end
+
+        thor.engine.prepare(thor.current_command_name, options.merge(sources: desired_sources, targets: desired_targets))
+        thor.engine.execute(thor.current_command_name, options)
+        exit 0 if thor.top_level?
+      end if defined?(base.after_initialize)
+    end
+
+    module ClassMethods
+      @@namespaces = []
+      @@commands = []
+      @@sources = []
+      @@targets = []
+
+      def skip
+        @@namespaces << namespace
+        @@sources << {skip: true}
+        @@targets << {skip: true}
+      end
+
+      def source(source_options = {})
+        @@namespaces << namespace
+        @@sources << source_options
+      end
+
+      def target(target_options = {})
+        @@targets << target_options
+      end
+
+      def create_command(*a)
+        super.tap do
+          @@commands += a
+        end
+      end
+
+      def engine
+        @@engine ||= Masamune::DataPlan::Builder.instance.build(@@namespaces, @@commands, @@sources, @@targets)
+      end
+
+      private
+
+      # If internal call to Thor::Base.start fails, exit
+      def exit_on_failure?
+        true
+      end
+    end
+  end
+end

data/lib/masamune/actions/date_parse.rb

@@ -0,0 +1,75 @@
+#  The MIT License (MIT)
+#
+#  Copyright (c) 2014-2015, VMware, Inc. All Rights Reserved.
+#
+#  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.
+
+require 'chronic'
+require 'active_support/concern'
+
+module Masamune::Actions
+  module DateParse
+    extend ActiveSupport::Concern
+
+    def parse_datetime_type(key)
+      value = options[key]
+      Chronic.parse(value).tap do |datetime_value|
+        console("Using '#{datetime_value}' for --#{key}") if value != datetime_value
+      end or raise Thor::MalformattedArgumentError, "Expected date time value for '--#{key}'; got #{value}"
+    end
+
+    private
+
+    included do |base|
+      base.class_eval do
+        attr_accessor :start_value
+        attr_accessor :stop_value
+        attr_accessor :exact_value
+
+        class_option :start, :aliases => '-a', :desc => 'Start time'
+        class_option :stop, :aliases => '-b', :desc => 'Stop time'
+        class_option :at, :desc => 'Exact time'
+
+        no_tasks do
+          def start_date
+            (start_value || exact_value).try(:to_date)
+          end
+
+          def start_time
+            (start_value || exact_value).try(:to_time)
+          end
+
+          def stop_date
+            (stop_value || exact_value).try(:to_date)
+          end
+
+          def stop_time
+            (stop_value || exact_value).try(:to_time)
+          end
+        end
+      end
+
+      base.after_initialize(:latest) do |thor, options|
+        thor.start_value = thor.parse_datetime_type(:start) if options[:start]
+        thor.exact_value = thor.parse_datetime_type(:at)    if options[:at]
+        thor.stop_value  = thor.parse_datetime_type(:stop)  if options[:stop]
+      end
+    end
+  end
+end