Sync Integ to Master
[sdc.git] / catalog-be / src / main / java / org / openecomp / sdc / common / transaction / impl / ESAction.java
1 /*-
2  * ============LICENSE_START=======================================================
3  * SDC
4  * ================================================================================
5  * Copyright (C) 2017 AT&T Intellectual Property. All rights reserved.
6  * ================================================================================
7  * Licensed under the Apache License, Version 2.0 (the "License");
8  * you may not use this file except in compliance with the License.
9  * You may obtain a copy of the License at
10  * 
11  *      http://www.apache.org/licenses/LICENSE-2.0
12  * 
13  * Unless required by applicable law or agreed to in writing, software
14  * distributed under the License is distributed on an "AS IS" BASIS,
15  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
16  * See the License for the specific language governing permissions and
17  * limitations under the License.
18  * ============LICENSE_END=========================================================
19  */
20
21 package org.openecomp.sdc.common.transaction.impl;
22
23 import org.openecomp.sdc.be.dao.impl.ESCatalogDAO;
24 import org.openecomp.sdc.be.resources.data.ESArtifactData;
25 import org.openecomp.sdc.be.resources.exception.ResourceDAOException;
26 import org.openecomp.sdc.be.tosca.CsarUtils;
27 import org.openecomp.sdc.common.transaction.api.IDBAction;
28 import org.openecomp.sdc.common.transaction.api.TransactionUtils.DBActionCodeEnum;
29 import org.openecomp.sdc.common.transaction.api.TransactionUtils.ESActionTypeEnum;
30 import org.slf4j.Logger;
31 import org.slf4j.LoggerFactory;
32
33 public class ESAction implements IDBAction {
34
35     private static final Logger log = LoggerFactory.getLogger(CsarUtils.class);
36
37     private ESCatalogDAO esCatalogDao;
38     private ESArtifactData artifactData;
39     private ESActionTypeEnum esActionType;
40
41     public ESAction(ESCatalogDAO esCatalogDao, ESArtifactData artifactData, ESActionTypeEnum esActiontype) {
42         this.esCatalogDao = esCatalogDao;
43         this.artifactData = artifactData;
44         this.esActionType = esActiontype;
45     }
46
47     @Override
48     public DBActionCodeEnum doAction() {
49         DBActionCodeEnum result = DBActionCodeEnum.SUCCESS;
50         try {
51             if (esActionType == ESActionTypeEnum.ADD_ARTIFACT || esActionType == ESActionTypeEnum.UPDATE_ARTIFACT) {
52                 esCatalogDao.writeArtifact(artifactData);
53             } else if (esActionType == ESActionTypeEnum.REMOVE_ARTIFACT) {
54                 esCatalogDao.deleteArtifact(artifactData.getId());
55             }
56
57         } catch (ResourceDAOException daoException) {
58             result = DBActionCodeEnum.FAIL_GENERAL;
59             log.error("#doAction - {}, es action failed with error : ", result, daoException);
60         }
61         return result;
62     }
63
64 }