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
|
#include <cerrno>
#include <cstring>
#include <iostream>
#include "test_helpers.hxx"
using namespace PGSTD;
using namespace pqxx;
// Example program for libpqxx. Test waiting for notification with timeout.
namespace
{
// Sample implementation of notification receiver.
class TestListener : public notification_receiver
{
bool m_done;
public:
explicit TestListener(connection_base &C, string Name) :
notification_receiver(C, Name), m_done(false)
{
}
virtual void operator()(const string &, int be_pid)
{
m_done = true;
PQXX_CHECK_EQUAL(
be_pid,
conn().backendpid(),
"Notification came from wrong backend.");
cout << "Received notification: " << channel() << " pid=" << be_pid << endl;
}
bool done() const { return m_done; }
};
// A transactor to trigger our notification listener
class Notify : public transactor<>
{
string m_notif;
public:
explicit Notify(string NotifName) :
transactor<>("Notifier"), m_notif(NotifName) { }
void operator()(argument_type &T)
{
T.exec("NOTIFY " + m_notif);
}
void on_abort(const char Reason[]) throw ()
{
try
{
cerr << "Notify failed!" << endl;
if (Reason) cerr << "Reason: " << Reason << endl;
}
catch (const exception &)
{
}
}
};
void test_079(transaction_base &orgT)
{
connection_base &C(orgT.conn());
orgT.abort();
const string NotifName = "mylistener";
cout << "Adding listener..." << endl;
TestListener L(C, NotifName);
// First see if the timeout really works: we're not expecting any notifs
int notifs = C.await_notification(0, 1);
PQXX_CHECK_EQUAL(notifs, 0, "Got unexpected notification.");
cout << "Sending notification..." << endl;
C.perform(Notify(L.channel()));
for (int i=0; (i < 20) && !L.done(); ++i)
{
PQXX_CHECK_EQUAL(notifs, 0, "Got notifications, but no handler called.");
cout << ".";
notifs = C.await_notification(1,0);
}
cout << endl;
PQXX_CHECK(L.done(), "No notifications received.");
PQXX_CHECK_EQUAL(notifs, 1, "Got unexpected notifications.");
}
} // namespace
PQXX_REGISTER_TEST_T(test_079, nontransaction)
|
