add scheduler support

app/models/project.rb

 class Project < ActiveRecord::Base
-  attr_accessible :name, :path, :scripts, :timeout, :token, :default_ref, :gitlab_url
+  attr_accessible :name, :path, :scripts, :timeout, :token, :default_ref, :gitlab_url, :always_build, :polling_interval
   validates_presence_of :name, :path, :scripts, :timeout, :token, :default_ref
@@ -10,6 +10,7 @@ class Project < ActiveRecord::Base
   validates_uniqueness_of :name
   before_validation :set_default_values
+  after_save :set_scheduler
   def set_default_values
     self.token = SecureRandom.hex(15) if self.token.blank?
@@ -113,6 +114,20 @@ class Project < ActiveRecord::Base
   def valid_token? token
     self.token && self.token == token
   end
+
+  def set_scheduler
+    if self.polling_interval.present?
+      Resque.set_schedule(self.token, {
+                            :class => 'SchedulerJob',
+                            :every => self.polling_interval,
+                            :queue => 'scheduler_task',
+                            :args => [:run, self.id],
+                            :description => self.name
+                          })
+    else
+      Resque.remove_schedule(self.token)
+    end
+  end
 end
@@ -130,4 +145,3 @@ end
 #  token       :string(255)
 #  default_ref :string(255)
 #
-

app/views/projects/_form.html.haml

@@ -17,6 +17,13 @@
         .field
           = f.label :token, "Token (Leave empty to generate random token)"
           = f.text_field :token, class: 'input-xlarge', placeholder: 'xEeFCaDAB89'
+        .field
+          = f.label :always_build, "always_build"
+          = f.check_box :always_build
+        .field
+          = f.label :polling_interval, "polling_interval"
+          = f.text_field :polling_interval
+
       %fieldset
         %legend Git
         .field

config/initializers/2_resque.rb

 require Rails.root.join('lib', 'runner')
+#require Rails.root.join('lib', 'scheduler_job')
+require 'scheduler_job'
 # Custom Redis configuration
 config_file = Rails.root.join('config', 'resque.yml')

config/routes.rb

 GitlabCi::Application.routes.draw do
   # Optionally, enable Resque here
   require 'resque/server'
+  require 'resque_scheduler'
+  require 'resque_scheduler/server'
   mount Resque::Server => '/ext/resque', as: 'ext_resque'
   resources :projects do

db/migrate/20121224092350_add_schedule_to_projects.rb

+class AddScheduleToProjects < ActiveRecord::Migration
+  def change
+    add_column :projects, :always_build, :boolean, :default => true
+    add_column :projects, :polling_interval, :string
+  end
+end

lib/scheduler_job.rb

+class SchedulerJob
+  @queue = :scheduler_task
+
+  def self.perform(method, *args)
+    self.send(method, *args)
+  end
+
+  # run scheduler job
+  def self.run(project_id)
+    @project = Project.find(project_id)
+
+    # when always_build not checked, do not build project in schedule if project not updated
+    return if !@project.always_build && (@project.builds.last.sha == @project.last_ref_sha(@project.default_ref))
+
+    # always_build not checked and project updated, build it
+    # always_build checked and no matter project updated or not, build it
+    @build = @project.register_build(ref: @project.default_ref)
+    if @build and @build.id
+      Runner.new(Build.find(@build.id)).run
+    end
+  end
+end

lib/tasks/resque.rake

@@ -2,13 +2,14 @@ require 'resque/tasks'
 require 'resque_scheduler/tasks'
 namespace :resque do
-  task :setup do
+  task :setup => :environment do
     require 'resque'
     require 'resque_scheduler'
-    require 'resque/scheduler'
+    #require 'resque_scheduler/server'
+    #require 'resque/scheduler'
     # you probably already have this somewhere
-    Resque.redis = 'localhost:6379'
+    #Resque.redis = 'localhost:6379'
     # If you want to be able to dynamically change the schedule,
     # uncomment this line.  A dynamic schedule can be updated via the
@@ -16,7 +17,7 @@ namespace :resque do
     # When dynamic is set to true, the scheduler process looks for
     # schedule changes and applies them on the fly.
     # Note: This feature is only available in >=2.0.0.
-    #Resque::Scheduler.dynamic = true
+    Resque::Scheduler.dynamic = true
     # The schedule doesn't need to be stored in a YAML, it just needs to
     # be a hash.  YAML is usually the easiest.

spec/models/project_spec.rb

@@ -22,6 +22,33 @@ describe Project do
     end
   end
+  describe 'after_save_with_schedule' do
+    it 'should not set schedule if polling_interval is blank' do
+      project = FactoryGirl.create :project
+      project.update_attribute(:polling_interval, nil)
+      Resque.get_schedule(project.token).should be_nil
+    end
+
+    it 'should set schedule if polling_interval is set' do
+      project = FactoryGirl.create :project
+      project.update_attribute(:polling_interval, '3m')
+      Resque.get_schedule(project.token).should.to_s == {
+        :class => 'SchedulerJob',
+        :every => project.polling_interval,
+        :args => [:run, project.id],
+        :description => project.name
+      }.to_s
+    end
+
+    it 'should cancel schedule if clear polling_interval' do
+      project = FactoryGirl.create :project
+      project.update_attribute(:polling_interval, '3m')
+      Resque.get_schedule(project.token).should_not be_nil
+      project.update_attribute(:polling_interval, nil)
+      Resque.get_schedule(project.token).should be_nil
+    end
+  end
+
   it { should validate_presence_of :name }
   it { should validate_presence_of :scripts }
   it { should validate_presence_of :timeout }
@@ -67,4 +94,3 @@ end
 #  token       :string(255)
 #  default_ref :string(255)
 #
-