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.\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.\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.\n" << reset;
}
Revision:	37
Committed:	2004-07-22T16:00:52-07:00 (20 years, 11 months ago) by douglas
File size:	1878 byte(s)
Log Message:	Changed colored output to be nicer, maybe.
#	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	37	cerr << bright << green << "Started publisher daemon.\n" << reset;
20	douglas	15
21	douglas	37	while (true)
22	douglas	31	{
23	douglas	37	Stamp now, next(now + Hour(1));
24			api::Thread thread(etl::Bind<0>(&Publisher::update, this));
25	douglas	31
26	douglas	37	next.setSeconds(0);
27			next.setMinutes(0);
28	douglas	31
29	douglas	37	if (Zoe::debug) cerr << "now = " << now << "\nnext = " << next << '\n';
30	douglas	35
31	douglas	37	while (now < next - Minute(1))
32	douglas	31	{
33	douglas	37	sleep(Minute(1));
34	douglas	31
35	douglas	37	now = Stamp();
36	douglas	31	}
37	douglas	37
38			while (now < next)
39			{
40			sleep(1);
41
42			now = Stamp();
43			}
44
45			thread.Join();
46	douglas	31	}
47	douglas	27	}
48	douglas	25
49	douglas	37	int Publisher::update()
50	douglas	27	{
51	douglas	37	cerr << bright << yellow << "Started updating feeds.\n" << reset;
52	douglas	31
53	douglas	27	Stamp stamp;
54	douglas	37	ext::Handle<dbi::Connection> db(dbi::Connect(database.driver, database.host,
55			database.user, database.password, database.db));
56	douglas	25
57	douglas	34	db->Execute("DELETE FROM messages WHERE stamp<'" + ext::String(stamp - Day(30)) + "'");
58	douglas	27
59	douglas	37	for (std::set<Buddy>::const_iterator buddy(buddies.begin());
60			buddy != buddies.end(); ++buddy)
61	douglas	27	{
62	douglas	34	ext::Handle<dbi::ResultSet> messages(db->Execute("SELECT stamp, message FROM messages WHERE id='"
63			+ lexical_cast<ext::String>(buddy->getId()) + "'"));
64	douglas	27	std::vector<AwayMessage> messages_;
65
66	douglas	34	while (messages->MoveNext()) messages_.push_back(AwayMessage(messages->GetString("message"),
67	douglas	27	messages->GetString("stamp")));
68
69			Rss rss(*buddy, messages_, stamp);
70	douglas	35	Atom atom(*buddy, messages_, stamp);
71	douglas	37
72			cerr << bright << yellow << "Updated feeds for " << blue << *buddy
73			<< yellow << ".\n" << reset;
74	douglas	27	}
75	douglas	37
76			cerr << bright << yellow << "Finished updating feeds.\n" << reset;
77	douglas	15	}