Contents

# encoding : utf-8
require 'diy/packet'
require 'drb'
require 'thread'

module DIY
  class Worker
    
    include DRbUndumped
    
    def initialize(live)
      @live = live
      @recv_t = nil
      @start = false
      @queue = Queue.new
      @running = false
      loop_recv
      loop_callback
    end
  
    # 发包
    def inject(pkts)
      pkts.each do |pkt|
        DIY::Logger.info "send pkt: #{pkt.pretty_print}"
        @live.send_packet(pkt.content)
      end
    end
    
    def loop_recv
      @recv_t = Thread.new do
        DIY::Logger.info "start thread recving pkt..."
        @live.loop do |this, pkt|
          next unless @start
          @queue.push(pkt.body)
        end
        DIY::Logger.debug "worker: stopped loop recv"
      end
    end
    
    def loop_callback
      @running = true
      @callback_t = Thread.new do 
        #~ DIY::Logger.info "start thread callbacking pkt..."
        while @running do
          begin
            pkt = @queue.pop
            #~ DIY::Logger.info "callback: #{pkt}"
            @block.call(pkt) if @block
          rescue DRb::DRbConnError
            DIY::Logger.info "closed connection by controller"
            @queue.clear
          end
        end
        DIY::Logger.debug "stopped loop callback"
      end
    end
    
    #收包
    def ready(&block)
      DIY::Logger.info("start recv pkt")
      @block = block
      @queue.clear
      @start = true
    end
    
    # 停止收发
    def terminal
      DIY::Logger.info("stop recv pkt")
      @start = false
      @queue.clear
    end
    
    # 停止线程
    def stop
      @running = false
      @queue.push nil
      @live.break
      Utils.wait_until { @recv_t && ! @recv_t.alive? }
      Utils.wait_until { @callback_t && ! @callback_t.alive? }    
    end
    
    # 过滤器
    def filter(reg)
      @live.set_filter(reg)
    end
    
    def inspect
      "<Worker: #{@live.net}>"
    end
  
  end
end

Version data entries

3 entries across 3 versions & 1 rubygems

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