rapidjson组装map和数组array的代码示例

972 篇文章 329 订阅
166 篇文章 14 订阅

         直接上码:

 

#include <iostream>
#include <map>

// 请自己下载开源的rapidjson
#include "rapidjson/prettywriter.h"
#include "rapidjson/rapidjson.h"
#include "rapidjson/document.h"
#include "rapidjson/stringbuffer.h"
#include "rapidjson/writer.h"
#include "rapidjson/memorystream.h"
#include <sys/types.h>
#include <vector>

using namespace std;
using rapidjson::Document;
using rapidjson::StringBuffer;
using rapidjson::Writer;
using namespace rapidjson;


// 注意int和uint64_t
map<string, uint64_t> g_mChildInt;
map<string, string> g_mChildString;
string formJson(const map<string, int> &mInt, const map<string, string> &mString,
		        const string &strChild="", const map<string, uint64_t> &mChildInt=g_mChildInt, const map<string, string> &mChildString=g_mChildString,
		        const string &strChild2="", const map<string, uint64_t> &mChildInt2=g_mChildInt, const map<string, string> &mChildString2=g_mChildString)
{
	Document document;

    Document::AllocatorType& allocator = document.GetAllocator(); 
    Value root(kObjectType);

    Value key(kStringType);  
    Value value(kStringType); 

	// 当前级别
	for(map<string, int>::const_iterator it = mInt.begin(); it != mInt.end(); ++it) 
	{
		key.SetString(it->first.c_str(), allocator);  
    	root.AddMember(key, it->second, allocator);
	}

	for(map<string, string>::const_iterator it = mString.begin(); it != mString.end(); ++it)
	{
		key.SetString(it->first.c_str(), allocator);  
   		value.SetString(it->second.c_str(), allocator);  
    	root.AddMember(key, value, allocator);
	}

	// 孩子级别
	if(!strChild.empty())
	{
		Value child(kObjectType);
		for(map<string, uint64_t>::const_iterator it = mChildInt.begin(); it != mChildInt.end(); ++it) 
		{
			key.SetString(it->first.c_str(), allocator);  
	    	child.AddMember(key, it->second, allocator);
		}

		for(map<string, string>::const_iterator it = mChildString.begin(); it != mChildString.end(); ++it)
		{
			key.SetString(it->first.c_str(), allocator);  
	   		value.SetString(it->second.c_str(), allocator);  
	    	child.AddMember(key, value, allocator);
		}

		key.SetString(strChild.c_str(), allocator); 
		root.AddMember(key, child, allocator);
	}

	// 孩子级别
	if(!strChild2.empty())
	{
		Value child(kObjectType);
		for(map<string, uint64_t>::const_iterator it = mChildInt2.begin(); it != mChildInt2.end(); ++it) 
		{
			key.SetString(it->first.c_str(), allocator);  
	    	child.AddMember(key, it->second, allocator);
		}

		for(map<string, string>::const_iterator it = mChildString2.begin(); it != mChildString2.end(); ++it)
		{
			key.SetString(it->first.c_str(), allocator);  
	   		value.SetString(it->second.c_str(), allocator);  
	    	child.AddMember(key, value, allocator);
		}

		key.SetString(strChild2.c_str(), allocator); 
		root.AddMember(key, child, allocator);
	}

    StringBuffer buffer;  
    Writer<StringBuffer> writer(buffer);  
    root.Accept(writer);  
    return buffer.GetString();  

}


string formJsonWithArray(const map<string, int> &mInt, const map<string, string> &mString,
				const string &strChild1, const map<string, uint64_t> &mChildInt, const map<string, string> &mChildString,
		        string &strChild2, vector<map<string, uint64_t> >&mVecChildInt, vector<map<string, string> >&mVecChildString)
		        
