Client#run_schedule returns [ScheduledJob]

treasure-data · Mar 12, 2012 · f46ad5f · f46ad5f
1 parent 2ed24f1
commit f46ad5f
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git a/lib/td/client.rb b/lib/td/client.rb
@@ -195,10 +195,12 @@ def history(name, from=nil, to=nil)
     }
   end
 
-  # TODO nil -> [ScheduledJob] or [Job]
+  # [ScheduledJob]
   def run_schedule(name, time, num)
-    @api.run_schedule(name, time, num)
+    results = @api.run_schedule(name, time, num)
-    nil
+    results.map {|job_id,type,scheduled_at|
+      ScheduledJob.new(self, scheduled_at, job_id, type, nil)
+    }
   end
 
   # => time:Flaot

diff --git a/lib/td/client/api.rb b/lib/td/client/api.rb
@@ -477,7 +477,8 @@ def run_schedule(name, time, num)
     js['jobs'].each {|m|
       job_id = m['job_id']
       scheduled_at = m['scheduled_at']
-      result << [job_id, scheduled_at]
+      type = (m['type'] || '?').to_sym
+      result << [job_id, type, scheduled_at]
     }
     return result
   end