21e2694a90cfdbd57081518b590ab0effd965197
[ccsdk/features.git] /
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.onap.ccsdk.features.sdnr.wt.common.database.HtDatabaseClient;
25 import org.onap.ccsdk.features.sdnr.wt.common.database.SearchHit;
26 import org.onap.ccsdk.features.sdnr.wt.common.database.SearchResult;
27 import org.onap.ccsdk.features.sdnr.wt.common.database.queries.QueryBuilders;
28 import org.onap.ccsdk.features.sdnr.wt.dataprovider.database.sqldb.data.HtUserdataManagerBase;
29 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.data.provider.rev201110.Entity;
30
31 public class HtUserdataManagerImpl extends HtUserdataManagerBase {
32
33     private final HtDatabaseClient dbClient;
34
35     public HtUserdataManagerImpl(HtDatabaseClient rawClient) {
36         this.dbClient = rawClient;
37     }
38
39     @Override
40     protected String readUserdata(String username, String defaultValue) {
41
42         SearchResult<SearchHit> result = this.dbClient.doReadByQueryJsonData(Entity.Userdata.getName(),
43                 QueryBuilders.matchQuery("_id", username));
44         return result.getHits().size() > 0 ? result.getHits().get(0).getSourceAsString() : defaultValue;
45     }
46
47     @Override
48     public boolean setUserdata(String username, String data) {
49         return this.dbClient.doWriteRaw(Entity.Userdata.getName(), username, data) != null;
50     }
51
52     @Override
53     public boolean removeUserdata(String username) {
54         return this.dbClient.doRemove(Entity.Userdata.getName(), username);
55     }
56
57 }