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
  40   public CrudGraphDataService(GraphDao dao) throws CrudException {
  41     super();
  42     this.dao = dao;
  43     this.daoForGet = dao;
  44   }
  45
  46   public CrudGraphDataService(GraphDao dao, GraphDao daoForGet) throws CrudException {
  47     super();
  48     this.dao = dao;
  49     this.daoForGet = daoForGet;
  50   }
  51
  52   public String addVertex(String version, String type, VertexPayload payload) throws CrudException {
  53     Vertex vertex = OxmModelValidator.validateIncomingUpsertPayload(null, version, type, payload.getProperties());
  54     return addVertex(version, vertex);
  55   }
  56
  57   private String addVertex(String version, Vertex vertex) throws CrudException {
  58     Vertex addedVertex = dao.addVertex(vertex.getType(), vertex.getProperties(), version);
  59     return CrudResponseBuilder
  60         .buildUpsertVertexResponse(OxmModelValidator.validateOutgoingPayload(version, addedVertex), version);
  61   }
  62
  63   public String addEdge(String version, String type, EdgePayload payload) throws CrudException {
  64     Edge edge = RelationshipSchemaValidator.validateIncomingAddPayload(version, type, payload);
  65     return addEdge(version, edge);
  66   }
  67
  68   private String addEdge(String version, Edge edge) throws CrudException {
  69     Edge addedEdge = dao.addEdge(edge.getType(), edge.getSource(), edge.getTarget(), edge.getProperties(), version);
  70     return CrudResponseBuilder
  71         .buildUpsertEdgeResponse(RelationshipSchemaValidator.validateOutgoingPayload(version, addedEdge), version);
  72   }
  73
  74   public String updateVertex(String version, String id, String type, VertexPayload payload) throws CrudException {
  75     Vertex vertex = OxmModelValidator.validateIncomingUpsertPayload(id, version, type, payload.getProperties());
  76     return updateVertex(version, vertex);
  77
  78   }
  79
  80   private String updateVertex(String version, Vertex vertex) throws CrudException {
  81     Vertex updatedVertex = dao.updateVertex(vertex.getId().get(), vertex.getType(), vertex.getProperties(), version);
  82     return CrudResponseBuilder
  83         .buildUpsertVertexResponse(OxmModelValidator.validateOutgoingPayload(version, updatedVertex), version);
  84   }
  85
  86   public String patchVertex(String version, String id, String type, VertexPayload payload) throws CrudException {
  87     Vertex existingVertex = dao.getVertex(id, OxmModelValidator.resolveCollectionType(version, type), version);
  88     Vertex vertex = OxmModelValidator.validateIncomingPatchPayload(id, version, type, payload.getProperties(),
  89         existingVertex);
  90     return updateVertex(version, vertex);
  91   }
  92
  93   public String deleteVertex(String version, String id, String type) throws CrudException {
  94     type = OxmModelValidator.resolveCollectionType(version, type);
  95     dao.deleteVertex(id, type);
  96     return "";
  97   }
  98
  99   public String deleteEdge(String version, String id, String type) throws CrudException {
 100     RelationshipSchemaValidator.validateType(version, type);
 101     dao.deleteEdge(id, type);
 102     return "";
 103   }
 104
 105   public String updateEdge(String version, String id, String type, EdgePayload payload) throws CrudException {
 106     Edge edge = dao.getEdge(id, type);
 107     Edge validatedEdge = RelationshipSchemaValidator.validateIncomingUpdatePayload(edge, version, payload);
 108     return updateEdge(version, validatedEdge);
 109   }
 110
 111   private String updateEdge(String version, Edge edge) throws CrudException {
 112     Edge updatedEdge = dao.updateEdge(edge);
 113     return CrudResponseBuilder
 114         .buildUpsertEdgeResponse(RelationshipSchemaValidator.validateOutgoingPayload(version, updatedEdge), version);
 115   }
 116
 117   public String patchEdge(String version, String id, String type, EdgePayload payload) throws CrudException {
 118     Edge edge = dao.getEdge(id, type);
 119     Edge patchedEdge = RelationshipSchemaValidator.validateIncomingPatchPayload(edge, version, payload);
 120     return updateEdge(version, patchedEdge);
 121
 122   }
 123
 124   @Override
 125   protected Vertex addBulkVertex(Vertex vertex, String version, String dbTransId) throws CrudException {
 126     return dao.addVertex(vertex.getType(), vertex.getProperties(), version, dbTransId);
 127   }
 128
 129   @Override
 130   protected Vertex updateBulkVertex(Vertex vertex, String id, String version, String dbTransId) throws CrudException {
 131     return dao.updateVertex(id, vertex.getType(), vertex.getProperties(), version, dbTransId);
 132   }
 133
 134   @Override
 135   protected void deleteBulkVertex(String id, String version, String type, String dbTransId) throws CrudException {
 136     dao.deleteVertex(id, type, dbTransId);
 137   }
 138
 139   @Override
 140   protected Edge addBulkEdge(Edge edge, String version, String dbTransId) throws CrudException {
 141     return dao.addEdge(edge.getType(), edge.getSource(), edge.getTarget(), edge.getProperties(), version, dbTransId);
 142   }
 143
 144   @Override
 145   protected Edge updateBulkEdge(Edge edge, String version, String dbTransId) throws CrudException {
 146     return dao.updateEdge(edge, dbTransId);
 147   }
 148
 149   @Override
 150   protected void deleteBulkEdge(String id, String version, String type, String dbTransId) throws CrudException {
 151     dao.deleteEdge(id, type, dbTransId);
 152   }
 153 }