jackross/threequel

# Public: ActiveRecord class to store log entries.
# Stores data in the threequel.log_entries table.
#
module Threequel
  class DBLoggerStorage < ActiveRecord::Base
    TABLE_NAME = "threequel.log_entries"
    self.table_name = TABLE_NAME

    def self.construct
      ActiveRecord::Migration.class_eval do
        create_table TABLE_NAME do |t|
          t.string   :stage,         :null => false, :limit => 20
          t.string   :name,          :null => false, :limit => 128
          t.string   :command,       :null => false, :limit => 128
          t.string   :statement,     :null => true,  :limit => 128
          t.text     :sql,           :null => false
          t.string   :status,        :null => false, :default => 'executing'
          t.integer  :rows_affected, :null => true
          t.text     :message,       :null => true
          t.float    :duration,      :null => true
          t.datetime :started_at,    :null => false
          t.datetime :finished_at,   :null => true
        end
      end
    end

    def self.drop
      ActiveRecord::Migration.class_eval do
        drop_table TABLE_NAME
      end
    end

    def initialize(data = {})
      super()
      assign_attributes legal_attributes(data) 
    end

    # Public: Convenience method to save and update a log entry.
    #
    # stage - A Symbol that represents the stage of execution.
    # data  - A Hash that holds the attributes to be saved in the log entry.
    #
    # Examples
    #
    #   log_execution_for :executing, { :name => "User.reset[statement0]", :command => "User.reset", :statement => "statement0", :started_at => Time.now, :sql => "UPDATE users SET logged_in = 0;\nGO" }
    #
    # Returns the Boolean result.    
    def log_execution_for(stage, data)
      self.update_attributes legal_attributes(data.merge(:stage => stage))
    end

    def legal_attributes(data)
      data.keep_if{|attr| self.has_attribute?(attr)}
    end

  end
end