workflow_manager 0.7.3 → 0.7.7

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 49b41b8057a7487ed992b96a9f0b81a155bef54d2fe9b63cde6c2503a84d2257
-  data.tar.gz: 373b56284df027f151a0a77f7d28fc0cd8331b51f9e2ef22e4484b0db8732154
+  metadata.gz: aff589c219150c4a2b7e8676d8540343143a12099ce4bfed847663e37f946a99
+  data.tar.gz: 64461e67991613603227c3894aa2b728db4952739071d140c007caefd8564601
 SHA512:
-  metadata.gz: df2c18b28e56ee900c2845b0ae43ee826a59fbab3b39ec6f93b5f62880b3bf2fcbc7ab4f3804899a6234e0ae5f9f7abebacd77fd24900515da31cf4d6b1215e9
-  data.tar.gz: e6ca40a3e1428284c73f578e09be5028e55bd630ce673699288d8ce3b76ccf107e5e265eba3a731401e38b7b1e01fcb5fa5e6152aeec642333cc35f681131074
+  metadata.gz: 4e79f54d8335b4df34e466055d76cb8f64fa5c53787bf83047a7e259e5d564b0514e39d2b08ba56830dc73e9235164904806a34210e76e6e1018516e1f3bd243
+  data.tar.gz: 9e3f4b1ce2adddc6b4b13e77e7bedbef74c9b635253ba77f9dbd0d731a5a6896a515404b481aa0c7f23b1a739f55142f5825fa7b7ffa8ecb8f7c7a3ba8754166

data/Gemfile

@@ -2,5 +2,5 @@ source 'https://rubygems.org'
 
 gem 'redis'
 gem 'sidekiq'
-#gem 'workflow_manager', :path => '/srv/GT/analysis/masaomi/FGCZ/prototype_workflow_manager_with_sidekiq_20210122/workflow_manager/'
-gem 'workflow_manager'
+gem 'workflow_manager', :path => './'
+#gem 'workflow_manager'

data/config/environments/redis.conf

@@ -1,9 +1,9 @@
 loglevel debug
 logfile ./logs/redis.log
-databases 4
+databases 5
 save 300 10
 rdbcompression yes
 dir ./dbs
 dbfilename redis.rdb
 maxmemory 10gb
-port 6380
+port 6379

data/lib/job_checker.rb

@@ -1,6 +1,9 @@
 require 'sidekiq'
 require 'redis'
 
+require 'uri'
+require 'net/http'
+
 WORKER_INTERVAL = 10 # [s]
 REDIS_CONF = File.expand_path("../../config/environments/redis.conf", __FILE__)
 PORT = if File.exist?(REDIS_CONF)
@@ -40,15 +43,15 @@ class JobChecker
     end
     new_job_script
   end
-  def update_time_status(status, script_basename, user, project_number)
+  def update_time_status(status, script_basename, user, project_number, next_dataset_id, rails_host)
     unless @start_time
       @start_time = Time.now.strftime("%Y-%m-%d %H:%M:%S")
     end
     time = Time.now.strftime("%Y-%m-%d %H:%M:%S")
-    [status, script_basename, [@start_time, time].join("/"), user, project_number].join(',')
+    [status, script_basename, [@start_time, time].join("/"), user, project_number, next_dataset_id, rails_host].join(',')
   end
 
-  def perform(job_id, script_basename, log_file, user, project_id)
+  def perform(job_id, script_basename, log_file, user, project_id, next_dataset_id=nil, rails_host=nil)
     puts "JobID (in JobChecker): #{job_id}"
     db0 = Redis.new(port: PORT, db: 0) # state + alpha DB
     db1 = Redis.new(port: PORT, db: 1) # log DB
@@ -63,10 +66,10 @@ class JobChecker
       #print ret
       state = ret.split(/\n/).last.strip
       #puts "state: #{state}"
-      db0[job_id] = update_time_status(state, script_basename, user, project_id)
+      db0[job_id] = update_time_status(state, script_basename, user, project_id, next_dataset_id, rails_host)
 
       unless state == pre_state
-        db0[job_id] = update_time_status(state, script_basename, user, project_id)
+        db0[job_id] = update_time_status(state, script_basename, user, project_id, next_dataset_id, rails_host)
         project_jobs = eval((db2[project_id]||[]).to_s)
         project_jobs = Hash[*project_jobs]
         project_jobs[job_id] = state
@@ -76,6 +79,11 @@ class JobChecker
       pre_state = state
       sleep WORKER_INTERVAL
     end while state =~ /RUNNING/  or state =~ /PENDING/ or state =~ /---/
+    if next_dataset_id and rails_host
+      uri = URI("#{rails_host}/data_set/#{next_dataset_id}/update_completed_samples")
+      #p uri
+      res = Net::HTTP.get_response(uri)
+    end
   end
 end
 

data/lib/workflow_manager/server.rb

@@ -163,6 +163,7 @@ module WorkflowManager
                     RedisDB.new(1, @redis_conf)
                 end
       @jobs = RedisDB.new(2, @redis_conf)
+      @trees = RedisDB.new(4, @redis_conf)
 
       @system_log = File.join(@log_dir, "system.log")
       @mutex = Mutex.new
@@ -186,12 +187,12 @@ module WorkflowManager
         statuses.each do |job_id, status|
           # puts [job_id, status].join(",")
           # 120249,RUNNING,QC_ventricles_100k.sh,2021-07-30 09:47:04/2021-07-30 09:47:04,masaomi,1535
