Catch the newly thrown NotFoundException when checking for exisiting …

…configs
spinnaker · Nov 16, 2017 · 538709f · 538709f
1 parent 457724b
commit 538709f
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/kayenta-web/src/main/java/com/netflix/kayenta/controllers/CanaryConfigController.java b/kayenta-web/src/main/java/com/netflix/kayenta/controllers/CanaryConfigController.java
@@ -24,6 +24,7 @@
 import com.netflix.kayenta.storage.ObjectType;
 import com.netflix.kayenta.storage.StorageService;
 import com.netflix.kayenta.storage.StorageServiceRepository;
+import com.netflix.spinnaker.kork.web.exceptions.NotFoundException;
 import io.swagger.annotations.ApiOperation;
 import lombok.extern.slf4j.Slf4j;
 import org.springframework.beans.factory.annotation.Autowired;
@@ -112,7 +113,7 @@ public CanaryConfigUpdateResponse storeCanaryConfig(@RequestParam(required = fal
 
     try {
       configurationService.loadObject(resolvedConfigurationAccountName, ObjectType.CANARY_CONFIG, canaryConfigId);
-    } catch (IllegalArgumentException e) {
+    } catch (NotFoundException e) {
       configurationService.storeObject(resolvedConfigurationAccountName, ObjectType.CANARY_CONFIG, canaryConfigId, canaryConfig, canaryConfig.getName() + ".json", false);
 
       return CanaryConfigUpdateResponse.builder().canaryConfigId(canaryConfigId).build();