/[pgswish]/trunk/pgswish.c
This is repository of my old source code which isn't updated any more. Go to git.rot13.org for current projects!
ViewVC logotype

Diff of /trunk/pgswish.c

Parent Directory Parent Directory | Revision Log Revision Log | View Patch Patch

revision 10 by dpavlin, Sat Feb 19 11:39:11 2005 UTC revision 18 by dpavlin, Sun Feb 20 22:58:25 2005 UTC
# Line 25  Line 25 
25  #include "fmgr.h"  #include "fmgr.h"
26  #include "funcapi.h"  #include "funcapi.h"
27  #include "utils/builtins.h"  #include "utils/builtins.h"
28    #include "utils/array.h"
29    #include "miscadmin.h"
30  #include <swish-e.h>  #include <swish-e.h>
31    
32  #define _textin(str) DirectFunctionCall1(textin, CStringGetDatum(str))  #define _textin(str) DirectFunctionCall1(textin, CStringGetDatum(str))
33  #define _textout(str) DatumGetPointer(DirectFunctionCall1(textout, PointerGetDatum(str)))  #define _textout(str) DatumGetPointer(DirectFunctionCall1(textout, PointerGetDatum(str)))
34    #define GET_STR(textp) DatumGetCString(DirectFunctionCall1(textout, PointerGetDatum(textp)))
35    #define GET_TEXT(cstrp) DatumGetTextP(DirectFunctionCall1(textin, CStringGetDatum(cstrp)))
36    
37    
38  SW_HANDLE   swish_handle = NULL;/* Database handle */  SW_HANDLE   swish_handle = NULL;/* Database handle */
39  SW_SEARCH   search = NULL;      /* search handle -- holds search parameters */  SW_SEARCH   search = NULL;      /* search handle -- holds search parameters */
40  SW_RESULTS  results = NULL;     /* results handle -- holds list of results */  SW_RESULTS  swish_results = NULL;       /* results handle -- holds list of results */
41    SW_RESULT   *sw_res = NULL;     /* one row from swish-e results */
42    
43  /* define PostgreSQL v1 function */  /* define PostgreSQL v1 function */
44  PG_FUNCTION_INFO_V1(pgswish);  PG_FUNCTION_INFO_V1(pgswish);
# Line 45  Datum pgswish(PG_FUNCTION_ARGS) { Line 50  Datum pgswish(PG_FUNCTION_ARGS) {
50          TupleDesc       tupdesc;          TupleDesc       tupdesc;
51          TupleTableSlot  *slot;          TupleTableSlot  *slot;
52          AttInMetadata   *attinmeta;          AttInMetadata   *attinmeta;
         SW_HANDLE       swish_handle = NULL;    /* Database handle */  
         SW_SEARCH       search = NULL;          /* search handle -- holds search parameters */  
         SW_RESULTS      results = NULL;         /* results handle -- holds list of results */  
53          char            *index_path;          char            *index_path;
54          char            *query;          char            *query;
55    
# Line 79  Datum pgswish(PG_FUNCTION_ARGS) { Line 81  Datum pgswish(PG_FUNCTION_ARGS) {
81                          SRF_RETURN_DONE(funcctx);                          SRF_RETURN_DONE(funcctx);
82                  }                  }
83                                    
84                  if ( SwishError( swish_handle ) ) error_or_abort( swish_handle );                  error_or_abort( swish_handle );
85                  /* set ranking scheme. default is 0 */                  /* set ranking scheme. default is 0 */
86                  SwishRankScheme( swish_handle, 0 );                  SwishRankScheme( swish_handle, 0 );
87                    error_or_abort( swish_handle );
                 /* Check for errors after every call */  
                 if ( SwishError( swish_handle ) )  
                         error_or_abort( swish_handle );  /* print an error or abort -- see below */  
