RIC-E2-TERMINATION/mapWrapper.h

   1 /*
   2  * Copyright 2019 AT&T Intellectual Property
   3  * Copyright 2019 Nokia
   4  *
   5  * Licensed under the Apache License, Version 2.0 (the "License");
   6  * you may not use this file except in compliance with the License.
   7  * You may obtain a copy of the License at
   8  *
   9  *      http://www.apache.org/licenses/LICENSE-2.0
  10  *
  11  * Unless required by applicable law or agreed to in writing, software
  12  * distributed under the License is distributed on an "AS IS" BASIS,
  13  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  14  * See the License for the specific language governing permissions and
  15  * limitations under the License.
  16  */
  17
  18 /*
  19  * This source code is part of the near-RT RIC (RAN Intelligent Controller)
  20  * platform project (RICP).
  21  */
  22
  23
  24 //
  25 // Created by adi ENZEL on 8/28/19.
  26 //
  27
  28 #ifndef E2_MAPWRAPPER_H
  29 #define E2_MAPWRAPPER_H
  30
  31 #include <unordered_map>
  32 #include <mutex>
  33 #include <shared_mutex>
  34 #include <thread>
  35 #include <string>
  36 #include <iostream>
  37
  38 using namespace std;
  39
  40 class mapWrapper {
  41 public:
  42     void *find(char *key) {
  43         shared_lock<shared_timed_mutex> read(fence);
  44         auto entry = keyMap.find(key);
  45         if (entry == keyMap.end()) {
  46             return nullptr;
  47         }
  48         return entry->second;
  49     }
  50
  51     void setkey(char *key, void *val) {
  52         unique_lock<shared_timed_mutex> write(fence);
  53         keyMap[key] = val;
  54     }
  55
  56     void *erase(char *key) {
  57         unique_lock<shared_timed_mutex> write(fence);
  58         return (void *)keyMap.erase(key);
  59     }
  60
  61     void clear() {
  62         unique_lock<shared_timed_mutex> write(fence);
  63         keyMap.clear();
  64     }
  65
  66     void getKeys(vector<char *> &v) {
  67         shared_lock<shared_timed_mutex> read(fence);
  68         for (auto const &e : keyMap) {
  69             v.emplace_back((char *)e.first.c_str());
  70         }
  71     }
  72
  73
  74
  75 private:
  76     std::unordered_map<string, void *> keyMap;
  77     shared_timed_mutex fence;
  78
  79 };
  80 #endif //E2_MAPWRAPPER_H