From 744acd7abf4231a2db9711038a86807b6812dd7e Mon Sep 17 00:00:00 2001
From: George Kalampokis <gkalampokis@cite.gr>
Date: Tue, 21 Apr 2020 12:41:37 +0300
Subject: [PATCH] Fixed issue when creating new Datasets

---
 .../elastic/repository/DatasetRepository.java     | 15 ++++++++++-----
 1 file changed, 10 insertions(+), 5 deletions(-)

diff --git a/dmp-backend/elastic/src/main/java/eu/eudat/elastic/repository/DatasetRepository.java b/dmp-backend/elastic/src/main/java/eu/eudat/elastic/repository/DatasetRepository.java
index 34a314bb1..ebf02c512 100644
--- a/dmp-backend/elastic/src/main/java/eu/eudat/elastic/repository/DatasetRepository.java
+++ b/dmp-backend/elastic/src/main/java/eu/eudat/elastic/repository/DatasetRepository.java
@@ -49,14 +49,19 @@ public class DatasetRepository extends ElasticRepository<Dataset, DatasetCriteri
             XContentBuilder builder = XContentFactory.jsonBuilder();
             Dmp dmp = this.dmpRepository.findDocument(entity.getDmp().toString());
             boolean found = false;
-            for(int i = 0; i < dmp.getDatasets().size(); i++) {
-                if (dmp.getDatasets().get(i).getId().equals(entity.getId())) {
-                    dmp.getDatasets().set(i, entity);
-                    found = true;
-                    break;
+            if (dmp.getDatasets() != null && !dmp.getDatasets().isEmpty()) {
+                for (int i = 0; i < dmp.getDatasets().size(); i++) {
+                    if (dmp.getDatasets().get(i).getId().equals(entity.getId())) {
+                        dmp.getDatasets().set(i, entity);
+                        found = true;
+                        break;
+                    }
                 }
             }
             if (!found) {
+                if (dmp.getDatasets() == null) {
+                    dmp.setDatasets(new ArrayList<>());
+                }
                 dmp.getDatasets().add(entity);
             }
             IndexRequest request = new IndexRequest("dmps").id(dmp.getId().toString()).source(dmp.toElasticEntity(builder));//new IndexRequest("datasets", "doc", entity.getId()).source(entity.toElasticEntity(builder));