#
#--
# Copyright (c) 2007, John Mettraux, OpenWFE.org
# All rights reserved.
# 
# Redistribution and use in source and binary forms, with or without 
# modification, are permitted provided that the following conditions are met:
# 
# . Redistributions of source code must retain the above copyright notice, this
#   list of conditions and the following disclaimer.  
# 
# . Redistributions in binary form must reproduce the above copyright notice, 
#   this list of conditions and the following disclaimer in the documentation 
#   and/or other materials provided with the distribution.
# 
# . Neither the name of the "OpenWFE" nor the names of its contributors may be
#   used to endorse or promote products derived from this software without
#   specific prior written permission.
# 
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" 
# AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 
# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 
# ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE 
# LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR 
# CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF 
# SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS 
# INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN 
# CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) 
# ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE 
# POSSIBILITY OF SUCH DAMAGE.
#++
#
# $Id: definitions.rb 2725 2006-06-02 13:26:32Z jmettraux $
#

#
# "made in Japan"
#
# John Mettraux at openwfe.org
#

require 'monitor'
require 'fileutils'

require 'openwfe/service'
require 'openwfe/rudefinitions'
require 'openwfe/flowexpressionid'
require 'openwfe/util/otime'
require 'openwfe/storage/yamlextras'
require 'openwfe/expool/journal_replay'


module OpenWFE
    
    #
    # Keeping a replayable track of the events in an OpenWFEru engine
    #
    class Journal < Service
        include MonitorMixin, OwfeServiceLocator
        include JournalReplay

        attr_reader :workdir

        FREQ = "1m"
            #
            # once per minute, makes sure the buckets are flushed
        
        def initialize (service_name, application_context)

            super

            @buckets = {}

            @workdir = OpenWFE::get_work_directory + "/journal"
            FileUtils.makedirs(@workdir) unless File.exist?(@workdir)

            get_expression_pool.add_observer(:all) do |event, *args|
                #ldebug { ":#{event}  for #{args[0].to_debug_s}" }
                queue_event(event, *args)
            end

            #@thread_id = get_scheduler.schedule_every(FREQ) do
            #    flush_buckets()
            #end
        end

        #
        # Will flush the journal of every open instance.
        #
        def stop
            get_scheduler.unschedule(@thread_id) if @thread_id
            flush_buckets()
        end

        protected

            def queue_event (event, *args)

                return if event == :stop
                return if event == :launch

                wfid = args[0].parent_wfid
                #puts "__#{wfid} : #{event}"

                e = serialize_event(event, *args)

                bucket = get_bucket(wfid) 
                bucket << e

                if event == :terminate
                    bucket.flush
                    @buckets.delete(wfid)
                end
            end

            def flush_buckets
                synchronize do
                end
            end

            def get_bucket (wfid)
                @buckets[wfid] ||= Bucket.new(@workdir, wfid)
            end

            def serialize_event (event, *args)
                args.insert(0, event)
                args.insert(1, Time.now)
                args.to_yaml
            end

            #
            # for each process instance, there is one bucket holding the
            # events waiting to get written in the journal
            #
            class Bucket
                include MonitorMixin

                attr_reader :wfid, :events

                def initialize (workdir, wfid)
                    super()
                    @workdir = workdir
                    @wfid = wfid
                    @events = []
                end

                def << (event)
                    synchronize do
                        @events << event
                    end
                end

                def size
                    @events.size
                end
                alias :length :size

                def flush
                    synchronize do

                        fn = @workdir + "/" + @wfid.to_s + ".journal"

                        File.open(fn, "w+") do |f|
                            @events.each do |e|
                                f.puts e
                            end
                        end
                        @events.clear
                    end
                end
            end

    end

end