khronos 0.0.0 → 0.0.1.pre1

data/lib/khronos/storage.rb

@@ -1,26 +1,47 @@
-require 'delegate'
 require 'uri'
 
 module Khronos
-  class Storage < Delegator
-    attr_reader :uri, :adapter
+  class Storage
+    autoload :Adapter, 'khronos/storage/adapter'
 
-    def initialize(url=ENV['KHRONOS_STORAGE'])
-      unless url.nil?
-        @uri = URI.parse(url)
+    def initialize(uri=ENV['KHRONOS_STORAGE'])
+      raise RuntimeError.new("Please configure 'KHRONOS_STORAGE' on your environment variables.") if uri.nil?
 
-        require "khronos/adapter/#{@uri.scheme}"
-        __setobj__(Adapter.const_get(@uri.scheme.capitalize).new(@uri))
-      end
+      @adapter = Adapter.get(uri)
+      self.migrate! if @adapter.name =~ /ActiveRecord/
+      self.class.send(:include, @adapter)
     end
 
-    def __getobj__
-      @adapter
+    def logger=(logger)
+      #
+      # Change logger current adapter's logger
+      #
+      raise NotImplementedMethod
     end
 
-    def __setobj__(adapter)
-      @adapter = adapter
+    def truncate!
+      Schedule.delete_all
+      ScheduleLog.delete_all
     end
 
+    protected
+
+      def migrate!
+        require 'khronos/storage/adapter/activerecord/migrations/schedule'
+        require 'khronos/storage/adapter/activerecord/migrations/schedule_log'
+
+        unless ActiveRecord::Base.connection.table_exists?(:schedules)
+          Adapter::ActiveRecord::CreateSchedule.up
+        else
+          Logger.debug "Schedules table already exists."
+        end
+
+        unless ActiveRecord::Base.connection.table_exists?(:schedule_logs)
+          Adapter::ActiveRecord::CreateScheduleLog.up
+        else
+          Logger.debug "ScheduleLog table already exists."
+        end
+      end
+
   end
 end

data/lib/khronos/storage/adapter.rb

@@ -0,0 +1,44 @@
+module Khronos
+  class Storage
+    module Adapter
+      @frameworks = {
+        'mongodb' => 'mongoid',
+        'postgresql' => 'activerecord',
+        'mysql2' => 'activerecord',
+        'sqlite3' => 'activerecord'
+      }
+      @classes = {
+        'activerecord' => 'ActiveRecord',
+        'mongoid' => 'Mongoid'
+      }
+      @adapters = {
+        'mongodb' => 'mongoid'
+      }
+
+      def self.parse_uri(uri)
+        data = URI.parse(uri)
+        {
+          :scheme     => data.scheme,
+          :host       => data.host,
+          :user       => data.user,
+          :password   => data.password,
+          :path       => data.path
+        }
+      end
+
+      def self.get(url)
+        uri = parse_uri(url)
+        framework = @frameworks[uri[:scheme]]
+
+        require "khronos/storage/adapter/#{framework}"
+
+        # Get and connect with the adapter class.
+        adapter = const_get(@classes[framework])
+
+        # uri.merge(:adapter => @adapters[uri[:scheme]] || uri[:scheme])
+        adapter.connect!(url)
+      end
+
+    end
+  end
+end

data/lib/khronos/storage/adapter/activerecord.rb

@@ -0,0 +1,39 @@
+require 'active_record'
+
+module Khronos
+  class Storage
+    module Adapter
+
+      module ActiveRecord
+        autoload :Schedule,     'khronos/storage/adapter/activerecord/schedule'
+        autoload :ScheduleLog,  'khronos/storage/adapter/activerecord/schedule_log'
+
+        def self.connect!(url)
+          puts "Connect! => #{url.inspect}"
+          if File.exists?("config/database.yml")
+            ::ActiveRecord::Base.establish_connection(YAML.load_file("config/database.yml")[ENV['RACK_ENV']])
+          else
+            ::ActiveRecord::Base.establish_connection(url)
+          end
+
+          #
+          # ::ActiveRecord::Base.logger = ::Logger.new(STDOUT)
+          #
+          self
+        end
+
+        def self.included(base)
+          #puts "included in #{base}"
+        end
+
+        def self.extended(base)
+          #puts "extended in #{base.inspect}"
+        end
+
+      end
+
+    end
+  end
+end
+
+

data/lib/khronos/storage/adapter/activerecord/migrations/schedule.rb

