src/yaz-my-server.cpp

   1 /*
   2  * Copyright (c) 1998-2001, Index Data.
   3  * See the file LICENSE for details.
   4  *
   5  * $Log: yaz-my-server.cpp,v $
   6  * Revision 1.7  2001-08-13 16:39:12  adam
   7  * PDU_Assoc keeps track of children. Using yaz_log instead of logf.
   8  *
   9  * Revision 1.6  2001/05/17 14:18:03  adam
  10  * New handler for old version item update for server:
  11  *  void update_service0 (Z_ExtendedServicesRequest *req,
  12  *                        Z_IU0Update *io, Z_ExtendedServicesResponse *res)
  13  *
  14  * Revision 1.5  2001/05/03 12:39:39  adam
  15  * Added Update server service.
  16  *
  17  * Revision 1.4  2001/04/05 13:09:44  adam
  18  * Removed ursula dependancy.
  19  *
  20  * Revision 1.3  2001/04/04 14:02:49  adam
  21  * URSULA / Z-ruth service.
  22  *
  23  * Revision 1.2  2001/03/29 15:14:26  adam
  24  * Minor updates.
  25  *
  26  * Revision 1.1  2001/03/27 14:47:45  adam
  27  * New server facility scheme.
  28  *
  29  * Revision 1.15  2001/03/26 14:43:49  adam
  30  * New threaded PDU association.
  31  *
  32  * Revision 1.14  2000/11/01 14:22:59  adam
  33  * Added fd parameter for method IYaz_PDU_Observer::clone.
  34  *
  35  * Revision 1.13  2000/10/11 11:58:16  adam
  36  * Moved header files to include/yaz++. Switched to libtool and automake.
  37  * Configure script creates yaz++-config script.
  38  *
  39  * Revision 1.12  2000/09/21 21:43:20  adam
  40  * Better high-level server API.
  41  *
  42  * Revision 1.11  2000/09/12 16:23:49  adam
  43  * Updated server example.
  44  *
  45  * Revision 1.10  2000/09/12 16:04:17  adam
  46  * Added comstack method for Yaz_PDU_Assoc..
  47  *
  48  * Revision 1.9  2000/09/12 12:09:53  adam
  49  * More work on high-level server.
  50  *
  51  * Revision 1.8  2000/09/08 10:23:42  adam
  52  * Added skeleton of yaz-z-server.
  53  *
  54  * Revision 1.7  1999/12/06 13:52:45  adam
  55  * Modified for new location of YAZ header files. Experimental threaded
  56  * operation.
  57  *
  58  * Revision 1.6  1999/04/21 12:09:01  adam
  59  * Many improvements. Modified to proxy server to work with "sessions"
  60  * based on cookies.
  61  *
  62  * Revision 1.5  1999/04/09 11:46:57  adam
  63  * Added object Yaz_Z_Assoc. Much more functional client.
  64  *
  65  * Revision 1.4  1999/03/23 14:17:57  adam
  66  * More work on timeout handling. Work on yaz-client.
  67  *
  68  * Revision 1.3  1999/02/02 14:01:22  adam
  69  * First WIN32 port of YAZ++.
  70  *
  71  * Revision 1.2  1999/01/28 13:08:47  adam
  72  * Yaz_PDU_Assoc better encapsulated. Memory leak fix in
  73  * yaz-socket-manager.cc.
  74  *
  75  * Revision 1.1.1.1  1999/01/28 09:41:07  adam
  76  * First implementation of YAZ++.
  77  *
  78  */
  79
  80 #include <yaz/log.h>
  81 #include <yaz/options.h>
  82 #include <yaz++/yaz-z-server.h>
  83 #include <yaz++/yaz-pdu-assoc.h>
  84 #include <yaz++/yaz-socket-manager.h>
  85
  86 class MyILL : public Yaz_Facility_ILL {
  87 public:
  88     void ill_service (Z_ExtendedServicesRequest *req,
  89                       Z_ItemOrder *io,
  90                       Z_ExtendedServicesResponse *res);
  91 };
  92
  93 class MyUpdate : public Yaz_Facility_Update {
  94 public:
  95     void update_service (Z_ExtendedServicesRequest *req,
  96                          Z_IUUpdate *io,
  97                          Z_ExtendedServicesResponse *res);
  98     void update_service0 (Z_ExtendedServicesRequest *req,
  99                          Z_IU0Update *io,
 100                          Z_ExtendedServicesResponse *res);
 101 };
 102
 103
 104 class MyRetrieval : public Yaz_Facility_Retrieval, Yaz_USMARC {
 105 public:
 106     int sr_init (Z_InitRequest *initRequest,
 107                  Z_InitResponse *initResponse);
 108     void sr_search (Z_SearchRequest *searchRequest,
 109                         Z_SearchResponse *searchResponse);
 110     void sr_present (Z_PresentRequest *presentRequest,
 111                          Z_PresentResponse *presentResponse);
 112     void sr_record (const char *resultSetName,
 113                     int position,
 114                     int *format,
 115                     Z_RecordComposition *comp,
 116                     Z_NamePlusRecord *namePlusRecord,
 117                     Z_Records *records);
 118 };
 119
 120 #if HAVE_YAZ_URSULA_H
 121 class MyUrsula : public Yaz_Facility_Ursula {
 122 public:
 123     void ursula_service (Z_ExtendedServicesRequest *req,
 124                          Z_UrsPDU *u,
 125                          Z_ExtendedServicesResponse *res);
 126 };
 127 #endif
 128
 129 class MyServer : public Yaz_Z_Server {
 130 public:
 131     ~MyServer();
 132     MyServer(IYaz_PDU_Observable *the_PDU_Observable);
 133     IYaz_PDU_Observer* sessionNotify(IYaz_PDU_Observable *the_PDU_Observable,
 134                                      int fd);
 135     void failNotify();
 136     void timeoutNotify();
 137     void connectNotify();
 138
 139 private:
 140     MyRetrieval m_retrieval;
 141     MyILL       m_ill;
 142     MyUpdate    m_update;
 143 #if HAVE_YAZ_URSULA_H
 144     MyUrsula    m_ursula;
 145 #endif
 146     int m_no;
 147 };
 148
 149 void MyILL::ill_service (Z_ExtendedServicesRequest *req,
 150                          Z_ItemOrder *io,
 151                          Z_ExtendedServicesResponse *res)
 152 {
 153     yaz_log (LOG_LOG, "MyServer::ill_service");
 154 }
 155
 156 void MyUpdate::update_service (Z_ExtendedServicesRequest *req,
 157                            Z_IUUpdate *io,
 158                            Z_ExtendedServicesResponse *res)
 159 {
 160     yaz_log (LOG_LOG, "MyServer::update_service (v1.1)");
 161 }
 162
 163 void MyUpdate::update_service0 (Z_ExtendedServicesRequest *req,
 164                            Z_IU0Update *io,
 165                                 Z_ExtendedServicesResponse *res)
 166 {
 167     yaz_log (LOG_LOG, "MyServer::update_service (v1.0)");
 168 }
 169
 170 #if HAVE_YAZ_URSULA_H
 171 void MyUrsula::ursula_service (Z_ExtendedServicesRequest *req,
 172                                Z_UrsPDU *u,
 173                                Z_ExtendedServicesResponse *res)
 174 {
 175     yaz_log (LOG_LOG, "MyServer::ursula_service");
 176     switch (u->which)
 177     {
 178     case  Z_UrsPDU_request:
 179         yaz_log(LOG_LOG, "request");
 180         if (u->u.request->libraryNo)
 181             yaz_log (LOG_LOG, "libraryNo: %s", u->u.request->libraryNo);
 182         break;
 183     case  Z_UrsPDU_update:
 184         yaz_log(LOG_LOG, "request");
 185         break;
 186     case  Z_UrsPDU_reservation:
 187         yaz_log(LOG_LOG, "request");
 188         break;
 189     case  Z_UrsPDU_renewal:
 190         yaz_log(LOG_LOG, "request");
 191         break;
 192     default:
 193         yaz_log(LOG_LOG, "unknown");
 194         break;
 195     }
 196 }
 197 #endif
 198
 199 int MyRetrieval::sr_init (Z_InitRequest *initRequest,
 200                        Z_InitResponse *initResponse)
 201 {
 202     yaz_log (LOG_LOG, "MyServer::sr_init");
 203     return 1;
 204 }
 205
 206 void MyRetrieval::sr_search (Z_SearchRequest *searchRequest,
 207                              Z_SearchResponse *searchResponse)
 208 {
 209     yaz_log (LOG_LOG, "MyServer::recv_Z_search");
 210     if (searchRequest->query->which == Z_Query_type_1)
 211     {
 212         Z_RPNStructure *s = searchRequest->query->u.type_1->RPNStructure;
 213         if (s->which == Z_RPNStructure_simple &&
 214             s->u.simple->which == Z_Operand_APT &&
 215             s->u.simple->u.attributesPlusTerm->term->which == Z_Term_general)
 216         {
 217             Odr_oct *term = s->u.simple->u.attributesPlusTerm->term->u.general;
 218             char *str = (char *) odr_malloc (odr_encode(), term->len+1);
 219             if (term->len)
 220                 memcpy (str, term->buf, term->len);
 221             str[term->len] = '\0';
 222             *searchResponse->resultCount = atoi(str);
 223         }
 224     }
 225 }
 226
 227 void MyRetrieval::sr_present (Z_PresentRequest *presentRequest,
 228                                Z_PresentResponse *presentResponse)
 229 {
 230     yaz_log (LOG_LOG, "MyServer::recv_Z_present");
 231 }
 232
 233 void MyRetrieval::sr_record (const char *resultSetName,
 234                              int position,
 235                              int *format,
 236                              Z_RecordComposition *comp,
 237                              Z_NamePlusRecord *namePlusRecord,
 238                              Z_Records *records)
 239 {
 240     yaz_log (LOG_LOG, "MyServer::recv_Z_record");
 241     const char *rec = get_record(position);
 242     create_databaseRecord (odr_encode(), namePlusRecord, 0, VAL_USMARC, rec,
 243                            strlen(rec));
 244 }
 245
 246 MyServer::~MyServer()
 247 {
 248 }
 249
 250 IYaz_PDU_Observer *MyServer::sessionNotify(
 251     IYaz_PDU_Observable *the_PDU_Observable, int fd)
 252 {
 253     MyServer *new_server;
 254     m_no++;
 255     new_server = new MyServer(the_PDU_Observable);
 256     new_server->timeout(900);
 257     new_server->facility_add(&new_server->m_retrieval, "my sr");
 258     new_server->facility_add(&new_server->m_ill, "my ill");
 259     new_server->facility_add(&new_server->m_update, "my update");
 260 #if HAVE_YAZ_URSULA_H
 261     new_server->facility_add(&new_server->m_ursula, "my ursula");
 262 #endif
 263
 264     new_server->set_APDU_log(get_APDU_log());
 265
 266     return new_server;
 267 }
 268
 269 MyServer::MyServer(IYaz_PDU_Observable *the_PDU_Observable) :
 270     Yaz_Z_Server (the_PDU_Observable)
 271 {
 272     m_no = 0;
 273 }
 274
 275 void MyServer::timeoutNotify()
 276 {
 277     yaz_log (LOG_LOG, "connection timed out");
 278     delete this;
 279 }
 280
 281 void MyServer::failNotify()
 282 {
 283     yaz_log (LOG_LOG, "connection closed by client");
 284     delete this;
 285 }
 286
 287 void MyServer::connectNotify()
 288 {
 289 }
 290
 291 void usage(char *prog)
 292 {
 293     fprintf (stderr, "%s: [-a log] [-v level] [-T] @:port\n", prog);
 294     exit (1);
 295 }
 296
 297 int main(int argc, char **argv)
 298 {
 299     int thread_flag = 0;
 300     char *arg;
 301     char *prog = *argv;
 302     char *addr = "tcp:@:9999";
 303     char *apdu_log = 0;
 304
 305     Yaz_SocketManager mySocketManager;
 306
 307     Yaz_PDU_Assoc *my_PDU_Assoc = 0;
 308
 309     MyServer *z = 0;
 310     int ret;
 311
 312     while ((ret = options("a:v:T", argv, argc, &arg)) != -2)
 313     {
 314         switch (ret)
 315         {
 316         case 0:
 317             addr = xstrdup(arg);
 318             break;
 319         case 'a':
 320             apdu_log = xstrdup(arg);
 321             break;
 322         case 'v':
 323             yaz_log_init_level (yaz_log_mask_str(arg));
 324             break;
 325         case 'T':
 326             thread_flag = 1;
 327             break;
 328         default:
 329             usage(prog);
 330             return 1;
 331         }
 332     }
 333     if (thread_flag)
 334         my_PDU_Assoc = new Yaz_PDU_AssocThread(&mySocketManager);
 335     else
 336         my_PDU_Assoc = new Yaz_PDU_Assoc(&mySocketManager);
 337
 338     z = new MyServer(my_PDU_Assoc);
 339     z->server(addr);
 340     if (apdu_log)
 341     {
 342         yaz_log (LOG_LOG, "set_APDU_log %s", apdu_log);
 343         z->set_APDU_log(apdu_log);
 344     }
 345
 346     while (mySocketManager.processEvent() > 0)
 347         ;
 348     delete z;
 349     return 0;
 350 }