From ffc140f84bc97154cff835c43f9a71705f721d07 Mon Sep 17 00:00:00 2001
From: George Kalampokis <gkalampokis@cite.gr>
Date: Thu, 12 Nov 2020 15:50:34 +0200
Subject: [PATCH] When creating or updating Datasets return also the
 DatasetProfileDefinition

---
 .../web/src/main/java/eu/eudat/controllers/Datasets.java      | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/dmp-backend/web/src/main/java/eu/eudat/controllers/Datasets.java b/dmp-backend/web/src/main/java/eu/eudat/controllers/Datasets.java
index 908f3476f..becff815b 100644
--- a/dmp-backend/web/src/main/java/eu/eudat/controllers/Datasets.java
+++ b/dmp-backend/web/src/main/java/eu/eudat/controllers/Datasets.java
@@ -248,7 +248,9 @@ public class Datasets extends BaseController {
     @RequestMapping(method = RequestMethod.POST, consumes = "application/json", produces = "application/json")
     public @ResponseBody
     ResponseEntity<ResponseItem<DatasetWizardModel>> createOrUpdate(@RequestBody DatasetWizardModel profile, Principal principal) throws Exception {
-    	DatasetWizardModel dataset = new DatasetWizardModel().fromDataModel(this.datasetManager.createOrUpdate(profile, principal));
+    	Dataset datasetEntity = this.datasetManager.createOrUpdate(profile, principal);
+    	DatasetWizardModel dataset = new DatasetWizardModel().fromDataModel(datasetEntity);
+    	dataset.setDatasetProfileDefinition(this.datasetManager.getPagedProfile(dataset, datasetEntity));
         return ResponseEntity.status(HttpStatus.OK).body(new ResponseItem<DatasetWizardModel>().status(ApiMessageCode.SUCCESS_MESSAGE).message("Created").payload(dataset));
     }