X-Git-Url: https://gerrit.onap.org/r/gitweb?a=blobdiff_plain;f=appc-sequence-generator%2Fappc-sequence-generator-bundle%2Fsrc%2Fmain%2Fjava%2Forg%2Fonap%2Fappc%2Fseqgen%2Fobjects%2FSequenceGeneratorInputBuilder.java;h=6066a9af18b4ca352fdbffcd2926579f00b6c6fa;hb=c751a9532c263b542f7f420071c545844fa56dc7;hp=21b8f7b55d96f7a628a421a017681989b3434fa0;hpb=796ca12ab0fba32db8b4a0dff4cc4e972652c35d;p=appc.git

diff --git a/appc-sequence-generator/appc-sequence-generator-bundle/src/main/java/org/onap/appc/seqgen/objects/SequenceGeneratorInputBuilder.java b/appc-sequence-generator/appc-sequence-generator-bundle/src/main/java/org/onap/appc/seqgen/objects/SequenceGeneratorInputBuilder.java
index 21b8f7b55..6066a9af1 100644
--- a/appc-sequence-generator/appc-sequence-generator-bundle/src/main/java/org/onap/appc/seqgen/objects/SequenceGeneratorInputBuilder.java
+++ b/appc-sequence-generator/appc-sequence-generator-bundle/src/main/java/org/onap/appc/seqgen/objects/SequenceGeneratorInputBuilder.java
@@ -40,21 +40,13 @@ public class SequenceGeneratorInputBuilder {
 
     private Map<String,String> tunableParams;
 
-    private Map<String,List<String>> capability;
-
+    private CapabilityModel capabilityModel;
+    
     public SequenceGeneratorInputBuilder requestInfo(RequestInfo requestInfo){
         this.requestInfo = requestInfo;
         return this;
     }
 
-    public SequenceGeneratorInputBuilder capability(String level,List<String> capabilities){
-        if(this.capability ==null){
-            this.capability = new HashMap<>();
-        }
-        this.capability.put(level,capabilities);
-        return this;
-    }
-
     public SequenceGeneratorInputBuilder tunableParameter(String key,String value){
         if(this.tunableParams ==null){
             this.tunableParams = new HashMap<>();
@@ -73,10 +65,14 @@ public class SequenceGeneratorInputBuilder {
         return this;
     }
 
+    public SequenceGeneratorInputBuilder capabilityModel(CapabilityModel model) {
+        this.capabilityModel = model;
+        return this;
+    }
     public SequenceGeneratorInput build(){
         SequenceGeneratorInput input = new SequenceGeneratorInput();
         input.setRequestInfo(this.requestInfo);
-        input.setCapability(this.capability);
+        input.setCapabilityModel(this.capabilityModel);
         input.setInventoryModel(this.inventoryModel);
         input.setDependencyModel(this.dependencyModel);
         input.setTunableParams(this.tunableParams);