src/json/json_elements.cc - master - Neitrinok/seastar

src/json/json_elements.cc
200 строк5 KB
Kefu Chai
src: include used headers for C++ modules build
10 янв 2024, 17:15
10 янв 2024, 17:15ac738b6
/*
 * This file is open source software, licensed to you under the terms
 * of the Apache License, Version 2.0 (the "License").  See the NOTICE file
 * distributed with this work for additional information regarding copyright
 * ownership.  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.
 */
/*
 * Copyright 2015 Cloudius Systems
 */

#ifdef SEASTAR_MODULE
module;
#endif

#include <string.h>
#include <string>
#include <vector>
#include <sstream>
#include <fmt/core.h>

#ifdef SEASTAR_MODULE
module seastar;
#else
#include <seastar/core/loop.hh>
#include <seastar/core/print.hh>
#include <seastar/json/json_elements.hh>
#endif

namespace seastar {

using namespace std;

namespace json {


class json_stream_builder;
/**
 * The json builder is a helper class
 * To help create a json object
 *
 */
class json_builder {
public:
    json_builder()
            : first(true) {
        result << OPEN;
    }

    /**
     * add a name value to an object
     * @param name the name of the element
     * @param str the value already formated
     */
    void add(const string& name, const string& str) {
        if (first) {
            first = false;
        } else {
            result << ", ";
        }
        result << '"' << name << "\": " << str;
    }

    /**
     * add a json element to the an object
     * @param element
     */
    void add(json_base_element* element) {
        if (element == nullptr || element->_set == false) {
            return;
        }
        try {
            add(element->_name, element->to_string());
        } catch (...) {
            std::throw_with_nested(std::runtime_error(fmt::format("Json generation failed for field: {}",element->_name)));
        }
    }

    /**
     * Get the string representation of the object
     * @return a string of accumulative object
     */
    string as_json() {
        result << CLOSE;
        return result.str();
    }

private:
    static const string OPEN;
    static const string CLOSE;
    friend class json_stream_builder;
    stringstream result;
    bool first;

};

/**
 * The json builder is a helper class
 * To help create a json object
 *
 */
class json_stream_builder {
public:
    json_stream_builder(output_stream<char>& s)
            : first(true), open(false), _s(s) {
    }

    /**
     * add a name value to an object
     * @param name the name of the element
     * @param str the value already formated
     */
    future<> add(const string& name, const json_base_element& element) {
        if (!open) {
            open = true;
            return _s.write(json_builder::OPEN).then([this, &name, &element] {
                return add(name, element);
            });
        }
        return _s.write(((first)? '"' + name : ",\"" + name) + "\":").then([this, &element] {
            first = false;
            return element.write(_s);
        });
    }

    /**
     * add a json element to the an object
     * @param element
     */
    future<> add(json_base_element* element) {
        if (element == nullptr || element->_set == false) {
            return make_ready_future<>();
        }
        return add(element->_name, *element);
    }

    /**
     * Get the string representation of the object
     * @return a string of accumulative object
     */
    future<> done() {
        return _s.write(json_builder::CLOSE);
    }

private:

    bool first;
    bool open;
    output_stream<char>& _s;
};

const string json_builder::OPEN("{");
const string json_builder::CLOSE("}");

void json_base::add(json_base_element* element, string name, bool mandatory) {
    element->_mandatory = mandatory;
    element->_name = name;
    _elements.push_back(element);
}

string json_base::to_json() const {
    json_builder res;
    for (auto i : _elements) {
        res.add(i);
    }
    return res.as_json();
}

future<> json_base::write(output_stream<char>& s) const {
    return do_with(json_stream_builder(s), [this] (json_stream_builder& builder) {
        return do_for_each(_elements, [&builder] (auto m) {
            return builder.add(m);
        }).then([&builder] {
            return builder.done();
        });
    });
}

bool json_base::is_verify() const {
    for (auto i : _elements) {
        if (!i->is_verify()) {
            return false;
        }
    }
    return true;
}

}

}