Contents

# encoding : utf-8
require 'logger'
module DIY
  class StrategyBuilder
    def initialize
      @ins = []
      @logger = DIY::Logger
    end
    attr_reader :queue
    
    def add(strategy)
      @ins << strategy
    end
    alias << add
    
    def logger=(logger)
      @logger = logger
    end
    
    def logger
      @logger
    end
    
    def call(hope_pkt, recv_pkt, queue)
      logger.debug("recv_pkt, I hope: #{ Utils.pp(hope_pkt) rescue nil }...")
      
      return if hope_pkt.nil?
      
      @ins.each do |strategy|
        begin
          ret = strategy.call(hope_pkt.content, recv_pkt.content, queue)
        rescue Exception => e
          #~ logger.error("user strategy exception: #{e.class} -> #{e.message}")
          raise StrategyCallError.new(e)
        else
          if ret == Strategy::OK
            logger.info("pkt same:")
            queue.shift
            return
          elsif ret == Strategy::OK_NO_POP
            logger.info("pkt skip:")
            return
          elsif ret == Strategy::FAIL
            logger.warn("pkt fail:")
            logger.warn("pkt fail: hope_pkt is #{hope_pkt.pretty_print}")
            logger.warn("pkt fail: recv_pkt is #{recv_pkt.pretty_print}")
            e = RuntimeError.new("Strategy FAIL: hope #{hope_pkt.pretty_print} but get #{recv_pkt.pretty_print}")
            e.set_backtrace(caller) # not used
            raise UnExpectPacketError.new(e)
          elsif ret == Strategy::NONE
            #~ logger.debug("pkt jumpped:")
            next
          elsif ret == Strategy::NONE_HOPE_POP
            queue.shift
            # skip this round if nil found
            return unless hope_pkt = queue.first
            # redo strategy
            retry
          end # end of if
        end # end of begin
      end # end of each
    end
    
  end
end

Version data entries

3 entries across 3 versions & 1 rubygems

Version	Path
DIY-pcap-0.3.5	lib/diy/strategy_builder.rb
DIY-pcap-0.3.4	lib/diy/strategy_builder.rb
DIY-pcap-0.3.3	lib/diy/strategy_builder.rb