summaryrefslogtreecommitdiff
path: root/p2pvr/lib/maintenance/network.cpp
blob: 6b0ebc6b5f44187c4f8b1aea3b68fa81115e96ce (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
#include "../maintenance.h"
#include "../siParsers/network.h"
#include <sqlMergeTask.h>
#include <Ice/Communicator.h>
#include "../p2Helpers.h"
#include "../dvbsiHelpers.h"
#include "../containerIterator.h"
#include "../singleIterator.h"
#include "../temporaryIceAdapterObject.h"

class SiNetworkInformationMerger : public SiNetworkInformationParser {
	public:
		SiNetworkInformationMerger(CommonObjects * co) : commonObjects(co) { }

		void HandleTable(DVBSI::NetworkPtr n)
		{
			Logger()->messagebf(LOG_DEBUG, "Network Id: %d Name: %s", n->NetworkId, *n->Name);
			BOOST_FOREACH(const auto & ts, n->TransportStreams) {
				Logger()->messagebf(LOG_DEBUG, "\tTransport Stream Id: %d Original Network Id: %d", ts->TransportStreamId, ts->OriginalNetworkId);
				BOOST_FOREACH(const auto & s, ts->Services) {
					Logger()->messagebf(LOG_DEBUG, "\t\tService Id: %d Service Type: %d", s.ServiceId, s.ServiceType);
				}
				if (ts->Terrestrial) {
					Logger()->messagebf(LOG_DEBUG, "\t\tDVB-T: Frequency: %d", ts->Terrestrial->Frequency);
				}
			}

			SqlMergeTask mergeNetwork("postgres", "networks");
			CreateColumns<DVBSI::NetworkPtr>(boost::bind(&Maintenance::SqlMergeColumnsInserter, &mergeNetwork, _1, _2));
			std::vector<DVBSI::NetworkPtr> networks = { n };
			mergeNetwork.sources.insert(new ContainerIterator<std::vector<DVBSI::NetworkPtr>>(&networks));
			mergeNetwork.loadComplete(commonObjects);
			mergeNetwork.execute(NULL);

			SqlMergeTask mergeTransports("postgres", "transportstreams");
			CreateColumns<DVBSI::NetworkTransportStreamPtr>(boost::bind(&Maintenance::SqlMergeColumnsInserter, &mergeTransports, _1, _2));
			mergeTransports.sources.insert(new ContainerIterator<DVBSI::NetworkTransportStreams>(&n->TransportStreams, n));
			mergeTransports.loadComplete(commonObjects);
			mergeTransports.execute(NULL);

			SqlMergeTask mergeDvbt("postgres", "delivery_dvbt");
			CreateColumns<DVBSI::TerrestrialDeliveryPtr>(boost::bind(&Maintenance::SqlMergeColumnsInserter, &mergeDvbt, _1, _2));
			BOOST_FOREACH(const auto & s, n->TransportStreams) {
				if (s->Terrestrial) {
					mergeDvbt.sources.insert(new SingleIterator<DVBSI::TerrestrialDeliveryPtr>(&s->Terrestrial, s));
				}
			}
			mergeDvbt.loadComplete(commonObjects);
			mergeDvbt.execute(NULL);

			SqlMergeTask mergeServices("postgres", "services");
			CreateColumns<DVBSI::NetworkService>(boost::bind(&Maintenance::SqlMergeColumnsInserter, &mergeServices, _1, _2));
			BOOST_FOREACH(const auto & s, n->TransportStreams) {
				mergeServices.sources.insert(new ContainerIterator<DVBSI::NetworkServiceList>(&s->Services, s));
			}
			mergeServices.loadComplete(commonObjects);
			mergeServices.execute(NULL);
		}
	private:
		CommonObjects * commonObjects;
};

void
Maintenance::UpdateNetwork(short type, const Ice::Current & ice)
{
	auto ic = ice.adapter->getCommunicator();
	auto devs = P2PVR::DevicesPrx::checkedCast(ice.adapter->createProxy(ic->stringToIdentity("GlobalDevices")));
	auto si = P2PVR::SIPrx::checkedCast(ice.adapter->createProxy(ic->stringToIdentity("SI")));
	auto siparser = new SiNetworkInformationMerger(this);
	TemporarayIceAdapterObject<P2PVR::RawDataClient> parser(ice.adapter, siparser);

	if (!devs) {
		throw std::runtime_error("bad proxy(s)");
	}
	const auto transports = si->GetAllDeliveries(type);
	// Attempt to just download fresh data
	BOOST_FOREACH(const auto & transport, transports) {
		P2PVR::TunerPrx tuner;
		try {
			tuner = devs->GetTunerAny(type, transport, time(NULL) + 300);
		}
		catch (...) {
			Logger()->messagebf(LOG_WARNING, "%s: Failed to get a suitable tuner", __PRETTY_FUNCTION__);
			continue;
		}
		if (!tuner) {
			continue;
		}
		try {
			tuner->SendNetworkInformation(parser);
			devs->ReleaseTuner(tuner);
			BOOST_FOREACH(const CommonObjects::DataSources::value_type & ds, CommonObjects::datasources) {
				ds.second->commit();
			}
			return;
		}
		catch (const std::exception & ex) {
			Logger()->messagebf(LOG_WARNING, "%s: Failed to fetch network information: %s", __PRETTY_FUNCTION__, ex.what());
			devs->ReleaseTuner(tuner);
			throw;
		}
		catch (...) {
			Logger()->messagebf(LOG_WARNING, "%s: Failed to fetch network information", __PRETTY_FUNCTION__);
			devs->ReleaseTuner(tuner);
			throw;
		}
	}
	// If we can't do that, do a complete scan
	auto tuner = devs->GetPrivateTuner(type, time(NULL) + 300);
	tuner->ScanAndSendNetworkInformation(parser);
	devs->ReleaseTuner(tuner);
}