zoe/trunk/Publisher.cpp

// Zoe AIM Away Message RSS Feed Generator
//
// Seth King and Douglas Thrift
//
// $Id$

#include "Publisher.hpp"
#include "Rss.hpp"
#include "Atom.hpp"

Publisher::Publisher(const std::set<Buddy>& buddies, const Database& database, bool start) : buddies(buddies),
        database(database), start(start)
{
        if (start) publisher.Spawn(etl::Bind<0>(&Publisher::publish, this));
}

int Publisher::publish()
{
        cerr << bright << green << "Started publisher daemon at " << Stamp()
                << ".\n" << reset;

        while (true)
        {
                Stamp now, next(now + Hour(1));
                api::Thread thread(etl::Bind<0>(&Publisher::update, this));

                next.setSeconds(0);
                next.setMinutes(0);

                if (Zoe::debug) cerr << "now = " << now << "\nnext = " << next << '\n';

                while (now < next - Minute(1))
                {
                        sleep(Minute(1));

                        now = Stamp();
                }

                while (now < next)
                {
                        sleep(1);

                        now = Stamp();
                }

                thread.Join();
        }
}

int Publisher::update()
{
        cerr << bright << yellow << "Started updating feeds at " << Stamp()
                << ".\n" << reset;

        Stamp stamp;
        ext::Handle<dbi::Connection> db(dbi::Connect(database.driver, database.host,
                database.user, database.password, database.db));

        db->Execute("DELETE FROM messages WHERE stamp<'" + ext::String(stamp - Day(30)) + "'");

        for (std::set<Buddy>::const_iterator buddy(buddies.begin());
                buddy != buddies.end(); ++buddy)
        {
                ext::Handle<dbi::ResultSet> messages(db->Execute("SELECT stamp, message FROM messages WHERE id='"
                        + lexical_cast<ext::String>(buddy->getId()) + "'"));
                std::vector<AwayMessage> messages_;

                while (messages->MoveNext()) messages_.push_back(AwayMessage(messages->GetString("message"),
                        messages->GetString("stamp")));

                Rss rss(*buddy, messages_, stamp);
                Atom atom(*buddy, messages_, stamp);

                cerr << bright << yellow << "Updated feeds for " << blue << *buddy
                        << yellow << ".\n" << reset;
        }

        cerr << bright << yellow << "Finished updating feeds at " << Stamp()
                << ".\n" << reset;
}
Revision:	38
Committed:	2004-07-22T22:42:40-07:00 (20 years, 11 months ago) by douglas
File size:	1947 byte(s)
Log Message:	Yeah, did stuff.
#	User	Rev	Content
1	douglas	13	// Zoe AIM Away Message RSS Feed Generator
2			//
3			// Seth King and Douglas Thrift
4			//
5			// $Id$
6
7			#include "Publisher.hpp"
8	douglas	25	#include "Rss.hpp"
9	douglas	35	#include "Atom.hpp"
10	douglas	13
11	douglas	34	Publisher::Publisher(const std::set<Buddy>& buddies, const Database& database, bool start) : buddies(buddies),
12			database(database), start(start)
13	douglas	13	{
14	douglas	34	if (start) publisher.Spawn(etl::Bind<0>(&Publisher::publish, this));
15	douglas	13	}
16	douglas	15
17	douglas	21	int Publisher::publish()
18	douglas	15	{
19	douglas	38	cerr << bright << green << "Started publisher daemon at " << Stamp()
20			<< ".\n" << reset;
21	douglas	15
22	douglas	37	while (true)
23	douglas	31	{
24	douglas	37	Stamp now, next(now + Hour(1));
25			api::Thread thread(etl::Bind<0>(&Publisher::update, this));
26	douglas	31
27	douglas	37	next.setSeconds(0);
28			next.setMinutes(0);
29	douglas	31
30	douglas	37	if (Zoe::debug) cerr << "now = " << now << "\nnext = " << next << '\n';
31	douglas	35
32	douglas	37	while (now < next - Minute(1))
33	douglas	31	{
34	douglas	37	sleep(Minute(1));
35	douglas	31
36	douglas	37	now = Stamp();
37	douglas	31	}
38	douglas	37
39			while (now < next)
40			{
41			sleep(1);
42
43			now = Stamp();
44			}
45
46			thread.Join();
47	douglas	31	}
48	douglas	27	}
49	douglas	25
50	douglas	37	int Publisher::update()
51	douglas	27	{
52	douglas	38	cerr << bright << yellow << "Started updating feeds at " << Stamp()
53			<< ".\n" << reset;
54	douglas	31
55	douglas	27	Stamp stamp;
56	douglas	37	ext::Handle<dbi::Connection> db(dbi::Connect(database.driver, database.host,
57			database.user, database.password, database.db));
58	douglas	25
59	douglas	34	db->Execute("DELETE FROM messages WHERE stamp<'" + ext::String(stamp - Day(30)) + "'");
60	douglas	27
61	douglas	37	for (std::set<Buddy>::const_iterator buddy(buddies.begin());
62			buddy != buddies.end(); ++buddy)
63	douglas	27	{
64	douglas	34	ext::Handle<dbi::ResultSet> messages(db->Execute("SELECT stamp, message FROM messages WHERE id='"
65			+ lexical_cast<ext::String>(buddy->getId()) + "'"));
66	douglas	27	std::vector<AwayMessage> messages_;
67
68	douglas	34	while (messages->MoveNext()) messages_.push_back(AwayMessage(messages->GetString("message"),
69	douglas	27	messages->GetString("stamp")));
70
71			Rss rss(*buddy, messages_, stamp);
72	douglas	35	Atom atom(*buddy, messages_, stamp);
73	douglas	37
74			cerr << bright << yellow << "Updated feeds for " << blue << *buddy
75			<< yellow << ".\n" << reset;
76	douglas	27	}
77	douglas	37
78	douglas	38	cerr << bright << yellow << "Finished updating feeds at " << Stamp()
79			<< ".\n" << reset;
80	douglas	15	}