{
	Document document;
	
    Document::AllocatorType& allocator = document.GetAllocator(); 
    Value root(kObjectType);

    Value key(kStringType);  
    Value value(kStringType); 

	// 当前级别
	for(map<string, int>::const_iterator it = mInt.begin(); it != mInt.end(); ++it) 
	{
		key.SetString(it->first.c_str(), allocator);  
    	root.AddMember(key, it->second, allocator);
	}

	for(map<string, string>::const_iterator it = mString.begin(); it != mString.end(); ++it)
	{
		key.SetString(it->first.c_str(), allocator);  
   		value.SetString(it->second.c_str(), allocator);  
    	root.AddMember(key, value, allocator);
	}

	// 孩子级别
	if(!strChild1.empty())
	{
		Value child(kObjectType);
		for(map<string, uint64_t>::const_iterator it = mChildInt.begin(); it != mChildInt.end(); ++it) 
		{
			key.SetString(it->first.c_str(), allocator);  
	    	child.AddMember(key, it->second, allocator);
		}

		for(map<string, string>::const_iterator it = mChildString.begin(); it != mChildString.end(); ++it)
		{
			key.SetString(it->first.c_str(), allocator);  
	   		value.SetString(it->second.c_str(), allocator);  
	    	child.AddMember(key, value, allocator);
		}

		key.SetString(strChild1.c_str(), allocator); 
		root.AddMember(key, child, allocator);
	}

	// 孩子级别
	unsigned int uiSize1 = mVecChildInt.size();
	unsigned int uiSize2 = mVecChildString.size();
	if(!strChild2.empty() && uiSize1 == uiSize2)
	{
		Value array(rapidjson::kArrayType);  
		for(unsigned int i = 0; i < uiSize1; ++i)
		{
			Value child(kObjectType);
			for(map<string, uint64_t>::iterator it = mVecChildInt[i].begin(); it != mVecChildInt[i].end(); ++it) 
			{
				key.SetString(it->first.c_str(), allocator);  
		    	child.AddMember(key, it->second, allocator);
	
			}
		
			for(map<string, string>::iterator it = mVecChildString[i].begin(); it != mVecChildString[i].end(); ++it)
			{
				key.SetString(it->first.c_str(), allocator);  
		   		value.SetString(it->second.c_str(), allocator);  
		    	child.AddMember(key, value, allocator);
			}

			array.PushBack(child, allocator);  

		}

		key.SetString(strChild2.c_str(), allocator); 
		root.AddMember(key, array, allocator);

	}

    StringBuffer buffer;  
    Writer<StringBuffer> writer(buffer);  
    root.Accept(writer);  
    return buffer.GetString();  
}


void test1()
{
	map<string, int> mInt;
	map<string, string> mString;
	mInt["code"] = 0;
	mString["msg"] = "ok";

	string strChild1 = "xxx";
	map<string, uint64_t> mChildInt1;
	map<string, string> mChildString1;
	mChildInt1["key"] = 729;
	mChildString1["kk"] = "vv";

	string strChild2 = "yyy";
	map<string, uint64_t> mChildInt2;
	map<string, string> mChildString2;
	mChildInt2["key"] = 730;
	mChildString2["kkk"] = "vvv";

	string s = formJson(mInt, mString, strChild1, mChildInt1, mChildString1,strChild2, mChildInt2, mChildString2);
	cout << s << endl;
}


void test2()
{
	map<string, int> mInt;
	map<string, string> mString;
	mInt["code"] = 0;
	mString["msg"] = "ok";

	string strChild1 = "xxx";
	map<string, uint64_t> mChildInt;
	map<string, string> mChildString;
	mChildString["kk"] = "vv";
	mChildInt["key"] = 729;


	string strChild2 = "data";
	vector<map<string, uint64_t> >mVecChildInt; 
	vector<map<string, string> >mVecChildString;

	{

		map<string, uint64_t> mChildInt; 
		map<string, string> mChildString;

		mChildInt["id"] = 1;
		mChildString["path"] = "pa";
		mChildString["sha"] = "sh";
			
		mVecChildInt.push_back(mChildInt);
		mVecChildString.push_back(mChildString);
	}

	{

		map<string, uint64_t> mChildInt; 
		map<string, string> mChildString;

		mChildInt["id"] = 2;
		mChildString["path"] = "pa";
		mChildString["sha"] = "sh";
			
		mVecChildInt.push_back(mChildInt);
		mVecChildString.push_back(mChildString);
	}

	string s = formJsonWithArray(mInt, mString, strChild1, mChildInt, mChildString, strChild2, mVecChildInt, mVecChildString);
	cout << s << endl;
}


int main(int argc, char *argv[])
{
	test1();
	test2();

	return 0;
}

         结果:

 

{"code":0,"msg":"ok","xxx":{"key":729,"kk":"vv"},"yyy":{"key":730,"kkk":"vvv"}}
{"code":0,"msg":"ok","xxx":{"key":729,"kk":"vv"},"data":[{"id":1,"path":"pa","sha":"sh"},{"id":2,"path":"pa","sha":"sh"}]}
 

 

        不多说, 睡觉。

 

评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值