export.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 "phosphor/platform/core.h"
#include "phosphor/trace_buffer.h"
#include "phosphor/trace_context.h"
#include "phosphor/trace_log.h"

namespace phosphor {
    namespace tools {

        class PHOSPHOR_API JSONExport {
        public:
            JSONExport(const TraceContext& _context);

            ~JSONExport();

            size_t read(char* out, size_t length);

            StringPtr read(size_t length);

            StringPtr read();

            bool done();

        protected:
            enum class State {
                opening,
                /* first_event is only used if first_thread wasn't */
                first_event,
                other_events,
                first_thread,
                other_threads,
                footer,
                dead
            };

            const TraceContext& context;
            TraceBuffer::event_iterator it;
            std::unordered_map<uint64_t, std::string>::const_iterator tit;

            State state = State::opening;
            std::string cache;
        };

        class PHOSPHOR_API FileStopCallback : public TracingStoppedCallback {
        public:
            FileStopCallback(
                const std::string& _file_path = "phosphor.%p.json");

            ~FileStopCallback();

            void operator()(TraceLog& log, std::lock_guard<TraceLog>& lh) override;

        protected:
            /*
             * Exposed for testing
             */
            StringPtr generateFilePathAsPtr();

        private:
            std::string file_path;
            std::string generateFilePath();
        };
    }
}