From b349cb51c4e46e1ae1bb309e9c589f1cff89e5af Mon Sep 17 00:00:00 2001 From: gebele Date: Fri, 2 Aug 2019 10:57:06 +0000 Subject: optimize upload task with wait for dataset id --- application.rb | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) (limited to 'application.rb') diff --git a/application.rb b/application.rb index 1c738f1..965b7ec 100644 --- a/application.rb +++ b/application.rb @@ -180,14 +180,14 @@ post '/predict/?' do uploadDataset = Task.run do t = uploadTask t.update_percent(1) - $logger.debug "Processing '#{params[:fileselect][:filename]}'" - @input = Dataset.from_csv_file File.join("tmp", params[:fileselect][:filename]) + puts "Processing '#{params[:fileselect][:filename]}'" + input = Dataset.from_csv_file File.join("tmp", params[:fileselect][:filename]) + t[:dataset_id] = input.id t.update_percent(100) t.save end @upid = uploadTask.id - #TODO route for compound size @compounds_size = 0 #@input.compounds.size @models = params[:selection].keys @tasks = [] @@ -201,12 +201,12 @@ post '/predict/?' do prediction = {} model = Model::Validation.find model_id t.update_percent(10) - input = Dataset.find_by(:source => "tmp/"+@filename) - until input + until uploadTask.dataset_id sleep 1 - input = Dataset.find_by(:source => "tmp/"+@filename) + uploadTask = Task.find @upid end - prediction_dataset = model.predict input + @input = Dataset.find uploadTask.dataset_id + prediction_dataset = model.predict @input t.update_percent(70) t[:dataset_id] = prediction_dataset.id t.update_percent(75) -- cgit v1.2.3