-          stat, script_basename, time, user, project_number = status.split(",")
+          stat, script_basename, time, user, project_number, next_dataset_id, rails_host = status.split(",")
           if stat == "RUNNING" or stat == "PENDING"
             log_file = logs[job_id]
             log_puts("JobID (in recovery check): #{job_id}")
             puts "JobID (in recovery check): #{job_id}"
-            JobChecker.perform_async(job_id, script_basename, log_file, user, project_number)
+            JobChecker.perform_async(job_id, script_basename, log_file, user, project_number, next_dataset_id, rails_host)
           end
         end
       end
@@ -296,7 +297,7 @@ module WorkflowManager
         Thread.current.kill
       end
     end
-    def start_monitoring3(script_path, script_content, user='sushi_lover', project_number=0, sge_options='', log_dir='')
+    def start_monitoring3(script_path, script_content, user='sushi_lover', project_number=0, sge_options='', log_dir='', next_dataset_id='', rails_host=nil)
       script_basename = File.basename(script_path)
       job_id, log_file, command = @cluster.submit_job(script_path, script_content, sge_options)
       #p command
@@ -304,7 +305,7 @@ module WorkflowManager
       #p job_id
       puts "JobID (in WorkflowManager): #{job_id}"
       sleep 1
-      JobChecker.perform_async(job_id, script_basename, log_file, user, project_number)
+      JobChecker.perform_async(job_id, script_basename, log_file, user, project_number, next_dataset_id, rails_host)
       job_id
     end
     def start_monitoring2(script_path, script_content, user='sushi_lover', project_number=0, sge_options='', log_dir='')
@@ -473,17 +474,23 @@ module WorkflowManager
       job_idsh = if job_ids
                    Hash[*(job_ids.split(',')).map{|job_id| [job_id, true]}.flatten]
                  end
-      s_ = {}
-      unless job_ids
+      if project_number
+        s_ = {}
         @jobs.transaction do |jobs|
           if project_jobs = jobs[project_number]
             s_ = Hash[*eval(project_jobs)]
           end
         end
-      end
-      @statuses.transaction do |statuses|
-        s_.each do |job_id, stat|
-          s << [job_id, statuses[job_id]]
+        @statuses.transaction do |statuses|
+          s_.each do |job_id, stat|
+            s << [job_id, statuses[job_id]]
+          end
+        end
+      else
+        @statuses.transaction do |statuses|
+          statuses.each do |key, value|
+            s << [key, value]
+          end
         end
       end
       if job_ids
@@ -549,6 +556,19 @@ module WorkflowManager
     def cluster_node_list
       @cluster.node_list
     end
+    def save_dataset_tree(project_number, json)
+      @trees.transaction do |trees|
+        trees[project_number] = json
+      end
+      json
+    end
+    def load_dataset_tree(project_number)
+      json = nil
+      @trees.transaction do |trees|
+        json = trees[project_number]
+      end
+      json
+    end
   end
 end
 

data/lib/workflow_manager/version.rb

@@ -1,3 +1,3 @@
 module WorkflowManager
-  VERSION = "0.7.3"
+  VERSION = "0.7.7"
 end

data/start_workflow_manager.sh

@@ -0,0 +1,11 @@
+#!/usr/bin/bash
+source /usr/local/ngseq/etc/lmod_profile
+module load Dev/Ruby/2.6.7
+module load Tools/Redis/6.0.1
+conda activate gtools_env
+which python
+which g-sub
+which g-req
+mkdir -p logs
+mkdir -p dbs
+bundle exec workflow_manager -d druby://fgcz-h-032:40002

data/test/job_list.rb

@@ -1,13 +1,13 @@
 #!/usr/bin/env ruby
 # encoding: utf-8
-# Version = '20210723-134812'
+# Version = '20211001-104513'
 
 PORT = (ARGV[0]||6380).to_i
 require 'redis'
 db0 = Redis.new(port: PORT, db: 0)
 db1 = Redis.new(port: PORT, db: 1)
 db2 = Redis.new(port: PORT, db: 2)
-#db3 = Redis.new(port: 6380, db: 3)
+db4 = Redis.new(port: PORT, db: 4)
 
 class Redis
   def show_all
@@ -18,8 +18,8 @@ class Redis
   end
 end
 
-dbs = [db0, db1, db2]
-db_notes = ["state DB", "log DB", "project job DB"]
+dbs = [db0, db1, db2, db4]
+db_notes = ["state DB", "log DB", "project job DB", "JS tree DB"]
 
 dbs.each.with_index do |db, i|
   note = db_notes[i]
@@ -48,3 +48,10 @@ db2.keys.sort.each do |key|
   value = db2.get(key)
   puts [key, value].join("\t")
 end
+
+puts
+puts "db3, status DB3, project specific"
+db3.keys.sort.each do |key|
+  value = db3.get(key)
+  puts [key, value].join("\t")
+end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: workflow_manager
 version: !ruby/object:Gem::Version
-  version: 0.7.3
+  version: 0.7.7
 platform: ruby
 authors:
 - Functional Genomics Center Zurich
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2021-07-31 00:00:00.000000000 Z
+date: 2021-10-15 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -87,6 +87,7 @@ files:
 - spec/cluster_spec.rb
 - spec/server_spec.rb
 - spec/spec_helper.rb
+- start_workflow_manager.sh
 - test/call_worker4.rb
 - test/call_worker_method.rb
 - test/job_list.rb