trace_context.h

/* -*- Mode: C++; tab-width: 4; c-basic-offset: 4; indent-tabs-mode: nil -*- */
/*
 *     Copyright 2016 Couchbase, 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.
 */

#pragma once

#include <memory>
#include <string>
#include <unordered_map>

#include "phosphor/platform/core.h"

namespace phosphor {

    // Forward declare
    class TraceBuffer;

    class PHOSPHOR_API TraceContext{
    public:
        using ThreadNamesMap = std::unordered_map<uint64_t, std::string>;

        ~TraceContext();

        TraceContext(std::unique_ptr<TraceBuffer>&& buffer);

        TraceContext(std::unique_ptr<TraceBuffer>&& buffer,
                     const ThreadNamesMap& _thread_names);

        TraceContext(TraceContext&& other);

        TraceContext& operator=(TraceContext&& other);

        const TraceBuffer* getBuffer() const {
            return trace_buffer.get();
        }

        TraceBuffer* getBuffer() {
            return trace_buffer.get();
        }

        const ThreadNamesMap& getThreadNames() const {
            return thread_names;
        }

    protected:

        void addThreadName(uint64_t id, const std::string& name);

    private:
        std::unique_ptr<TraceBuffer> trace_buffer;

        ThreadNamesMap thread_names;
    };

}