music-core/src/main/java/org/onap/music/datastore/Condition.java

   1 /*
   2  * ============LICENSE_START==========================================
   3  * org.onap.music
   4  * ===================================================================
   5  *  Copyright (c) 2017 AT&T Intellectual Property
   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  *
  19  * ============LICENSE_END=============================================
  20  * ====================================================================
  21  */
  22
  23 package org.onap.music.datastore;
  24
  25 import java.util.Map;
  26 import org.onap.music.exceptions.MusicServiceException;
  27 import org.onap.music.main.MusicCore;
  28
  29 import com.datastax.driver.core.ResultSet;
  30 import com.datastax.driver.core.Row;
  31
  32 public class Condition {
  33         private Map<String, Object> conditions;
  34         private PreparedQueryObject selectQueryForTheRow;
  35
  36         public Condition(Map<String, Object> conditions, PreparedQueryObject selectQueryForTheRow) {
  37             this.conditions = conditions;
  38             this.selectQueryForTheRow = selectQueryForTheRow;
  39         }
  40
  41         public boolean testCondition() throws Exception {
  42             // first generate the row
  43             ResultSet results = quorumGet(selectQueryForTheRow);
  44             Row row = null;
  45             if(results != null) {
  46                 row = results.one();
  47             }
  48             if(row == null) {
  49                 throw new Exception(" No data found to update");
  50             }
  51             return getDSHandle().doesRowSatisfyCondition(row, conditions);
  52         }
  53
  54         /* For JUnit testing only */
  55         public ResultSet quorumGet(PreparedQueryObject selectQueryForTheRow) {
  56             return MusicCore.quorumGet(selectQueryForTheRow);
  57         }
  58
  59         /* For JUnit testing only */
  60         public MusicDataStore getDSHandle() throws MusicServiceException {
  61             return MusicDataStoreHandle.getDSHandle();
  62         }
  63     }