tengine_core 0.5.28

data/lib/tengine/core/connection_test/.gitignore

@@ -0,0 +1 @@
+/VERSION

data/lib/tengine/core/connection_test/fire_bar_on_foo.rb

@@ -0,0 +1,16 @@
+# -*- coding: utf-8 -*-
+require 'tengine/core'
+
+class << Tengine
+  attr_accessor :callback_for_test
+end
+
+driver :connection_test_driver do
+  on :foo do
+    Tengine.callback_for_test.call(:foo) if Tengine.callback_for_test
+    fire :bar
+  end
+  on :bar do
+    Tengine.callback_for_test.call(:bar) if Tengine.callback_for_test
+  end
+end

data/lib/tengine/core/driveable.rb

@@ -0,0 +1,213 @@
+# -*- coding: utf-8 -*-
+require 'tengine/core'
+require 'active_support/core_ext/kernel/singleton_class'
+
+# イベントドライバ定義モジュール
+#
+module Tengine::Core::Driveable
+  extend ActiveSupport::Concern
+
+  included do
+    @__context__ = self.singleton_class
+    @__context__.extend(ClassMethods)
+    @__context__.instance_eval do
+      def driver; @driver; end
+      def driver=(val); @driver = val; end
+
+      def __on_args__; @__on_args__; end
+      def __on_args__=(val); @__on_args__ = val; end
+
+      def __creating_driver__; @__creating_driver__; end
+      def __creating_driver__=(val); @__creating_driver__ = val; end
+    end
+
+    driver = nil
+    if @__context__.__creating_driver__ = self.driver.nil? # ==比較じゃなくて、代入した結果で条件分岐してます
+      config = @__context__.respond_to?(:config) ? @__context__.config : nil
+      options = @__context__.respond_to?(:options) ? @__context__.options : {}
+      driver = Tengine::Core::Driver.new({
+          :name => self.driver_name, # self.name.gsub(/:/, 'Colon'),
+          :version => Tengine::Core::Setting.dsl_version,
+          :enabled => config ? !config[:tengined][:skip_enablement] : true,   # driverを有効化して登録するかのオプション
+          :enabled_on_activation => options[:enabled_on_activation].nil? || options[:enabled_on_activation],  # DSLに記述されているオプション
+          :target_class_name => self.name,
+        })
+      driver.create_session
+      begin
+        driver.save!
+      rescue Mongoid::Errors::Validations => e
+        raise e unless e.document.errors[:name].any?{|s| s =~ /taken/}
+        Tengine::Core.stdout_logger.debug("[#{Process.pid}] Occured #{e.inspect}")
+        driver = Tengine::Core::Driver.where(:name => self.driver_name, :version => Tengine::Core::Setting.dsl_version).first
+      rescue Mongo::OperationFailure => e
+        raise e unless e.message =~ /E11000 duplicate key error/
+        Tengine::Core.stdout_logger.debug("[#{Process.pid}] Occured #{e.inspect}")
+        driver = Tengine::Core::Driver.where(:name => self.driver_name, :version => Tengine::Core::Setting.dsl_version).first
+      end
+    end
+    @__context__.driver = driver || self.driver
+
+    def self.method_added(method_name)
+      return unless @__context__.__on_args__
+      args = @__context__.__on_args__
+      @__context__.__on_args__ = nil
+      return unless @__context__.__creating_driver__
+      driver = @__context__.driver
+      driver.reload
+      options = args.extract_options!
+      handler = driver.handlers.create!({
+          :event_type_names => args,
+          :target_instantiation_key => :instance_method,
+          :target_method_name => method_name.to_s
+        }.update(options))
+      args.each do |event_type_name|
+        driver.handler_paths.create!(:event_type_name => event_type_name, :handler_id => handler.id)
+      end
+    end
+
+    def self.singleton_method_added(method_name)
+      return if method_name == :singleton_method_added
+      # def self.hoge... と class << self; def baz...; end ではselfが異なる
+      # (前者はclass自身、後者はclassのsingleton_classになる)ので、差異を吸収するために
+      # selfが前者の場合@__context__には後者への参照が設定されており、
+      # selfが後者の場合には、@__context__はnilなので、selfを使うことによって
+      # 同じインスタンスの__on_args__を使用することが可能になります。
+      context = @__context__ || self
+      return unless context.__on_args__
+      args = context.__on_args__
+      context.__on_args__ = nil
+      return unless @__context__.__creating_driver__
+      driver = context.driver
+      driver.reload
+      options = args.extract_options!
+      handler = driver.handlers.create!({
+          :event_type_names => args,
+          :target_instantiation_key => :static,
+          :target_method_name => method_name.to_s
+        }.update(options))
+      # puts "handler: #{handler.inspect}\n#{args.inspect}"
+      args.each do |event_type_name|
+        driver.handler_paths.create!(:event_type_name => event_type_name, :handler_id => handler.id)
+      end
+    end
+  end
+
+  module ClassMethods
+    def driver_name
+      return nil if self.name.nil?
+      (self < Tengine::Core::Driveable::ByDsl) ? self.name.underscore : self.name
+    end
+
+    def driver
+      unless @driver
+        driver_attrs = {
+          :name => self.driver_name, # self.name.gsub(/:/, 'Colon'),
+          :version => Tengine::Core::Setting.dsl_version
+        }
+        @driver = Tengine::Core::Driver.first(:conditions => driver_attrs)
+      end
+      @driver
+    end
+
+    def session
+      @__session_wrapper__ ||= Tengine::Core::SessionWrapper.new(driver.session)
+    end
+
+    def on(*args, &block)
+      context = @__context__ || self
+      options = args.extract_options!
+      event_type_names = args
+      if block
+        filepath, lineno = *block.source_location
+        filepath = context.config.relative_path_from_dsl_dir(filepath.dup.force_encoding(Encoding.default_external).freeze)
+        options.update(:filepath => filepath, :lineno => lineno)
+        filter_def = nil
+        handler = nil
+        if event_type_names.length == 1 && event_type_names.first.is_a?(Tengine::Core::DslFilterDef)
+          filter_def = event_type_names.first
+          options[:filter] = filter_def.filter
+          event_type_names = filter_def.event_type_names
+          base_method_name = event_type_names.map(&:to_s).join("_")
+          driver = context.driver
+          driver.reload
+          handler = driver.handlers.new({
+              :event_type_names => event_type_names,
+              :target_instantiation_key => :instance_method,
+            }.update(options))
+          # フィルタ付きの場合は単純なイベントハンドラ名だけではメソッド名として表現できないので
+          # handler自身のIDをメソッド名に含めます。
+          method_name = "#{base_method_name}_#{handler.id.to_s}"
+          handler.target_method_name = method_name.to_s
+          handler.save!
+          event_type_names.each do |event_type_name|
+            driver.handler_paths.create!(:event_type_name => event_type_name, :handler_id => handler.id)
+          end
+        else
+          method_name = event_type_names.map(&:to_s).join("_")
+          context.__on_args__ = event_type_names.map(&:to_s) + [options]
+        end
+        case block.arity
+        when 1 then
+          define_method(method_name, &block)
+        when 0 then
+          impl_method_name = "__#{method_name}_impl__"
+          self.instance_eval do
+            define_method(method_name) do |event|
+              @__event__ = event
+              begin
+                send(impl_method_name)
+              ensure
+                @__event__ = nil
+              end
+            end
+            define_method(impl_method_name, &block)
+          end
+        else
+          raise Tengine::Core::DslError, "#{block.artity} aritties block given"
+        end
+        # handler.save! if handler
+      else
+        filepath, lineno = *caller.first.sub(/:in.+\Z/, '').split(/:/, 2)
+        options.update(:filepath => filepath, :lineno => lineno)
+        context.__on_args__ = ( args + [options] )
+      end
+    end
+  end
+
+  def session
+    self.class.session
+  end
+
+  def event
+    @__event_wrapper__ ||= (@__event__ ? Tengine::Core::EventWrapper.new(@__event__) : nil)
+  end
+
+  def kernel
+    ev = @__event__
+    ev ? ev.kernel : nil
+  end
+
+  def ack?; kernel.ack?; end
+  def submit; kernel.submit; end
+
+  def fire(*args, &block)
+    kernel.fire(*args, &block)
+  end
+
+
+  module ByDsl
+    extend ActiveSupport::Concern
+
+    included do
+      @__context__ = self.singleton_class
+      @__context__.instance_eval do
+        def config; @config; end
+        def config=(val); @config = val; end
+
+        def options; @options; end
+        def options=(val); @options = val; end
+      end
+    end
+  end
+
+end