@@ -0,0 +1,29 @@
+module Khronos
+  class Storage
+    module Adapter
+      module ActiveRecord
+
+        class CreateSchedule < ::ActiveRecord::Migration
+          def self.up
+            create_table :schedules do |t|
+              t.string    :context,    :null => false, :limit => 100
+              t.datetime  :at,         :null => false
+              t.string    :task_url,   :null => false
+              t.integer   :recurrency, :null => false
+              t.string    :callbacks,  :null => true,  :limit => 500
+              t.boolean   :active,     :null => false, :default => true
+            end
+
+            add_index :schedules, :at
+            add_index :schedules, :context
+          end
+
+          def self.down
+            drop_table :schedules
+          end
+        end
+
+      end
+    end
+  end
+end

data/lib/khronos/storage/adapter/activerecord/migrations/schedule_log.rb

@@ -0,0 +1,23 @@
+module Khronos
+  class Storage
+    module Adapter
+      module ActiveRecord
+
+        class CreateScheduleLog < ::ActiveRecord::Migration
+          def self.up
+            create_table :schedule_logs do |t|
+              t.integer  :schedule_id,   :null => false
+              t.datetime :started_at,    :null => false
+              t.datetime :finished_at,   :null => true
+            end
+          end
+
+          def self.down
+            drop_table :schedule_logs
+          end
+        end
+
+      end
+    end
+  end
+end

data/lib/khronos/storage/adapter/activerecord/schedule.rb

@@ -0,0 +1,24 @@
+require 'json'
+
+module Khronos
+  class Storage
+    module Adapter
+
+      module ActiveRecord
+        class Schedule < ::ActiveRecord::Base
+          attr_accessible :context, :at, :recurrency, :task_url, :callbacks, :active
+          has_many :logs, :class_name => ScheduleLog
+
+          def callbacks=(options)
+            write_attribute(:callbacks, options.to_json)
+          end
+
+          def callbacks
+            JSON.parse(read_attribute(:callbacks) || '{}')
+          end
+        end
+      end
+
+    end
+  end
+end

data/lib/khronos/storage/adapter/activerecord/schedule_log.rb

@@ -0,0 +1,14 @@
+module Khronos
+  class Storage
+    module Adapter
+
+      module ActiveRecord
+        class ScheduleLog < ::ActiveRecord::Base
+          attr_accessible :schedule_id, :started_at, :finished_at
+          belongs_to :schedule
+        end
+      end
+
+    end
+  end
+end

data/lib/khronos/storage/adapter/mongoid.rb

@@ -0,0 +1,37 @@
+require 'mongoid'
+require 'mongo'
+
+module Khronos
+  class Storage
+    module Adapter
+
+      module Mongoid
+        autoload :Schedule,     'khronos/storage/adapter/mongoid/schedule'
+        autoload :ScheduleLog,  'khronos/storage/adapter/mongoid/schedule_log'
+
+        def self.connect!(uri)
+          if File.exists?("config/mongoid.yml")
+            ::Mongoid.load!("config/mongoid.yml")
+          else
+            ::Mongoid.configure do |config|
+              config.connect_to uri[:host][1..-1]
+            end
+          end
+
+          self
+        end
+
+        def self.included(base)
+          #puts "included in #{base}"
+        end
+
+        def self.extended(base)
+          #puts "extended in #{base.inspect}"
+        end
+
+      end
+
+    end
+  end
+end
+

data/lib/khronos/storage/adapter/{mongo → mongoid}/schedule.rb

@@ -1,15 +1,17 @@
 module Khronos
-  module Storage
+  class Storage
     module Adapter
 
-      module Mongo
+      module Mongoid
         class Schedule
-          include Mongoid::Document
+          include ::Mongoid::Document
 
           field :namespace,   :type => String
           field :context,     :type => String
           field :at,          :type => DateTime
-          field :recurring,   :type => Integer
+          field :recurrency,  :type => Integer
+          field :callbacks,   :type => Hash
+          field :active,      :type => Boolean
 
           has_many :logs, :class_name => ScheduleLog
         end

data/lib/khronos/storage/adapter/{mongo → mongoid}/schedule_log.rb

@@ -1,11 +1,10 @@
-
 module Khronos
-  module Storage
+  class Storage
     module Adapter
 
-      module Mongo
+      module Mongoid
         class ScheduleLog
-          include Mongoid::Document
+          include ::Mongoid::Document
 
           field :started_at,  :type => DateTime
           field :finished_at, :type => DateTime

data/lib/khronos/tasks/db.rake

