#include "stdafx.h"
#include "CSubscribeManager2.h"


CSubscribeManager2::CSubscribeManager2()
{
}


CSubscribeManager2::~CSubscribeManager2()
{
}

// 订阅，返回需要API订阅的合约
set<string> CSubscribeManager2::Subscribe(string sym, set<string> ss)
{
	auto it = m_maps.find(sym);
	if (it == m_maps.end())
	{
		m_maps.insert(pair<string, set<string>>(sym, ss));
	}
	else
	{
		m_maps[sym] = ss;
	}
	return ss;
}

// 取消订阅，返回需要API取消的合约
set<string> CSubscribeManager2::Unsubscribe(string sym)
{
	m_maps.erase(sym);
	return GetChildenInstruments();
}

set<string> CSubscribeManager2::GetChildenInstruments()
{
	lock_guard<mutex> cl(m_cs);

	set<string> ss;
	for (auto i = m_maps.begin(); i != m_maps.end(); ++i)
	{
		for (auto j = i->second.begin(); j != i->second.end(); ++j)
		{
			ss.insert(j->data());
		}
	}
	return ss;
}

set<string> CSubscribeManager2::GetParentInstruments()
{
	lock_guard<mutex> cl(m_cs);

	set<string> ss;
	for (auto i = m_maps.begin(); i != m_maps.end(); ++i)
	{
		ss.insert(i->first);
	}
	return ss;
}

void CSubscribeManager2::Clear()
{
	lock_guard<mutex> cl(m_cs);
	m_maps.clear();
}