data/lib/tengine/core/driver.rb

@@ -0,0 +1,69 @@
+# -*- coding: utf-8 -*-
+require 'tengine/core'
+
+# イベントドライバ
+#
+# イベントに対する処理はイベントハンドラによって実行されますが、イベントドライバはそのイベントハンドラの上位の概念です。
+# イベントハンドラは必ずイベントドライバの中に定義されます。
+#
+# またイベントドライバは有効化／無効化を設定する単位であり、起動時の設定あるいはユーザーの指定によって変更することができます。
+#
+class Tengine::Core::Driver
+  autoload :Finder, 'tengine/core/driver/finder'
+
+  include Mongoid::Document
+  include Mongoid::Timestamps
+  include Tengine::Core::Validation
+  include Tengine::Core::FindByName
+
+  # @attribute 名前
+  field :name, :type => String
+
+  # @attribute バージョン。デプロイされた際に設定されます。
+  field :version, :type => String
+
+  # @attribute 有効／無効
+  field :enabled, :type => Boolean
+
+  # @attribute 実行時有効／無効
+  field :enabled_on_activation, :type => Boolean, :default => true
+
+  # @attribute 対象クラス名
+  field :target_class_name, :type => String
+
+  index([ [:_id, Mongo::ASCENDING], [:enabled, Mongo::ASCENDING], [:version, Mongo::ASCENDING], ])
+  index([ [:name, Mongo::ASCENDING], [:version, Mongo::ASCENDING], ], :unique => true)
+  index([ [:version, Mongo::ASCENDING], [:enabled_on_activation, Mongo::ASCENDING], ])
+  index([ [:version, Mongo::ASCENDING], ])
+  index([ [:_id, Mongo::ASCENDING], [:name, Mongo::ASCENDING], ])
+
+  validates(:name, :presence => true,
+    :uniqueness => {:scope => :version, :message => "is already taken in same version"},
+    :format => BASE_NAME.options
+    )
+  validates :version, :presence => true
+
+  embeds_many :handlers, :class_name => "Tengine::Core::Handler"
+
+  belongs_to :session, :index => true, :class_name => "Tengine::Core::Session"
+  has_many :handler_paths, :class_name => "Tengine::Core::HandlerPath"
+
+  after_create :update_handler_path
+  before_create :create_session # has_oneによって追加されるメソッドcreate_sessionのように振る舞うメソッドです
+
+  def update_handler_path
+    handlers.each(&:update_handler_path)
+  end
+
+  def create_session
+    self.session ||= Tengine::Core::Session.create
+  end
+
+  class << self
+    # Tengine::Core::FindByName で定義しているクラスメソッドfind_by_nameを上書きしています
+    def find_by_name(name, options = {})
+      version = options[:version] || Tengine::Core::Setting.dsl_version
+      first(:conditions => {:name => name, :version => version})
+    end
+  end
+end

