Add a control to read the PNN number of the local node to libctdb
[sahlberg/ctdb.git] / libctdb / tst.c
1 #include <stdio.h>
2 #include <stdint.h>
3 #include <stdlib.h>
4 #include <poll.h>
5 #include <fcntl.h>
6 #include <string.h>
7 #include "lib/tdb/include/tdb.h"
8 #include "include/ctdb.h"
9
10 void msg_h(struct ctdb_context *ctdb, uint64_t srvid, TDB_DATA data, void *private_data)
11 {
12         printf("Message received on port %d : %s\n", (int)srvid, data.dptr);
13 }
14
15
16 void pnn_cb(int32_t status, int32_t pnn, void *private_data)
17 {
18         printf("status:%d pnn:%d\n", status, pnn);
19 }
20
21 void rm_cb(int32_t status, int32_t recmaster, void *private_data)
22 {
23         printf("status:%d recmaster:%d\n", status, recmaster);
24 }
25
26 int main(int argc, char *argv[])
27 {
28         struct ctdb_context *ctdb_context;
29         ctdb_handle *handle;
30         struct pollfd pfd;
31         int ret;
32         TDB_DATA msg;
33
34         ctdb_context = ctdb_connect("/tmp/ctdb.socket");
35
36
37         handle = ctdb_set_message_handler_send(ctdb_context, 55, NULL, msg_h, NULL);
38         if (handle == NULL) {
39                 printf("Failed to register message port\n");
40                 exit(10);
41         }
42         ret = ctdb_set_message_handler_recv(ctdb_context, handle);
43         if (ret != 0) {
44                 printf("Failed to receive set_message_handler reply\n");
45                 exit(10);
46         }
47
48         msg.dptr="HelloWorld";
49         msg.dsize = strlen(msg.dptr);
50
51         ret = ctdb_send_message(ctdb_context, 0, 55, msg);
52         if (ret != 0) {
53                 printf("Failed to send message. Aborting\n");
54                 exit(10);
55         }
56
57         handle = ctdb_getpnn_send(ctdb_context, pnn_cb, NULL);
58         if (handle == NULL) {
59                 printf("Failed to send get_pnn control\n");
60                 exit(10);
61         }
62
63         handle = ctdb_getrecmaster_send(ctdb_context, 0, rm_cb, NULL);
64         if (handle == NULL) {
65                 printf("Failed to send get_recmaster control\n");
66                 exit(10);
67         }
68
69
70         pfd.fd = ctdb_get_fd(ctdb_context);
71         for (;;) {
72           pfd.events = ctdb_which_events(ctdb_context);
73           if (poll(&pfd, 1, -1) < 0) {
74             printf("Poll failed");
75             exit(10);
76           }
77           ctdb_service(ctdb_context);
78         }
79
80         return 0;
81 }