From db25d962e9bb39e8aa76c0e331679e0bc697572b Mon Sep 17 00:00:00 2001
From: Mirco Nasuti <mirco.nasuti@chuv.ch>
Date: Thu, 14 Jul 2016 11:07:17 +0200
Subject: [PATCH] bugfixes

---
 src/main/java/org/hbp/mip/controllers/ModelsApi.java | 12 ++++++++----
 src/main/resources/db/migration/V1_1__Create.sql     |  2 +-
 2 files changed, 9 insertions(+), 5 deletions(-)

diff --git a/src/main/java/org/hbp/mip/controllers/ModelsApi.java b/src/main/java/org/hbp/mip/controllers/ModelsApi.java
index 7d31f7e81..21acd04af 100644
--- a/src/main/java/org/hbp/mip/controllers/ModelsApi.java
+++ b/src/main/java/org/hbp/mip/controllers/ModelsApi.java
@@ -9,6 +9,7 @@ import io.swagger.annotations.*;
 import org.apache.log4j.Logger;
 import org.hbp.mip.configuration.SecurityConfiguration;
 import org.hbp.mip.model.*;
+import org.hbp.mip.repositories.ConfigRepository;
 import org.hbp.mip.repositories.DatasetRepository;
 import org.hbp.mip.repositories.ModelRepository;
 import org.hbp.mip.repositories.QueryRepository;
@@ -46,6 +47,9 @@ public class ModelsApi {
     @Autowired
     QueryRepository queryRepository;
 
+    @Autowired
+    ConfigRepository configRepository;
+
     private static final String DATA_FILE = "data/values.csv";
 
     @ApiOperation(value = "Get models", response = Model.class, responseContainer = "List")
@@ -75,6 +79,7 @@ public class ModelsApi {
             for (Iterator<Model> i = models.iterator(); i.hasNext(); )
             {
                 Model m = i.next();
+                m.setDataset(datasetRepository.findOne(m.getDataset().getCode()));
                 if(valid != m.getValid())
                 {
                     i.remove();
@@ -82,10 +87,6 @@ public class ModelsApi {
             }
         }
 
-        for(Model model:models){
-            model.setDataset(datasetRepository.findOne(model.getDataset().getCode()));
-        }
-
         return new ResponseEntity<List<Model>>(HttpStatus.OK).ok(models);
 
     }
@@ -153,6 +154,9 @@ public class ModelsApi {
         map.put("text", model.getTitle());
         model.getConfig().setTitle(map);
 
+        configRepository.save(model.getConfig());
+        queryRepository.save(model.getQuery());
+        datasetRepository.save(model.getDataset());
         modelRepository.save(model);
 
         return new ResponseEntity<Model>(HttpStatus.CREATED).ok(model);
diff --git a/src/main/resources/db/migration/V1_1__Create.sql b/src/main/resources/db/migration/V1_1__Create.sql
index 777f0f39d..2dfa346e9 100644
--- a/src/main/resources/db/migration/V1_1__Create.sql
+++ b/src/main/resources/db/migration/V1_1__Create.sql
@@ -66,7 +66,7 @@ ALTER TABLE app OWNER TO postgres;
 
 CREATE TABLE article (
     slug character varying(255) NOT NULL,
-    _abstract text,
+    abstract text,
     content text,
     createdat timestamp without time zone,
     publishedat timestamp without time zone,
-- 
GitLab