data/lib/tengine/core/driver/finder.rb

@@ -0,0 +1,42 @@
+# -*- coding: utf-8 -*-
+require 'tengine/core/driver'
+
+class Tengine::Core::Driver::Finder
+
+  ATTRIBUTE_NAMES = [
+    :name,
+    :version,
+    :enabled,
+    :enabled_on_activation
+  ].freeze
+
+  ATTRIBUTE_NAMES.each{|name| attr_accessor(name)}
+
+  def initialize(attrs = {})
+    attrs ||= {}
+    attrs.each do |attr, v| 
+      send("#{attr}=", v) unless v.blank?
+    end
+  end
+
+  def attributes
+    ATTRIBUTE_NAMES.inject({}){|d, name| d[name] = send(name); d}
+  end
+
+  def paginate(page)
+    scope(Tengine::Core::Driver).page(page)
+  end
+
+  def scope(criteria)
+    result = criteria
+    result = result.where(name: name) if name
+    result = result.where(version: version) if version
+    result = result.where(enabled: enabled) if enabled
+    result = result.where(enabled_on_activation: enabled_on_activation) if enabled_on_activation
+    # ソート
+    result = result.asc(:_id, :name)
+    result
+  end
+
+end
+

data/lib/tengine/core/dsl_evaluator.rb