88    
89                  elog(INFO, "pgswish: SwishQuery(%s)", query);                  elog(INFO, "pgswish: SwishQuery(%s)", query);
90                  /* Here's a short-cut to searching that creates a search object and searches at the same time */                  /* Here's a short-cut to searching that creates a search object and searches at the same time */
91                  results = SwishQuery( swish_handle, query);                  swish_results = SwishQuery( swish_handle, query);
92                  if ( SwishError( swish_handle ) ) error_or_abort( swish_handle );                  error_or_abort( swish_handle );
93    
94                  /* total number of tuples to be returned */                  /* total number of tuples to be returned */
95                  funcctx->max_calls = SwishHits( results );                  funcctx->max_calls = SwishHits( swish_results );
96    
97                  /* check if results exists */                  /* check if results exists */
98                  if ( 0 == funcctx->max_calls )                  if ( 0 == funcctx->max_calls )
# Line 118  Datum pgswish(PG_FUNCTION_ARGS) { Line 117  Datum pgswish(PG_FUNCTION_ARGS) {
117                  funcctx->attinmeta = attinmeta;                  funcctx->attinmeta = attinmeta;
118    
119                  MemoryContextSwitchTo(oldcontext);                  MemoryContextSwitchTo(oldcontext);
120    
121                    elog(INFO, "SRF_IS_FIRSTCALL done");
122          }          }
123    
124          /* stuff done on every call of the function */          /* stuff done on every call of the function */
# Line 133  Datum pgswish(PG_FUNCTION_ARGS) { Line 134  Datum pgswish(PG_FUNCTION_ARGS) {
134                  HeapTuple       tuple;                  HeapTuple       tuple;
135                  Datum           result;                  Datum           result;
136    
137  if (0) {                  elog(INFO, "pgswish: loop count %d", call_cntr);
138    
139                    if (! swish_results) {
140                            elog(ERROR, "pgswish: no swish-e results");
141                            SRF_RETURN_DONE(funcctx);
142                    }
143                    
144                    elog(DEBUG1, "pgswish: check for swish-e error");
145                    error_or_abort( swish_handle );
146    
147                  /*                  /*
148                   * Prepare a values array for storage in our slot.                   * Prepare a values array for storage in our slot.
149                   * This should be an array of C strings which will                   * This should be an array of C strings which will
150                   * be processed later by the type input functions.                   * be processed later by the type input functions.
151                   */                   */
152                  values = (char **) palloc(5 * sizeof(char *));  
153                  values[0] = _textout( SwishResultPropertyULong ( result, "swishrank" ) );                  sw_res = SwishNextResult( swish_results );
154                  values[1] = _textout( SwishResultPropertyStr   ( result, "swishdocpath" ) );                  if (! sw_res) {
155                  values[2] = _textout( SwishResultPropertyStr   ( result, "swishtitle" ) );                          elog(ERROR, "pgswish: swish-e sort result list: %d rows expected %d", call_cntr, max_calls - 1);
156                  values[3] = _textout( SwishResultPropertyStr   ( result, "swishdocsize" ) );                          SRF_RETURN_DONE(funcctx);
157                  values[4] = _textout( SwishResultPropertyStr   ( result, "swishdbfile" ) );                  }
158                    
159                    elog(INFO, "Path: %s\n  Rank: %lu\n  Size: %lu\n  Title: %s\n  Index: %s\n  Modified: %s\n  Record #: %lu\n  File   #: %lu\n\n",
160                            SwishResultPropertyStr   ( sw_res, "swishdocpath" ),
161                            SwishResultPropertyULong ( sw_res, "swishrank" ),
162                            SwishResultPropertyULong ( sw_res, "swishdocsize" ),
163                            SwishResultPropertyStr   ( sw_res, "swishtitle"),
164                            SwishResultPropertyStr   ( sw_res, "swishdbfile" ),
165                            SwishResultPropertyStr   ( sw_res, "swishlastmodified" ),
166                            SwishResultPropertyULong ( sw_res, "swishreccount" ),  /* can figure this out in loop, of course */
167                            SwishResultPropertyULong ( sw_res, "swishfilenum" )
168                    );
169    
170                    values = (char **) palloc(4 * sizeof(char *));
171    
172                    values[0] = prop2int( sw_res, "swishrank" );
173                    values[1] = prop2text( sw_res, "swishdocpath" );
174                    values[2] = prop2text( sw_res, "swishtitle" );
175                    values[3] = prop2int( sw_res, "swishdocsize" );
176    
177    /*
178                    values[0] = (char *) palloc(16 * sizeof(char));
179                    snprintf(values[0], 16, "%d", 1);
180                    values[1] = (char *) palloc(16 * sizeof(char));
181                    snprintf(values[1], 16, "%d", 2);
182                    values[2] = (char *) palloc(16 * sizeof(char));
183                    snprintf(values[2], 16, "%d", 3);
184                    values[3] = (char *) palloc(16 * sizeof(char));
185                    snprintf(values[3], 16, "%d", 4);
186    */
187    
188                  /* build a tuple */                  /* build a tuple */
189                  tuple = BuildTupleFromCStrings(attinmeta, values);                  tuple = BuildTupleFromCStrings(attinmeta, values);
# Line 152  if (0) { Line 191  if (0) {
191                  /* make the tuple into a datum */                  /* make the tuple into a datum */
192                  result = TupleGetDatum(slot, tuple);                  result = TupleGetDatum(slot, tuple);
193    
194  }                  /* clean up ? */
195                  /* clean up (this is not really necessary) */                  pfree(values[0]);
196                    pfree(values[1]);
197                    pfree(values[2]);
198                    pfree(values[3]);
199                    pfree(values);
200                    
201                    elog(DEBUG1, "row: %s|%s|%s|%s",values[0],values[1],values[2],values[3]);
202            
203                  SRF_RETURN_NEXT(funcctx, result);                  SRF_RETURN_NEXT(funcctx, result);
204          } else {          } else {
205                    elog(INFO, "loop over");
206    
207                  /* free swish object and close */                  /* free swish object and close */
208                  Free_Search_Object( search );                  Free_Search_Object( search );
209                  SwishClose( swish_handle );                  SwishClose( swish_handle );
# Line 166  if (0) { Line 213  if (0) {
213          }          }
214  }  }
215    
216    /* work in progress */
217    PG_FUNCTION_INFO_V1(pgswish2);
218    Datum pgswish2(PG_FUNCTION_ARGS)
219    {
220            int             nrows = 3;
221            int16           typlen;
222            bool            typbyval;
223            char            typalign;
224            ReturnSetInfo   *rsinfo = (ReturnSetInfo *) fcinfo->resultinfo;
225            AttInMetadata   *attinmeta;
226            TupleDesc       tupdesc;
227            Tuplestorestate *tupstore = NULL;
228            HeapTuple       tuple;
229            MemoryContext   per_query_ctx;
230            MemoryContext   oldcontext;
231            Datum           dvalue;
232            char            **values;
233            int             ncols;
234            int             i, j;
235    
236            /* check to see if caller supports us returning a tuplestore */
237            if (!rsinfo || !(rsinfo->allowedModes & SFRM_Materialize))
238                    ereport(ERROR,
239                                    (errcode(ERRCODE_SYNTAX_ERROR),
240                                     errmsg("materialize mode required, but it is not " \
241                                                    "allowed in this context")));
242    
243            /* get the requested return tuple description */
244            tupdesc = rsinfo->expectedDesc;
245            ncols = tupdesc->natts;
246    
247            /*
248             * The requested tuple description better match up with the array
249             * we were given.
250             */
251            /* OK, use it */
252            attinmeta = TupleDescGetAttInMetadata(tupdesc);
253    
254            /* Now go to work */
255            rsinfo->returnMode = SFRM_Materialize;
256    
257            per_query_ctx = fcinfo->flinfo->fn_mcxt;
258            oldcontext = MemoryContextSwitchTo(per_query_ctx);
259    
260            /* initialize our tuplestore */
261            tupstore = tuplestore_begin_heap(true, false, SortMem);
262    
263            values = (char **) palloc(ncols * sizeof(char *));
264    
265            for (i = 0; i < nrows; i++)
266            {
267                    for (j = 0; j < ncols; j++)
268                    {
269                            values[j] = DatumGetCString( "foo" );
270                    }
271                    /* construct the tuple */
272                    tuple = BuildTupleFromCStrings(attinmeta, values);
273    
274                    /* now store it */
275                    tuplestore_puttuple(tupstore, tuple);
276            }
277    
278            tuplestore_donestoring(tupstore);
279            rsinfo->setResult = tupstore;
280    
281            /*
282             * SFRM_Materialize mode expects us to return a NULL Datum. The actual
283             * tuples are in our tuplestore and passed back through
284             * rsinfo->setResult. rsinfo->setDesc is set to the tuple description
285             * that we actually used to build our tuples with, so the caller can
286             * verify we did what it was expecting.
287             */
288            rsinfo->setDesc = tupdesc;
289            MemoryContextSwitchTo(oldcontext);
290    
291            return (Datum) 0;
292    }
293    
294    
295    /* make text var prom property */
296    char *prop2text(SW_RESULT sw_res, char *propname) {
297            char *val;
298            char *prop;
299            int len;
300    
301            elog(DEBUG2, "prop2text(%s)", propname);
302    
303            prop = SwishResultPropertyStr( sw_res, propname );
304            error_or_abort( swish_handle );
305    
306            len = strlen(prop);
307            elog(DEBUG1, "prop2text(%s) = '%s' %d bytes", propname, prop, len);
308    
309            len++;
310            len *= sizeof(char);
311    
312            elog(DEBUG2, "palloc(%d)", len);
313    
314            val = palloc(len);
315    
316            memset(val, 0, len);
317            strncpy(val, prop, len);
318    
319            elog(DEBUG2, "val=%s", val);
320    
321            return val;
322    }
323    
324    /* make integer variable from property */
325    char *prop2int(SW_RESULT sw_res, char *propname) {
326            char *val;
327            unsigned long prop;
328            int len;
329    
330            elog(DEBUG2, "prop2int(%s)", propname);
331    
332            prop = SwishResultPropertyULong( sw_res, propname );
333            error_or_abort( swish_handle );
334    
335            elog(DEBUG1, "prop2int(%s) = %lu", propname, prop);
336    
337            len = 128 * sizeof(char);
338            elog(DEBUG2, "palloc(%d)", len);
339    
340            val = palloc(len);
341            memset(val, 0, len);
342    
343            snprintf(val, len, "%lu", prop);
344    
345            elog(DEBUG2, "val=%s", val);
346    
347            return val;
348    }
349    
350    
351  /*  /*
352   * elog errors   * check if swish has returned error, and elog it.
  *  
353   */   */
   
354  static void error_or_abort( SW_HANDLE swish_handle ) {  static void error_or_abort( SW_HANDLE swish_handle ) {
355          if ( !SwishError( swish_handle ) )          if ( !SwishError( swish_handle ) )
356                  return;                  return;
# Line 184  static void error_or_abort( SW_HANDLE sw Line 364  static void error_or_abort( SW_HANDLE sw
364          );          );
365          if ( search ) Free_Search_Object( search );          if ( search ) Free_Search_Object( search );
366          SwishClose( swish_handle );          SwishClose( swish_handle );
   
         /* do when there is no more left */  
367  }  }
368    
Legend:
Removed from v.10  
changed lines
  Added in v.18
  ViewVC Help
Powered by ViewVC 1.1.26