src/main/java/org/onap/crud/service/CrudGraphDataService.java

   1 /**
   2  * ============LICENSE_START=======================================================
   3  * Gizmo
   4  * ================================================================================
   5  * Copyright © 2017 AT&T Intellectual Property.
   6  * Copyright © 2017 Amdocs
   7  * All rights reserved.
   8  * ================================================================================
   9  * Licensed under the Apache License, Version 2.0 (the "License");
  10  * you may not use this file except in compliance with the License.
  11  * You may obtain a copy of the License at
  12  *
  13  *    http://www.apache.org/licenses/LICENSE-2.0
  14  *
  15  * Unless required by applicable law or agreed to in writing, software
  16  * distributed under the License is distributed on an "AS IS" BASIS,
  17  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  18  * See the License for the specific language governing permissions and
  19  * limitations under the License.
  20  * ============LICENSE_END=========================================================
  21  *
  22  * ECOMP is a trademark and service mark of AT&T Intellectual Property.
  23  */
  24 package org.onap.crud.service;
  25
  26
  27 import org.onap.crud.dao.GraphDao;
  28 import org.onap.crud.entity.Edge;
  29
  30 import org.onap.crud.entity.Vertex;
  31 import org.onap.crud.exception.CrudException;
  32 import org.onap.crud.parser.CrudResponseBuilder;
  33 import org.onap.schema.OxmModelValidator;
  34 import org.onap.schema.RelationshipSchemaValidator;
  35
  36
  37 public class CrudGraphDataService extends AbstractGraphDataService {
  38
  39   public CrudGraphDataService(GraphDao dao) throws CrudException {
  40     super(dao);
  41   }
  42
  43   public String addVertex(String version, String type, VertexPayload payload) throws CrudException {
  44     Vertex vertex = OxmModelValidator.validateIncomingUpsertPayload(null, version, type, payload.getProperties());
  45     return addVertex(version, vertex);
  46   }
  47
  48   private String addVertex(String version, Vertex vertex) throws CrudException {
  49     Vertex addedVertex = dao.addVertex(vertex.getType(), vertex.getProperties());
  50     return CrudResponseBuilder
  51         .buildUpsertVertexResponse(OxmModelValidator.validateOutgoingPayload(version, addedVertex), version);
  52   }
  53
  54   public String addEdge(String version, String type, EdgePayload payload) throws CrudException {
  55     Edge edge = RelationshipSchemaValidator.validateIncomingAddPayload(version, type, payload);
  56     return addEdge(version, edge);
  57   }
  58
  59   private String addEdge(String version, Edge edge) throws CrudException {
  60     Edge addedEdge = dao.addEdge(edge.getType(), edge.getSource(), edge.getTarget(), edge.getProperties());
  61     return CrudResponseBuilder
  62         .buildUpsertEdgeResponse(RelationshipSchemaValidator.validateOutgoingPayload(version, addedEdge), version);
  63   }
  64
  65   public String updateVertex(String version, String id, String type, VertexPayload payload) throws CrudException {
  66     Vertex vertex = OxmModelValidator.validateIncomingUpsertPayload(id, version, type, payload.getProperties());
  67     return updateVertex(version, vertex);
  68
  69   }
  70
  71   private String updateVertex(String version, Vertex vertex) throws CrudException {
  72     Vertex updatedVertex = dao.updateVertex(vertex.getId().get(), vertex.getType(), vertex.getProperties());
  73     return CrudResponseBuilder
  74         .buildUpsertVertexResponse(OxmModelValidator.validateOutgoingPayload(version, updatedVertex), version);
  75   }
  76
  77   public String patchVertex(String version, String id, String type, VertexPayload payload) throws CrudException {
  78     Vertex existingVertex = dao.getVertex(id, OxmModelValidator.resolveCollectionType(version, type));
  79     Vertex vertex = OxmModelValidator.validateIncomingPatchPayload(id, version, type, payload.getProperties(),
  80         existingVertex);
  81     return updateVertex(version, vertex);
  82   }
  83
  84   public String deleteVertex(String version, String id, String type) throws CrudException {
  85     type = OxmModelValidator.resolveCollectionType(version, type);
  86     dao.deleteVertex(id, type);
  87     return "";
  88   }
  89
  90   public String deleteEdge(String version, String id, String type) throws CrudException {
  91     RelationshipSchemaValidator.validateType(version, type);
  92     dao.deleteEdge(id, type);
  93     return "";
  94   }
  95
  96   public String updateEdge(String version, String id, String type, EdgePayload payload) throws CrudException {
  97     Edge edge = dao.getEdge(id, type);
  98     Edge validatedEdge = RelationshipSchemaValidator.validateIncomingUpdatePayload(edge, version, payload);
  99     return updateEdge(version, validatedEdge);
 100   }
 101
 102   private String updateEdge(String version, Edge edge) throws CrudException {
 103     Edge updatedEdge = dao.updateEdge(edge);
 104     return CrudResponseBuilder
 105         .buildUpsertEdgeResponse(RelationshipSchemaValidator.validateOutgoingPayload(version, updatedEdge), version);
 106   }
 107
 108   public String patchEdge(String version, String id, String type, EdgePayload payload) throws CrudException {
 109     Edge edge = dao.getEdge(id, type);
 110     Edge patchedEdge = RelationshipSchemaValidator.validateIncomingPatchPayload(edge, version, payload);
 111     return updateEdge(version, patchedEdge);
 112
 113   }
 114
 115   public Vertex getVertex(String id) throws CrudException {
 116     return dao.getVertex(id);
 117   }
 118
 119   @Override
 120   protected Vertex addBulkVertex(Vertex vertex, String version, String dbTransId) throws CrudException {
 121     return dao.addVertex(vertex.getType(), vertex.getProperties(), dbTransId);
 122   }
 123
 124   @Override
 125   protected Vertex updateBulkVertex(Vertex vertex, String id, String version, String dbTransId) throws CrudException {
 126     return dao.updateVertex(id, vertex.getType(), vertex.getProperties(), dbTransId);
 127   }
 128
 129   @Override
 130   protected void deleteBulkVertex(String id, String version, String type, String dbTransId) throws CrudException {
 131     dao.deleteVertex(id, type, dbTransId);
 132   }
 133
 134   @Override
 135   protected Edge addBulkEdge(Edge edge, String version, String dbTransId) throws CrudException {
 136     return dao.addEdge(edge.getType(), edge.getSource(), edge.getTarget(), edge.getProperties(), dbTransId);
 137   }
 138
 139   @Override
 140   protected Edge updateBulkEdge(Edge edge, String version, String dbTransId) throws CrudException {
 141     return dao.updateEdge(edge, dbTransId);
 142   }
 143
 144   @Override
 145   protected void deleteBulkEdge(String id, String version, String type, String dbTransId) throws CrudException {
 146     dao.deleteEdge(id, type, dbTransId);
 147   }
 148 }