@@ -0,0 +1,110 @@
+# -*- coding: utf-8 -*-
+require 'tengine/core'
+
+require 'active_support/dependencies'
+
+module Tengine::Core::DslEvaluator
+  attr_accessor :config
+
+  def __evaluate__
+    __setup_core_ext__
+    begin
+      Tengine.plugins.notify(self, :__evaluate__) do
+        ActiveSupport::Dependencies.mechanism = config.tengined.cache_drivers ? :require : :load
+        config.dsl_dir_path.tap do |dsl_dir_path|
+          $LOAD_PATH.unshift(dsl_dir_path) unless $LOAD_PATH.include?(dsl_dir_path)
+          ActiveSupport::Dependencies.autoload_paths = [dsl_dir_path]
+        end
+        config.dsl_file_paths.each do |f|
+          Tengine::Core.stdout_logger.debug("now loading #{f}")
+          # self.instance_eval(File.read(f), f)
+          # require(f)
+          if config.tengined.cache_drivers
+            load(f)
+          else
+            begin
+              require_dependency( config.relative_path_from_dsl_dir(f) )
+            rescue NameError => e
+              # ロードするパスのディレクトリ名をcamelizeした際に不正なモジュール名(例えば先頭が数字)
+              # だと失敗してしまうのでその場合はloadし直す
+              if e.message =~ /wrong constant name/
+                load(f)
+              else
+                raise
+              end
+            end
+          end
+        end
+      end
+    ensure
+      __teardown_core_ext__
+    end
+  end
+
+  def __safety_event__(event)
+    @__event__ = event
+    begin
+      yield if block_given?
+    ensure
+      @__event__ = nil
+      @__event_wrapper__ = nil
+    end
+  end
+
+  def __safety_driver__(driver)
+    @__driver__ = driver
+    @__session__ = driver.session
+    begin
+      yield if block_given?
+    ensure
+      @__driver__ = nil
+      @__session__ = nil
+    end
+  end
+
+  private
+
+  def __setup_core_ext__
+    Symbol.class_eval do
+      def and(other)
+        Tengine::Core::DslFilterDef.new(
+          [self.to_s, other.to_s],
+          {
+            'method' => :and,
+            'children' => [
+              { 'pattern' => self, 'method' => :find_or_mark_in_session },
+              { 'pattern' => other, 'method' => :find_or_mark_in_session },
+            ]
+          })
+      end
+      alias_method :&, :and
+
+      def at(soruce_pattern)
+        Tengine::Core::DslFilterDef.new(
+          [self.to_s],
+          {
+            'method' => :match_source_name?,
+            'pattern' => soruce_pattern.to_s,
+          })
+      end
+    end
+  end
+
+  def __teardown_core_ext__
+    Symbol.class_eval do
+      remove_method(:&, :and)
+    end
+  end
+
+  # requireではなく、ファイルを文字列としてロードしてinstance_evalで評価される場合、
+  # Proc#source_locationが返す配列の一つ目の文字列がUTF-8ではなくASCII-8BITになってしまう。
+  # そのままこれを使って検索すると、ヒットするべき検索もヒットしない。
+  # これを回避するためにUTF-8として解釈するようにString#force_encodingを使用している。
+  # http://doc.ruby-lang.org/ja/1.9.2/method/String/i/force_encoding.html
+  def __source_location__(block)
+    filepath, lineno = *block.source_location
+    filepath = filepath.dup
+    filepath.force_encoding(Encoding::UTF_8)
+    return filepath, lineno
+  end
+end