@@ -0,0 +1,25 @@
+require 'khronos/storage/adapter/activerecord/migrations/schedule'
+require 'khronos/storage/adapter/activerecord/migrations/schedule_log'
+
+namespace :db do
+
+  desc 'Create the database.'
+  task :create do
+    adapter = Khronos::Storage::Adapter.get(ENV['KHRONOS_STORAGE'])
+    if adapter.name =~ /ActiveRecord/
+      CreateSchedule.up
+      CreateScheduleLog.up
+    end
+  end
+
+  desc 'Destroy entire database.'
+  task :drop do
+    adapter = Khronos::Storage::Adapter.get(ENV['KHRONOS_STORAGE'])
+    if adapter.name =~ /ActiveRecord/
+      CreateSchedule.down
+      CreateScheduleLog.down
+    end
+  end
+
+end
+

data/lib/khronos/version.rb

@@ -1,3 +1,3 @@
 module Khronos
-  VERSION = '0.0.0'
+  VERSION = '0.0.1.pre1'
 end

data/server.rb

@@ -0,0 +1,29 @@
+$: << 'lib'
+ENV['RACK_ENV'] = 'development'
+
+require 'rubygems'
+require 'bundler/setup'
+require 'active_support/all'
+require 'khronos'
+
+#controller = Khronos::Controller.new
+#controller.start!
+
+case ARGV[0]
+when 'runner'
+  puts "Running: Runner."
+
+  require 'khronos/server/runner'
+  EventMachine.run {
+    EventMachine.start_server Khronos::Config.instance.runner['host'], Khronos::Config.instance.runner['port'], Khronos::Server::Runner
+  }
+
+when 'scheduler'
+  puts "Running: Scheduler."
+  run Khronos::Server::Scheduler
+
+when 'controller'
+  controller = Khronos::Controller.new
+  controller.start!
+
+end

data/spec/functional/adapters_spec.rb

@@ -0,0 +1,61 @@
+require 'spec_helper'
+
+describe Khronos::Storage do
+  subject { Khronos::Storage }
+
+  describe Khronos::Storage::Adapter do
+
+    it "should identify activerecord adapter for postgres" do
+      client = subject.new('postgresql://localhost:5432/khronos')
+      Khronos::Storage::Schedule.should == Khronos::Storage::Adapter::ActiveRecord::Schedule
+      Khronos::Storage::ScheduleLog.should == Khronos::Storage::Adapter::ActiveRecord::ScheduleLog
+      Khronos::Storage::Schedule.create({
+        :context => "test:dummy",
+        :task_url => "http://some-service.com/task",
+        :at => Time.now,
+        :recurrency => 1.day
+      })
+      Khronos::Storage::Schedule.last.task_url.should == "http://some-service.com/task"
+    end
+
+    xit "should identify activerecord adapter for mysql" do
+      client = subject.new('mysql2://localhost:3306/khronos')
+      Khronos::Storage::Schedule.should == Khronos::Storage::Adapter::ActiveRecord::Schedule
+      Khronos::Storage::ScheduleLog.should == Khronos::Storage::Adapter::ActiveRecord::ScheduleLog
+      Khronos::Storage::Schedule.create({
+        :context => "test:dummy",
+        :task_url => "http://some-service.com/task",
+        :at => Time.now,
+        :recurrency => 1.day
+      })
+      Khronos::Storage::Schedule.last.task_url.should == "http://some-service.com/task"
+    end
+
+    it "should identify activerecord adapter for sqlite3" do
+      client = subject.new('sqlite3://localhost/spec/tmp/sqlite3.db')
+      Khronos::Storage::Schedule.should == Khronos::Storage::Adapter::ActiveRecord::Schedule
+      Khronos::Storage::ScheduleLog.should == Khronos::Storage::Adapter::ActiveRecord::ScheduleLog
+      Khronos::Storage::Schedule.create({
+        :context => "test:dummy",
+        :task_url => "http://some-service.com/task",
+        :at => Time.now,
+        :recurrency => 1.day
+      })
+      Khronos::Storage::Schedule.last.task_url.should == "http://some-service.com/task"
+    end
+
+    xit "should identify mongodb adapter" do
+      client = subject.new('mongodb://127.0.0.1:6379/test')
+      Khronos::Storage::Schedule.should == Khronos::Storage::Adapter::Mongoid::Schedule
+      Khronos::Storage::ScheduleLog.should == Khronos::Storage::Adapter::Mongoid::ScheduleLog
+      Khronos::Storage::Schedule.create({
+        :context => "test:dummy",
+        :task_url => "http://some-service.com/task",
+        :at => Time.now,
+        :recurrency => 1.day
+      })
+      Khronos::Storage::Schedule.last.task_url.should == "http://some-service.com/task"
+    end
+
+  end
+end

data/spec/functional/controller_spec.rb

@@ -0,0 +1,11 @@
+require 'spec_helper'
+
+describe Khronos::Server::Controller do
+  let(:controller) { Khronos::Server::Controller.new }
+
+  it "should retrieve" do
+    #controller.
+  end
+
+end
+