src/main/java/org/onap/nbi/apis/hub/service/CriteriaBuilderServiceOrder.java

   1 /**
   2  *     Copyright (c) 2018 Orange
   3  *
   4  *     Licensed under the Apache License, Version 2.0 (the "License");
   5  *     you may not use this file except in compliance with the License.
   6  *     You may obtain a copy of the License at
   7  *
   8  *         http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  *     Unless required by applicable law or agreed to in writing, software
  11  *     distributed under the License is distributed on an "AS IS" BASIS,
  12  *     WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  *     See the License for the specific language governing permissions and
  14  *     limitations under the License.
  15  */
  16 package org.onap.nbi.apis.hub.service;
  17
  18 import com.fasterxml.jackson.databind.JsonNode;
  19 import org.onap.nbi.apis.hub.model.Event;
  20 import org.springframework.data.mongodb.core.query.Criteria;
  21
  22 import java.util.ArrayList;
  23 import java.util.List;
  24
  25 public class CriteriaBuilderServiceOrder implements CriteriaBuilder {
  26     @Override
  27     public Criteria adjust(Criteria base, Event event) {
  28         switch (event.getEventType()) {
  29             case "ServiceOrderCreationNotification":
  30                 return base;
  31
  32             case "ServiceOrderStateChangeNotification":
  33                 JsonNode stateNode = event.getEvent().path("state");
  34                 if (stateNode.isValueNode())
  35                     return base.orOperator(
  36                             Criteria.where("query.serviceOrder__state").exists(false),
  37                             Criteria.where("query.serviceOrder__state").in(stateNode.textValue())
  38                     );
  39                 break;
  40
  41             case "ServiceOrderItemStateChangeNotification":
  42                 Object[] states = getStates(event);
  43                 if (states.length > 0)
  44                     return base.orOperator(
  45                             Criteria.where("query.serviceOrder__serviceOrderItem__state").exists(false),
  46                             Criteria.where("query.serviceOrder__serviceOrderItem__state").in(states)
  47                     );
  48                 break;
  49         }
  50
  51         return base;
  52     }
  53
  54     private String[] getStates(Event event) {
  55         List<String> states = new ArrayList<>();
  56
  57         JsonNode orderItems = event.getEvent().path("orderItem");
  58         if (orderItems.isArray()) {
  59             for (JsonNode node : orderItems) {
  60                 JsonNode stateNode = node.path("state");
  61                 if (stateNode.isValueNode()) {
  62                     states.add(stateNode.textValue());
  63                 }
  64             }
  65         }
  66
  67         return states.toArray(new String[0]);
  68     }
  69 }