/[pgestraier]/trunk/pgest.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/pgest.c

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

revision 2 by dpavlin, Fri May 20 13:00:46 2005 UTC revision 7 by dpavlin, Fri May 20 22:01:19 2005 UTC
# Line 34  Line 34 
34  #define GET_STR(textp) DatumGetCString(DirectFunctionCall1(textout, PointerGetDatum(textp)))  #define GET_STR(textp) DatumGetCString(DirectFunctionCall1(textout, PointerGetDatum(textp)))
35  #define GET_TEXT(cstrp) DatumGetTextP(DirectFunctionCall1(textin, CStringGetDatum(cstrp)))  #define GET_TEXT(cstrp) DatumGetTextP(DirectFunctionCall1(textin, CStringGetDatum(cstrp)))
36    
37    /* prototype */
38    char *attr2text(ESTDOC *doc, char *attr);
39    
40  ESTDB *db;  ESTDB *db;
41  ESTCOND *cond;  ESTCOND *cond;
42  ESTDOC *doc;  ESTDOC *doc;
43  const CBLIST *texts;  const CBLIST *texts;
44  int ecode, *est_result, resnum, i, j;  int ecode, *est_result, resnum, i, j;
45    int limit = 0;
46    int offset = 0;
47    
48  /* define PostgreSQL v1 function */  /* define PostgreSQL v1 function */
49  PG_FUNCTION_INFO_V1(pgest);  PG_FUNCTION_INFO_V1(pgest);
# Line 53  Datum pgest(PG_FUNCTION_ARGS) { Line 57  Datum pgest(PG_FUNCTION_ARGS) {
57          AttInMetadata   *attinmeta;          AttInMetadata   *attinmeta;
58          char            *index_path;          char            *index_path;
59          char            *query;          char            *query;
60            char            *attr;
61    
62          /* stuff done only on the first call of the function */          /* stuff done only on the first call of the function */
63          if (SRF_IS_FIRSTCALL()) {          if (SRF_IS_FIRSTCALL()) {
# Line 62  Datum pgest(PG_FUNCTION_ARGS) { Line 67  Datum pgest(PG_FUNCTION_ARGS) {
67                  //index_path = _textout(PG_GETARG_TEXT_P(0));                  //index_path = _textout(PG_GETARG_TEXT_P(0));
68                  index_path = _textout(PG_GETARG_TEXT_P(0));                  index_path = _textout(PG_GETARG_TEXT_P(0));
69                  query = _textout(PG_GETARG_TEXT_P(1));                  query = _textout(PG_GETARG_TEXT_P(1));
70                    attr = _textout(PG_GETARG_TEXT_P(2));
71                    limit = PG_GETARG_INT32(3);
72                    offset = PG_GETARG_INT32(4);
73    
74                  /* create a function context for cross-call persistence */                  /* create a function context for cross-call persistence */
75                  funcctx = SRF_FIRSTCALL_INIT();                  funcctx = SRF_FIRSTCALL_INIT();
# Line 77  Datum pgest(PG_FUNCTION_ARGS) { Line 85  Datum pgest(PG_FUNCTION_ARGS) {
85                          SRF_RETURN_DONE(funcctx);                          SRF_RETURN_DONE(funcctx);
86                  }                  }
87                                    
88                  elog(DEBUG2, "pgest: query=%s", query);                  elog(INFO, "pgest: query[%s] attr[%s] limit %d offset %d", query, attr, limit, offset);
89                                    
90                  /* create a search condition object */                  /* create a search condition object */
91                  if (!(cond = est_cond_new())) {                  if (!(cond = est_cond_new())) {
# Line 88  Datum pgest(PG_FUNCTION_ARGS) { Line 96  Datum pgest(PG_FUNCTION_ARGS) {
96                  /* set the search phrase to the search condition object */                  /* set the search phrase to the search condition object */
97                  est_cond_set_phrase(cond, query);                  est_cond_set_phrase(cond, query);
98    
99                    /* minimum valid attribute length is 10: @a STREQ a */
100                    if (attr != NULL && strlen(attr) >= 10) {
101                            elog(INFO,"est_cond_add_attr(%s)", attr);
102                            est_cond_add_attr(cond, attr);
103                    }
104    
105                  /* get the result of search */                  /* get the result of search */
106                  est_result = est_db_search(db, cond, &resnum, NULL);                  est_result = est_db_search(db, cond, &resnum, NULL);
107                                    
108                  /* total number of tuples to be returned */                  /* total number of tuples to be returned */
109                  funcctx->max_calls = resnum;                  if (limit && limit < resnum) {
110                            funcctx->max_calls = limit - offset;
111                    } else {
112                            funcctx->max_calls = resnum - offset;
113                    }
114    
115                  /* check if results exists */                  /* check if results exists */
116                  if ( 0 == funcctx->max_calls )                  if ( 0 == funcctx->max_calls )
117                          elog(INFO, "pgest: no results for: %s", query );                          elog(INFO, "pgest: no results for: %s", query );
118    
119                  elog(DEBUG1, "pgest: found %d hits for %s", funcctx->max_calls, query);                  elog(DEBUG1, "pgest: found %d hits for %s", resnum, query);
120    
121                  /* Build a tuple description for a __pgest tuple */                  /* Build a tuple description for a __pgest tuple */
122                  tupdesc = RelationNameGetTupleDesc("__pgest");                  tupdesc = RelationNameGetTupleDesc("__pgest");
# Line 128  Datum pgest(PG_FUNCTION_ARGS) { Line 146  Datum pgest(PG_FUNCTION_ARGS) {
146          max_calls = funcctx->max_calls;          max_calls = funcctx->max_calls;
147          slot = funcctx->slot;          slot = funcctx->slot;
148          attinmeta = funcctx->attinmeta;          attinmeta = funcctx->attinmeta;
149    
150            if (limit && call_cntr > limit - 1) {
151                    elog(INFO, "call_cntr: %d limit: %d", call_cntr, limit);
152                    SRF_RETURN_DONE(funcctx);
153            }
154    
155          if (call_cntr < max_calls) {          if (call_cntr < max_calls) {
156                  char            **values;                  char            **values;
157                  HeapTuple       tuple;                  HeapTuple       tuple;
# Line 147  Datum pgest(PG_FUNCTION_ARGS) { Line 170  Datum pgest(PG_FUNCTION_ARGS) {
170                   * be processed later by the type input functions.                   * be processed later by the type input functions.
171                   */                   */
172    
173                  if (doc = est_db_get_doc(db, est_result[call_cntr], 0)) {                  if (doc = est_db_get_doc(db, est_result[call_cntr + offset], 0)) {
174                                    
175                          elog(DEBUG1, "URI: %s\n Title: %s\n",                          elog(DEBUG1, "URI: %s\n Title: %s\n",
176                                  est_doc_attr(doc, "@uri"),                                  est_doc_attr(doc, "@uri"),
# Line 158  Datum pgest(PG_FUNCTION_ARGS) { Line 181  Datum pgest(PG_FUNCTION_ARGS) {
181    
182  //                      values[0] = (char *) palloc(strlen(_estval) * sizeof(char));  //                      values[0] = (char *) palloc(strlen(_estval) * sizeof(char));
183    
184                          values[0] = attr2text(doc,"@id");                          values[0] = (char *) attr2text(doc,"@id");
185                          values[1] = attr2text(doc,"@uri");                          values[1] = (char *) attr2text(doc,"@uri");
186                          values[2] = attr2text(doc,"@title");                          values[2] = (char *) attr2text(doc,"@title");
187                          values[3] = attr2text(doc,"@type");                          values[3] = (char *) attr2text(doc,"@type");
188    
189                          /* destloy the document object */                          /* destloy the document object */
190                          elog(DEBUG2, "est_doc_delete");                          elog(DEBUG2, "est_doc_delete");
191                          est_doc_delete(doc);                          est_doc_delete(doc);
192                  } else {                  } else {
193                          elog(INFO, "no result from estraier");                          elog(INFO, "no result from estraier");
194                          values[0] = NULL;                          values[0] = DatumGetCString( "" );
195                          values[1] = NULL;                          values[1] = DatumGetCString( "" );
196                          values[2] = NULL;                          values[2] = DatumGetCString( "" );
197                          values[3] = NULL;                          values[3] = DatumGetCString( "" );
198                  }                  }
199    
200    
# Line 292  char *attr2text(ESTDOC *doc, char *attr) Line 315  char *attr2text(ESTDOC *doc, char *attr)
315          char *val;          char *val;
316          const char *attrval;          const char *attrval;
317          int len;          int len;
318            int attrlen;
319    
320          elog(DEBUG1, "doc: %08x, attr: %s", doc, attr);          elog(DEBUG1, "doc: %08x, attr: %s", doc, attr);
321    
322          if (attrval = est_doc_attr(doc, attr)) {          if ( (attrval = est_doc_attr(doc, attr)) && (attrlen = strlen(attrval)) ) {
323                  val = (char *) palloc(strlen(attrval) * sizeof(char));                  val = (char *) palloc(attrlen * sizeof(char));
324          } else {          } else {
325                  return (Datum) NULL;                  return (Datum) NULL;
326          }          }
Legend:
Removed from v.2  
changed lines
  Added in v.7
  ViewVC Help
Powered by ViewVC 1.1.26