gerrit.onap Code Review - ccsdk/features.git/blob

   1 /*
   2  * ============LICENSE_START=======================================================
   3  * ONAP : ccsdk features
   4  * ================================================================================
   5  * Copyright (C) 2020 highstreet technologies GmbH Intellectual Property.
   6  * All rights reserved.
   7  * ================================================================================
   8  * Licensed under the Apache License, Version 2.0 (the "License");
   9  * you may not use this file except in compliance with the License.
  10  * You may obtain a copy of the License at
  11  *
  12  *     http://www.apache.org/licenses/LICENSE-2.0
  13  *
  14  * Unless required by applicable law or agreed to in writing, software
  15  * distributed under the License is distributed on an "AS IS" BASIS,
  16  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  17  * See the License for the specific language governing permissions and
  18  * limitations under the License.
  19  * ============LICENSE_END=========================================================
  20  *
  21  */
  22 package org.onap.ccsdk.features.sdnr.wt.dataprovider.database.elasticsearch.impl;
  23
  24 import org.json.JSONObject;
  25 import org.onap.ccsdk.features.sdnr.wt.common.database.HtDatabaseClient;
  26 import org.onap.ccsdk.features.sdnr.wt.common.database.SearchHit;
  27 import org.onap.ccsdk.features.sdnr.wt.common.database.SearchResult;
  28 import org.onap.ccsdk.features.sdnr.wt.common.database.queries.QueryBuilders;
  29 import org.onap.ccsdk.features.sdnr.wt.dataprovider.database.sqldb.data.HtUserdataManagerBase;
  30 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.data.provider.rev201110.Entity;
  31
  32 public class HtUserdataManagerImpl extends HtUserdataManagerBase {
  33
  34     private final HtDatabaseClient dbClient;
  35
  36     public HtUserdataManagerImpl(HtDatabaseClient rawClient) {
  37         this.dbClient = rawClient;
  38     }
  39
  40     @Override
  41     protected String readUserdata(String username, String defaultValue) {
  42
  43         SearchResult<SearchHit> result = this.dbClient.doReadByQueryJsonData(Entity.Userdata.getName(),
  44                 QueryBuilders.matchQuery("_id", username));
  45         return result.getHits().size() > 0 ? result.getHits().get(0).getSourceAsString() : defaultValue;
  46     }
  47
  48     @Override
  49     public boolean setUserdata(String username, String data) {
  50         JSONObject o = new JSONObject(this.getUserdata(username));
  51         JSONObject merge = mergeData(new JSONObject(data), o);
  52         return this.dbClient.doWriteRaw(Entity.Userdata.getName(), username, merge.toString()) != null;
  53     }
  54
  55     @Override
  56     public boolean removeUserdata(String username) {
  57         return this.dbClient.doRemove(Entity.Userdata.getName(), username);
  58     }
  59
  60 }