index/apitest.c

   1
   2 #include <stdio.h>
   3
   4 #include <log.h>
   5 #include <pquery.h>
   6 #include "zebraapi.h"
   7
   8 /* Small routine to display GRS-1 record variants ... */
   9 /* Copied verbatim from yaz/client/client.c */
  10 static void display_variant(Z_Variant *v, int level)
  11 {
  12     int i;
  13
  14     for (i = 0; i < v->num_triples; i++)
  15     {
  16         printf("%*sclass=%d,type=%d", level * 4, "", *v->triples[i]->zclass,
  17             *v->triples[i]->type);
  18         if (v->triples[i]->which == Z_Triple_internationalString)
  19             printf(",value=%s\n", v->triples[i]->value.internationalString);
  20         else
  21             printf("\n");
  22     }
  23 }
  24
  25 /* Small routine to display a GRS-1 record ... */
  26 /* Copied verbatim from yaz/client/client.c */
  27 static void display_grs1(Z_GenericRecord *r, int level)
  28 {
  29     int i;
  30
  31     if (!r)
  32         return;
  33     for (i = 0; i < r->num_elements; i++)
  34     {
  35         Z_TaggedElement *t;
  36
  37         printf("%*s", level * 4, "");
  38         t = r->elements[i];
  39         printf("(");
  40         if (t->tagType)
  41             printf("%d,", *t->tagType);
  42         else
  43             printf("?,");
  44         if (t->tagValue->which == Z_StringOrNumeric_numeric)
  45             printf("%d) ", *t->tagValue->u.numeric);
  46         else
  47             printf("%s) ", t->tagValue->u.string);
  48         if (t->content->which == Z_ElementData_subtree)
  49         {
  50             printf("\n");
  51             display_grs1(t->content->u.subtree, level+1);
  52         }
  53         else if (t->content->which == Z_ElementData_string)
  54             printf("%s\n", t->content->u.string);
  55         else if (t->content->which == Z_ElementData_numeric)
  56             printf("%d\n", *t->content->u.numeric);
  57         else if (t->content->which == Z_ElementData_oid)
  58         {
  59             int *ip = t->content->u.oid;
  60             oident *oent;
  61
  62             if ((oent = oid_getentbyoid(t->content->u.oid)))
  63                 printf("OID: %s\n", oent->desc);
  64             else
  65             {
  66                 printf("{");
  67                 while (ip && *ip >= 0)
  68                     printf(" %d", *(ip++));
  69                 printf(" }\n");
  70             }
  71         }
  72         else if (t->content->which == Z_ElementData_noDataRequested)
  73             printf("[No data requested]\n");
  74         else if (t->content->which == Z_ElementData_elementEmpty)
  75             printf("[Element empty]\n");
  76         else if (t->content->which == Z_ElementData_elementNotThere)
  77             printf("[Element not there]\n");
  78         else
  79             printf("??????\n");
  80         if (t->appliedVariant)
  81             display_variant(t->appliedVariant, level+1);
  82         if (t->metaData && t->metaData->supportedVariants)
  83         {
  84             int c;
  85
  86             printf("%*s---- variant list\n", (level+1)*4, "");
  87             for (c = 0; c < t->metaData->num_supportedVariants; c++)
  88             {
  89                 printf("%*svariant #%d\n", (level+1)*4, "", c);
  90                 display_variant(t->metaData->supportedVariants[c], level + 2);
  91             }
  92         }
  93     }
  94 }
  95
  96 /* Small test main to illustrate the use of the C api */
  97 int main (int argc, char **argv)
  98 {
  99     /* odr is a handle to memory assocated with RETURNED data from
 100        various functions */
 101     ODR odr = odr_createmem (ODR_ENCODE);
 102
 103     /* zh is our Zebra Handle - describes the server as a whole */
 104     ZebraHandle zh;
 105
 106     /* the database we specify in our example */
 107     char *base = "Default";
 108     int argno;
 109
 110     /* open Zebra */
 111     zh = zebra_open ("zebra.cfg");
 112     if (!zh)
 113     {
 114         printf ("Couldn't init zebra\n");
 115         exit (1);
 116     }
 117
 118     /* This call controls the logging facility in YAZ/Zebra */
 119 #if 0
 120     log_init(LOG_ALL, "", "out.log");
 121 #endif
 122
 123     /* Each argument to main will be a query */
 124     for (argno = 1; argno < argc; argno++)
 125     {
 126         /* parse the query and generate an RPN structure */
 127         Z_RPNQuery *query = p_query_rpn (odr, PROTO_Z3950, argv[argno]);
 128         char setname[64];
 129         int errCode;
 130         int i;
 131         const char *errString;
 132         char *errAdd;
 133         ZebraRetrievalRecord *records;
 134         int noOfRecordsToFetch;
 135
 136         /* bad query? */
 137         if (!query)
 138         {
 139             logf (LOG_WARN, "bad query %s\n", argv[argno]);
 140             odr_reset (odr);
 141             continue;
 142         }
 143
 144         /* result set name will be called 1,2, etc */
 145         sprintf (setname, "%d", i);
 146
 147         /* fire up the search */
 148         zebra_search_rpn (zh, odr, query, 1, &base, setname);
 149
 150         /* status ... */
 151         errCode = zebra_errCode (zh);
 152         errString = zebra_errString (zh);
 153         errAdd = zebra_errAdd (zh);
 154
 155         /* error? */
 156         if (errCode)
 157         {
 158             printf ("Zebra Search Error %d %s %s\n",
 159                     errCode, errString, errAdd ? errAdd : "");
 160             continue;
 161         }
 162         /* ok ... */
 163         printf ("Zebra Search gave %d hits\n", zebra_hits (zh));
 164
 165         /* Deterimine number of records to fetch ... */
 166         if (zebra_hits(zh) > 10)
 167             noOfRecordsToFetch = 10;
 168         else
 169             noOfRecordsToFetch = zebra_hits(zh);
 170
 171         /* reset our memory - we've finished dealing with search */
 172         odr_reset (odr);
 173
 174         /* prepare to fetch ... */
 175         records = malloc (sizeof(*records) * noOfRecordsToFetch);
 176         /* specify position of each record to fetch */
 177         /* first one is numbered 1 and NOT 0 */
 178         for (i = 0; i<noOfRecordsToFetch; i++)
 179             records[i].position = i+1;
 180         /* fetch them and request for GRS-1 records */
 181         zebra_records_retrieve (zh, odr, setname, NULL, VAL_GRS1,
 182                                 noOfRecordsToFetch, records);
 183
 184         /* status ... */
 185         errCode = zebra_errCode (zh);
 186         errString = zebra_errString (zh);
 187         errAdd = zebra_errAdd (zh);
 188
 189         /* error ? */
 190         if (errCode)
 191         {
 192             printf ("Zebra Search Error %d %s %s\n",
 193                     errCode, errString, errAdd ? errAdd : "");
 194         }
 195         else
 196         {
 197             /* inspect each record in result */
 198             for (i = 0; i<noOfRecordsToFetch; i++)
 199             {
 200                 printf ("Record %d\n", i+1);
 201                 /* error when fetching this record? */
 202                 if (records[i].errCode)
 203                 {
 204                     printf ("  Error %d\n", records[i].errCode);
 205                     continue;
 206                 }
 207                 /* GRS-1 record ? */
 208                 if (records[i].format == VAL_GRS1)
 209                 {
 210                     Z_GenericRecord *grs_record =
 211                         (Z_GenericRecord *) records[i].buf;
 212                     printf ("  GRS-1\n");
 213                     display_grs1(grs_record, 0);
 214                 }
 215                 /* some other record we don't handle yet... */
 216                 else
 217                 {
 218                     printf ("  Other record (ignored)\n");
 219                 }
 220             }
 221         }
 222         free (records);
 223         odr_reset (odr);  /* reset memory */
 224     }
 225     return 0;
 226 }