92 |
} |
} |
93 |
|
|
94 |
/* limit */ |
/* limit */ |
95 |
if (! PG_ARGISNULL(3)) limit = PG_GETARG_INT32(3); |
if (PG_ARGISNULL(3)) { |
96 |
|
limit = 0; |
97 |
|
} else { |
98 |
|
limit = PG_GETARG_INT32(3); |
99 |
|
} |
100 |
|
|
101 |
/* offset */ |
/* offset */ |
102 |
if (! PG_ARGISNULL(4)) offset = PG_GETARG_INT32(4); |
if (PG_ARGISNULL(4)) { |
103 |
|
offset = 0; |
104 |
|
} else { |
105 |
|
offset = PG_GETARG_INT32(4); |
106 |
|
} |
107 |
|
|
108 |
|
|
109 |
/* open the database */ |
/* open the database */ |
114 |
SRF_RETURN_DONE(funcctx); |
SRF_RETURN_DONE(funcctx); |
115 |
} |
} |
116 |
|
|
117 |
elog(INFO, "pgest: query[%s] attr[%s] limit %d offset %d", query, (PG_ARGISNULL(2) ? "NULL" : attr), limit, offset); |
elog(DEBUG1, "pgest: query[%s] attr[%s] limit %d offset %d", query, (PG_ARGISNULL(2) ? "NULL" : attr), limit, offset); |
118 |
|
|
119 |
/* create a search condition object */ |
/* create a search condition object */ |
120 |
if (!(cond = est_cond_new())) { |
if (!(cond = est_cond_new())) { |
128 |
|
|
129 |
/* minimum valid attribute length is 10: @a STREQ a */ |
/* minimum valid attribute length is 10: @a STREQ a */ |
130 |
if (! PG_ARGISNULL(2) && strlen(attr) >= 10) { |
if (! PG_ARGISNULL(2) && strlen(attr) >= 10) { |
131 |
elog(INFO,"est_cond_add_attr(%s)", attr); |
elog(DEBUG1,"est_cond_add_attr(%s)", attr); |
132 |
est_cond_add_attr(cond, attr); |
est_cond_add_attr(cond, attr); |
133 |
} |
} |
134 |
|
|
178 |
attinmeta = funcctx->attinmeta; |
attinmeta = funcctx->attinmeta; |
179 |
|
|
180 |
if (limit && call_cntr > limit - 1) { |
if (limit && call_cntr > limit - 1) { |
181 |
elog(INFO, "call_cntr: %d limit: %d", call_cntr, limit); |
elog(DEBUG1, "call_cntr: %d limit: %d", call_cntr, limit); |
182 |
SRF_RETURN_DONE(funcctx); |
SRF_RETURN_DONE(funcctx); |
183 |
} |
} |
184 |
|
|
214 |
values[0] = (char *) attr2text(doc,"@id"); |
values[0] = (char *) attr2text(doc,"@id"); |
215 |
values[1] = (char *) attr2text(doc,"@uri"); |
values[1] = (char *) attr2text(doc,"@uri"); |
216 |
values[2] = (char *) attr2text(doc,"@title"); |
values[2] = (char *) attr2text(doc,"@title"); |
217 |
values[3] = (char *) attr2text(doc,"@type"); |
values[3] = (char *) attr2text(doc,"@size"); |
218 |
|
|
219 |
/* destloy the document object */ |
/* destloy the document object */ |
220 |
elog(DEBUG2, "est_doc_delete"); |
elog(DEBUG2, "est_doc_delete"); |
262 |
} |
} |
263 |
|
|
264 |
/* work in progress */ |
/* work in progress */ |
265 |
PG_FUNCTION_INFO_V1(pgest2); |
PG_FUNCTION_INFO_V1(pgest_attr); |
266 |
Datum pgest2(PG_FUNCTION_ARGS) |
Datum pgest_attr(PG_FUNCTION_ARGS) |
267 |
{ |
{ |
268 |
int nrows = 3; |
ArrayType *attr_arr = PG_GETARG_ARRAYTYPE_P(5); |
269 |
|
Oid element_type = ARR_ELEMTYPE(attr_arr); |
270 |
|
int ndims = ARR_NDIM(attr_arr); |
271 |
|
int *dim_counts = ARR_DIMS(attr_arr); |
272 |
|
int *dim_lower_bounds = ARR_LBOUND(attr_arr); |
273 |
|
int ncols = 0; |
274 |
|
int nrows = 0; |
275 |
|
int indx[MAXDIM]; |
276 |
int16 typlen; |
int16 typlen; |
277 |
bool typbyval; |
bool typbyval; |
278 |
char typalign; |
char typalign; |
279 |
ReturnSetInfo *rsinfo = (ReturnSetInfo *) fcinfo->resultinfo; |
ReturnSetInfo *rsinfo = (ReturnSetInfo *) fcinfo->resultinfo; |
280 |
AttInMetadata *attinmeta; |
AttInMetadata *attinmeta; |
281 |
TupleDesc tupdesc; |
TupleDesc tupdesc; |
282 |
Tuplestorestate *tupstore = NULL; |
Tuplestorestate *tupstore = NULL; |
283 |
HeapTuple tuple; |
HeapTuple tuple; |
284 |
MemoryContext per_query_ctx; |
MemoryContext per_query_ctx; |
285 |
MemoryContext oldcontext; |
MemoryContext oldcontext; |
286 |
Datum dvalue; |
Datum dvalue; |
287 |
char **values; |
char **values; |
288 |
int ncols; |
int rsinfo_ncols; |
289 |
int i, j; |
int i, j; |
290 |
|
/* estvars */ |
291 |
|
char *index_path; |
292 |
|
char *query; |
293 |
|
char *attr; |
294 |
|
|
295 |
|
|
296 |
|
/* only allow 1D input array */ |
297 |
|
if (ndims == 1) |
298 |
|
{ |
299 |
|
ncols = dim_counts[0]; |
300 |
|
} |
301 |
|
else |
302 |
|
ereport(ERROR, |
303 |
|
(errcode(ERRCODE_INVALID_PARAMETER_VALUE), |
304 |
|
errmsg("invalid input array"), |
305 |
|
errdetail("Input array must have 1 dimension"))); |
306 |
|
|
307 |
/* check to see if caller supports us returning a tuplestore */ |
/* check to see if caller supports us returning a tuplestore */ |
308 |
if (!rsinfo || !(rsinfo->allowedModes & SFRM_Materialize)) |
if (!rsinfo || !(rsinfo->allowedModes & SFRM_Materialize)) |
309 |
ereport(ERROR, |
ereport(ERROR, |
311 |
errmsg("materialize mode required, but it is not " \ |
errmsg("materialize mode required, but it is not " \ |
312 |
"allowed in this context"))); |
"allowed in this context"))); |
313 |
|
|
314 |
|
/* get info about element type needed to construct the array */ |
315 |
|
get_typlenbyvalalign(element_type, &typlen, &typbyval, &typalign); |
316 |
|
|
317 |
/* get the requested return tuple description */ |
/* get the requested return tuple description */ |
318 |
tupdesc = rsinfo->expectedDesc; |
tupdesc = rsinfo->expectedDesc; |
319 |
ncols = tupdesc->natts; |
rsinfo_ncols = tupdesc->natts; |
320 |
|
|
321 |
/* |
/* |
322 |
* The requested tuple description better match up with the array |
* The requested tuple description better match up with the array |
323 |
* we were given. |
* we were given. |
324 |
*/ |
*/ |
325 |
|
if (rsinfo_ncols != ncols) |
326 |
|
ereport(ERROR, |
327 |
|
(errcode(ERRCODE_INVALID_PARAMETER_VALUE), |
328 |
|
errmsg("invalid input array"), |
329 |
|
errdetail("Number of elements in array must match number of query specified columns."))); |
330 |
|
|
331 |
/* OK, use it */ |
/* OK, use it */ |
332 |
attinmeta = TupleDescGetAttInMetadata(tupdesc); |
attinmeta = TupleDescGetAttInMetadata(tupdesc); |
333 |
|
|
340 |
/* initialize our tuplestore */ |
/* initialize our tuplestore */ |
341 |
tupstore = tuplestore_begin_heap(true, false, SortMem); |
tupstore = tuplestore_begin_heap(true, false, SortMem); |
342 |
|
|
343 |
|
|
344 |
|
/* take rest of arguments from function */ |
345 |
|
|
346 |
|
/* index path */ |
347 |
|
if (PG_ARGISNULL(0)) { |
348 |
|
ereport(ERROR, |
349 |
|
(errcode(ERRCODE_INVALID_PARAMETER_VALUE), |
350 |
|
errmsg("index path can't be null"), |
351 |
|
errdetail("Index path must be valid full path to HyperEstraier index"))); |
352 |
|
} |
353 |
|
index_path = _textout(PG_GETARG_TEXT_P(0)); |
354 |
|
|
355 |
|
/* query string */ |
356 |
|
if (PG_ARGISNULL(0)) { |
357 |
|
query = ""; |
358 |
|
} else { |
359 |
|
query = _textout(PG_GETARG_TEXT_P(1)); |
360 |
|
} |
361 |
|
|
362 |
|
/* atribute filter */ |
363 |
|
if (PG_ARGISNULL(2)) { |
364 |
|
attr = ""; |
365 |
|
} else { |
366 |
|
attr = _textout(PG_GETARG_TEXT_P(2)); |
367 |
|
} |
368 |
|
|
369 |
|
/* limit */ |
370 |
|
if (PG_ARGISNULL(3)) { |
371 |
|
limit = 0; |
372 |
|
} else { |
373 |
|
limit = PG_GETARG_INT32(3); |
374 |
|
} |
375 |
|
|
376 |
|
/* offset */ |
377 |
|
if (PG_ARGISNULL(4)) { |
378 |
|
offset = 0; |
379 |
|
} else { |
380 |
|
offset = PG_GETARG_INT32(4); |
381 |
|
} |
382 |
|
|
383 |
|
|
384 |
|
/* open the database */ |
385 |
|
elog(DEBUG1, "pgest_attr: est_db_open(%s)", index_path); |
386 |
|
|
387 |
|
if(!(db = est_db_open(index_path, ESTDBREADER, &ecode))){ |
388 |
|
ereport(ERROR, (errcode(ERRCODE_INVALID_PARAMETER_VALUE), |
389 |
|
errmsg("est_db_open: can't open %s: %d", index_path, ecode), |
390 |
|
errdetail(est_err_msg(ecode)))); |
391 |
|
} |
392 |
|
|
393 |
|
elog(DEBUG1, "pgest_attr: query[%s] attr[%s] limit %d offset %d", query, (PG_ARGISNULL(2) ? "NULL" : attr), limit, offset); |
394 |
|
|
395 |
|
/* create a search condition object */ |
396 |
|
if (!(cond = est_cond_new())) { |
397 |
|
ereport(ERROR, (errcode(ERRCODE_QUERY_CANCELED), |
398 |
|
errmsg("pgest_attr: est_cond_new failed"))); |
399 |
|
} |
400 |
|
|
401 |
|
/* set the search phrase to the search condition object */ |
402 |
|
if (! PG_ARGISNULL(1) && strlen(query) > 0) |
403 |
|
est_cond_set_phrase(cond, query); |
404 |
|
|
405 |
|
/* minimum valid attribute length is 10: @a STREQ a */ |
406 |
|
if (! PG_ARGISNULL(2) && strlen(attr) >= 10) { |
407 |
|
elog(DEBUG1,"est_cond_add_attr(%s)", attr); |
408 |
|
est_cond_add_attr(cond, attr); |
409 |
|
} |
410 |
|
|
411 |
|
/* get the result of search */ |
412 |
|
est_result = est_db_search(db, cond, &resnum, NULL); |
413 |
|
|
414 |
|
/* check if results exists */ |
415 |
|
if ( 0 == resnum ) { |
416 |
|
elog(INFO, "pgest_attr: no results for: %s", query ); |
417 |
|
} |
418 |
|
|
419 |
|
/* total number of tuples to be returned */ |
420 |
|
if (limit && limit < resnum) { |
421 |
|
nrows = limit - offset; |
422 |
|
} else { |
423 |
|
nrows = resnum - offset; |
424 |
|
} |
425 |
|
|
426 |
|
|
427 |
|
elog(DEBUG1, "pgest_attr: found %d hits for %s", resnum, query); |
428 |
|
|
429 |
|
|
430 |
values = (char **) palloc(ncols * sizeof(char *)); |
values = (char **) palloc(ncols * sizeof(char *)); |
431 |
|
|
432 |
for (i = 0; i < nrows; i++) |
for (i = 0; i < nrows; i++) |
433 |
{ |
{ |
434 |
|
|
435 |
|
/* get result from estraier */ |
436 |
|
if (! ( doc = est_db_get_doc(db, est_result[i + offset], 0)) ) { |
437 |
|
elog(INFO, "can't find result %d", i + offset); |
438 |
|
} else { |
439 |
|
elog(DEBUG1, "URI: %s\n Title: %s\n", |
440 |
|
est_doc_attr(doc, "@uri"), |
441 |
|
est_doc_attr(doc, "@title") |
442 |
|
); |
443 |
|
} |
444 |
|
|
445 |
|
/* iterate over results */ |
446 |
for (j = 0; j < ncols; j++) |
for (j = 0; j < ncols; j++) |
447 |
{ |
{ |
448 |
values[j] = DatumGetCString( "foo" ); |
bool isnull; |
449 |
|
|
450 |
|
/* array value of this position */ |
451 |
|
indx[0] = j + dim_lower_bounds[0]; |
452 |
|
|
453 |
|
dvalue = array_ref(attr_arr, ndims, indx, -1, typlen, typbyval, typalign, &isnull); |
454 |
|
|
455 |
|
if (!isnull && doc) |
456 |
|
values[j] = DatumGetCString( |
457 |
|
attr2text(doc, |
458 |
|
(char *)DirectFunctionCall1(textout, dvalue) |
459 |
|
)); |
460 |
|
else |
461 |
|
values[j] = NULL; |
462 |
} |
} |
463 |
/* construct the tuple */ |
/* construct the tuple */ |
464 |
tuple = BuildTupleFromCStrings(attinmeta, values); |
tuple = BuildTupleFromCStrings(attinmeta, values); |
465 |
|
|
466 |
/* now store it */ |
/* now store it */ |
467 |
tuplestore_puttuple(tupstore, tuple); |
tuplestore_puttuple(tupstore, tuple); |
468 |
|
|
469 |
|
|
470 |
|
/* delete estraier document object */ |
471 |
|
est_doc_delete(doc); |
472 |
} |
} |
473 |
|
|
474 |
tuplestore_donestoring(tupstore); |
tuplestore_donestoring(tupstore); |
484 |
rsinfo->setDesc = tupdesc; |
rsinfo->setDesc = tupdesc; |
485 |
MemoryContextSwitchTo(oldcontext); |
MemoryContextSwitchTo(oldcontext); |
486 |
|
|
487 |
|
if(!est_db_close(db, &ecode)){ |
488 |
|
ereport(ERROR, (errcode(ERRCODE_IO_ERROR), |
489 |
|
errmsg("est_db_close: %d", ecode), |
490 |
|
errdetail(est_err_msg(ecode)))); |
491 |
|
} |
492 |
|
|
493 |
return (Datum) 0; |
return (Datum) 0; |
494 |
} |
} |
495 |
|
|
527 |
return val; |
return val; |
528 |
} |
} |
529 |
|
|
|
/* make integer variable from property */ |
|
|
/* |
|
|
char *prop2int(SW_RESULT sw_res, char *propname) { |
|
|
char *val; |
|
|
unsigned long prop; |
|
|
int len; |
|
|
|
|
|
elog(DEBUG2, "prop2int(%s)", propname); |
|
|
|
|
|
prop = estResultPropertyULong( sw_res, propname ); |
|
|
if (error_or_abort( est_handle )) return NULL; |
|
|
|
|
|
elog(DEBUG1, "prop2int(%s) = %lu", propname, prop); |
|
|
|
|
|
len = 128 * sizeof(char); |
|
|
elog(DEBUG2, "palloc(%d)", len); |
|
|
|
|
|
val = palloc(len); |
|
|
memset(val, 0, len); |
|
|
|
|
|
snprintf(val, len, "%lu", prop); |
|
|
|
|
|
elog(DEBUG2, "val=%s", val); |
|
|
|
|
|
return val; |
|
|
} |
|
|
*/ |
|