diff --git a/server/website/website/models.py b/server/website/website/models.py index 3954a85..f51dc75 100644 --- a/server/website/website/models.py +++ b/server/website/website/models.py @@ -185,7 +185,8 @@ class Session(BaseModel): description = models.TextField(null=True, blank=True) dbms = models.ForeignKey(DBMSCatalog) hardware = models.ForeignKey(Hardware) - algorithm = models.IntegerField(choices=AlgorithmType.choices()) + algorithm = models.IntegerField(choices=AlgorithmType.choices(), + default=AlgorithmType.OTTERTUNE) project = models.ForeignKey(Project) creation_time = models.DateTimeField() diff --git a/server/website/website/views.py b/server/website/website/views.py index 7637896..2895839 100644 --- a/server/website/website/views.py +++ b/server/website/website/views.py @@ -527,9 +527,17 @@ def handle_result_files(session, files): return HttpResponse("Result stored successfully!") result_id = result.pk - response = chain(aggregate_target_results.s(result.pk), - map_workload.s(), - configuration_recommendation.s()).apply_async() + response = None + if session.algorithm == AlgorithmType.OTTERTUNE: + response = chain(aggregate_target_results.s(result.pk), + map_workload.s(), + configuration_recommendation.s()).apply_async() + elif session.algorithm == AlgorithmType.ALGORITHM1: + pass + elif session.algorithm == AlgorithmType.ALGORITHM2: + pass + elif session.algorithm == AlgorithmType.ALGORITHM3: + pass taskmeta_ids = [response.parent.parent.id, response.parent.id, response.id] result.task_ids = ','.join(taskmeta_ids) result.save()