external/vulkancts/vkscserver/vksStore.hpp

#ifndef _VKSSTORE_HPP
#define _VKSSTORE_HPP

/*-------------------------------------------------------------------------
 * Vulkan CTS Framework
 * --------------------
 *
 * Copyright (c) 2021 The Khronos Group Inc.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 *
 *-------------------------------------------------------------------------*/

#include "vksCommon.hpp"

#include <mutex>
#include <map>

namespace vksc_server
{

struct Store
{
    bool Get(const string &path, vector<u8> &content, bool removeAfter)
    {
        std::lock_guard<std::mutex> lock(FileMapMutex);

        auto it = FileMap.find(path);
        if (it != FileMap.end())
        {
            if (removeAfter)
            {
                content = std::move(it->second);
                FileMap.erase(it);
            }
            else
            {
                content = it->second;
            }
            return true;
        }
        return false;
    }

    bool Set(const string &uniqueFilename, const vector<u8> &content)
    {
        std::lock_guard<std::mutex> lock(FileMapMutex);
        FileMap[uniqueFilename] = std::move(content);
        return true;
    }

private:
    std::map<string, vector<u8>> FileMap;
    std::mutex FileMapMutex;
};

} // namespace vksc_server

#endif // _VKSSTORE_HPP