]> git.netwichtig.de Git - user/henk/code/inspircd.git/blob - src/modules/extra/m_testclient.cpp
Fix test client error cheecking on result types
[user/henk/code/inspircd.git] / src / modules / extra / m_testclient.cpp
1 #include <string>
2
3 #include "users.h"
4 #include "channels.h"
5 #include "modules.h"
6 #include "helperfuncs.h"
7 #include "inspircd.h"
8 #include "configreader.h"
9 #include "m_sqlv2.h"
10
11 class ModuleTestClient : public Module
12 {
13 private:
14         Server* Srv;
15
16 public:
17         ModuleTestClient(Server* Me)
18         : Module::Module(Me), Srv(Me)
19         {
20         }
21
22         void Implements(char* List)
23         {
24                 List[I_OnRequest] = List[I_OnBackgroundTimer] = 1;
25         }
26                 
27         virtual Version GetVersion()
28         {
29                 return Version(1, 0, 0, 0, VF_VENDOR);
30         }
31         
32         virtual void OnBackgroundTimer(time_t foo)
33         {
34                 Module* target = Srv->FindFeature("SQL");
35                 
36                 if(target)
37                 {
38                         SQLrequest foo = SQLreq(this, target, "foo", "SELECT foo, bar FROM ?", "rawr");
39                         
40                         if(foo.Send())
41                         {
42                                 log(DEBUG, "Sent query, got given ID %lu", foo.id);
43                         }
44                         else
45                         {
46                                 log(DEBUG, "SQLrequest failed: %s", foo.error.Str());
47                         }
48                 }
49         }
50         
51         virtual char* OnRequest(Request* request)
52         {
53                 if(strcmp(SQLRESID, request->GetData()) == 0)
54                 {
55                         log(DEBUG, "Got SQL result (%s)", request->GetData());
56                 
57                         SQLresult* res = (SQLresult*)request;
58
59                         if (res->error.Id() != NO_ERROR)
60                         {
61                                 log(DEBUG, "Got result with %d rows and %d columns", res->Rows(), res->Cols());
62
63                                 for (int r = 0; r < res->Rows(); r++)
64                                 {
65                                                 log(DEBUG, "Row %d:", r);
66                                         
67                                         for(int i = 0; i < res->Cols(); i++)
68                                         {
69                                                 log(DEBUG, "\t[%s]: %s", res->ColName(i).c_str(), res->GetValue(r, i).d.c_str());
70                                         }
71                                 }
72                         }
73                         else
74                         {
75                                 log(DEBUG, "SQLrequest failed: %s", res->error.Str());
76                                 
77                         }
78                 
79                         return SQLSUCCESS;
80                 }
81                 
82                 log(DEBUG, "Got unsupported API version string: %s", request->GetData());
83                 
84                 return NULL;
85         }
86         
87         virtual ~ModuleTestClient()
88         {
89         }       
90 };
91
92 class ModuleTestClientFactory : public ModuleFactory
93 {
94  public:
95         ModuleTestClientFactory()
96         {
97         }
98         
99         ~ModuleTestClientFactory()
100         {
101         }
102         
103         virtual Module * CreateModule(Server* Me)
104         {
105                 return new ModuleTestClient(Me);
106         }
107 };
108
109
110 extern "C" void * init_module( void )
111 {
112         return new ModuleTestClientFactory;
113 }