Class: Ml::ExperimentTracking::AssociateMlCandidateToPackageWorker

Inherits:
Object
  • Object
show all
Includes:
Gitlab::EventStore::Subscriber
Defined in:
app/workers/ml/experiment_tracking/associate_ml_candidate_to_package_worker.rb

Class Method Summary collapse

Instance Method Summary collapse

Methods included from Gitlab::EventStore::Subscriber

#perform

Class Method Details

.handles_event?(event) ⇒ Boolean

Returns:

  • (Boolean)


21
22
23
# File 'app/workers/ml/experiment_tracking/associate_ml_candidate_to_package_worker.rb', line 21

def self.handles_event?(event)
  event.generic? && Ml::Experiment.package_for_experiment?(event.data[:name])
end

Instance Method Details

#handle_event(event) ⇒ Object



13
14
15
16
17
18
19
# File 'app/workers/ml/experiment_tracking/associate_ml_candidate_to_package_worker.rb', line 13

def handle_event(event)
  return unless (candidate = Ml::Candidate.with_project_id_and_iid(event.data[:project_id], event.data[:version]))
  return unless (package = Packages::Package.find_by_id(event.data[:id]))

  candidate.package = package
  candidate.save!
end