Line data    Source code

       1              : /*
       2              :  * If not stated otherwise in this file or this component's LICENSE file the
       3              :  * following copyright and licenses apply:
       4              :  *
       5              :  * Copyright 2022 Sky UK
       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              : 
      20              : #ifndef FIREBOLT_RIALTO_CLIENT_KEY_IDMAP_H_
      21              : #define FIREBOLT_RIALTO_CLIENT_KEY_IDMAP_H_
      22              : 
      23              : #include <cstdint>
      24              : #include <map>
      25              : #include <mutex>
      26              : #include <vector>
      27              : 
      28              : namespace firebolt::rialto::client
      29              : {
      30              : class KeyIdMap
      31              : {
      32              : public:
      33              :     static KeyIdMap &instance();
      34              :     void addSession(std::int32_t keySessionId);
      35              :     bool updateKey(std::int32_t keySessionId, const std::vector<std::uint8_t> &keyId);
      36              :     std::vector<std::uint8_t> get(std::int32_t keySessionId) const;
      37              :     void erase(std::int32_t keySessionId);
      38              : 
      39              : private:
      40            1 :     KeyIdMap() = default;
      41            1 :     ~KeyIdMap() = default;
      42              :     KeyIdMap(const KeyIdMap &) = delete;
      43              :     KeyIdMap(KeyIdMap &&) = delete;
      44              :     KeyIdMap &operator=(const KeyIdMap &) = delete;
      45              :     KeyIdMap &operator=(KeyIdMap &&) = delete;
      46              : 
      47              : private:
      48              :     mutable std::mutex m_mutex;
      49              :     std::map<std::int32_t, std::vector<std::uint8_t>> m_keyIdMap;
      50              : };
      51              : } // namespace firebolt::rialto::client
      52              : 
      53              : #endif // FIREBOLT_RIALTO_CLIENT_KEY_IDMAP_H_