ft_eval.c
上传用户:romrleung
上传日期:2022-05-23
资源大小:18897k
文件大小:7k
源码类别:

MySQL数据库

开发平台:

Visual C++

  1. /* Copyright (C) 2000 MySQL AB & MySQL Finland AB & TCX DataKonsult AB
  2.    This program is free software; you can redistribute it and/or modify
  3.    it under the terms of the GNU General Public License as published by
  4.    the Free Software Foundation; either version 2 of the License, or
  5.    (at your option) any later version.
  6.    This program is distributed in the hope that it will be useful,
  7.    but WITHOUT ANY WARRANTY; without even the implied warranty of
  8.    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  9.    GNU General Public License for more details.
  10.    You should have received a copy of the GNU General Public License
  11.    along with this program; if not, write to the Free Software
  12.    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA */
  13. /* Written by Sergei A. Golubchik, who has a shared copyright to this code
  14.    added support for long options (my_getopt) 22.5.2002 by Jani Tolonen */
  15. #include "ftdefs.h"
  16. #include "ft_eval.h"
  17. #include <stdarg.h>
  18. #include <my_getopt.h>
  19. static void print_error(int exit_code, const char *fmt,...);
  20. static void get_options(int argc, char *argv[]);
  21. static int create_record(char *pos, FILE *file);
  22. static void usage();
  23. static struct my_option my_long_options[] =
  24. {
  25.   {"", 's', "", 0, 0, 0, GET_STR, REQUIRED_ARG, 0, 0, 0, 0, 0, 0},
  26.   {"", 'q', "", 0, 0, 0, GET_NO_ARG, NO_ARG, 0, 0, 0, 0, 0, 0},
  27.   {"", 'S', "", 0, 0, 0, GET_NO_ARG, NO_ARG, 0, 0, 0, 0, 0, 0},
  28.   {"", '#', "", 0, 0, 0, GET_STR, REQUIRED_ARG, 0, 0, 0, 0, 0, 0},
  29.   {"", 'V', "", 0, 0, 0, GET_NO_ARG, NO_ARG, 0, 0, 0, 0, 0, 0},
  30.   {"", '?', "", 0, 0, 0, GET_NO_ARG, NO_ARG, 0, 0, 0, 0, 0, 0},
  31.   {"", 'h', "", 0, 0, 0, GET_NO_ARG, NO_ARG, 0, 0, 0, 0, 0, 0},
  32.   { 0, 0, 0, 0, 0, 0, GET_NO_ARG, NO_ARG, 0, 0, 0, 0, 0, 0}
  33. };
  34. int main(int argc, char *argv[])
  35. {
  36.   MI_INFO *file;
  37.   int i,j;
  38.   MY_INIT(argv[0]);
  39.   get_options(argc,argv);
  40.   bzero((char*)recinfo,sizeof(recinfo));
  41.   /* First define 2 columns */
  42.   recinfo[0].type=FIELD_SKIP_ENDSPACE;
  43.   recinfo[0].length=docid_length;
  44.   recinfo[1].type=FIELD_BLOB;
  45.   recinfo[1].length= 4+mi_portable_sizeof_char_ptr;
  46.   /* Define a key over the first column */
  47.   keyinfo[0].seg=keyseg;
  48.   keyinfo[0].keysegs=1;
  49.   keyinfo[0].seg[0].type= HA_KEYTYPE_TEXT;
  50.   keyinfo[0].seg[0].flag= HA_BLOB_PART;
  51.   keyinfo[0].seg[0].start=recinfo[0].length;
  52.   keyinfo[0].seg[0].length=key_length;
  53.   keyinfo[0].seg[0].null_bit=0;
  54.   keyinfo[0].seg[0].null_pos=0;
  55.   keyinfo[0].seg[0].bit_start=4;
  56.   keyinfo[0].seg[0].language=MY_CHARSET_CURRENT;
  57.   keyinfo[0].flag = HA_FULLTEXT;
  58.   if (!silent)
  59.     printf("- Creating isam-filen");
  60.   if (mi_create(filename,1,keyinfo,2,recinfo,0,NULL,(MI_CREATE_INFO*) 0,0))
  61.     goto err;
  62.   if (!(file=mi_open(filename,2,0)))
  63.     goto err;
  64.   if (!silent)
  65.     printf("Initializing stopwordsn");
  66.   ft_init_stopwords(stopwordlist);
  67.   if (!silent)
  68.     printf("- Writing key:sn");
  69.   my_errno=0;
  70.   i=0;
  71.   while (create_record(record,df))
  72.   {
  73.     error=mi_write(file,record);
  74.     if (error)
  75.       printf("I= %2d  mi_write: %d  errno: %dn",i,error,my_errno);
  76.     i++;
  77.   }
  78.   fclose(df);
  79.   if (mi_close(file)) goto err;
  80.   if (!silent)
  81.     printf("- Reopening filen");
  82.   if (!(file=mi_open(filename,2,0))) goto err;
  83.   if (!silent)
  84.     printf("- Reading rows with keyn");
  85.   for (i=1;create_record(record,qf);i++)
  86.   {
  87.     FT_DOCLIST *result;
  88.     double w;
  89.     int t, err;
  90.     result=ft_nlq_init_search(file,0,blob_record,(uint) strlen(blob_record),1);
  91.     if (!result)
  92.     {
  93.       printf("Query %d failed with errno %3dn",i,my_errno);
  94.       goto err;
  95.     }
  96.     if (!silent)
  97.       printf("Query %d. Found: %d.n",i,result->ndocs);
  98.     for (j=0;(err=ft_nlq_read_next(result, read_record))==0;j++)
  99.     {
  100.       t=uint2korr(read_record);
  101.       w=ft_nlq_get_relevance(result);
  102.       printf("%d %.*s %fn",i,t,read_record+2,w);
  103.     }
  104.     if (err != HA_ERR_END_OF_FILE)
  105.     {
  106.       printf("ft_read_next %d failed with errno %3dn",j,my_errno);
  107.       goto err;
  108.     }
  109.     ft_nlq_close_search(result);
  110.   }
  111.   if (mi_close(file)) goto err;
  112.   my_end(MY_CHECK_ERROR);
  113.   return (0);
  114.  err:
  115.   printf("got error: %3d when using myisam-databasen",my_errno);
  116.   return 1; /* skip warning */
  117. }
  118. static my_bool
  119. get_one_option(int optid, const struct my_option *opt __attribute__((unused)),
  120.        char *argument)
  121. {
  122.   switch (optid) {
  123.   case 's':
  124.     if (stopwordlist && stopwordlist != ft_precompiled_stopwords)
  125.       break;
  126.     {
  127.       FILE *f; char s[HA_FT_MAXLEN]; int i=0,n=SWL_INIT;
  128.       if (!(stopwordlist=(const char**) malloc(n*sizeof(char *))))
  129. print_error(1,"malloc(%d)",n*sizeof(char *));
  130.       if (!(f=fopen(argument,"r")))
  131. print_error(1,"fopen(%s)",argument);
  132.       while (!feof(f))
  133.       {
  134. if (!(fgets(s,HA_FT_MAXLEN,f)))
  135.   print_error(1,"fgets(s,%d,%s)",HA_FT_MAXLEN,argument);
  136. if (!(stopwordlist[i++]=strdup(s)))
  137.   print_error(1,"strdup(%s)",s);
  138. if (i >= n)
  139. {
  140.   n+=SWL_PLUS;
  141.   if (!(stopwordlist=(const char**) realloc((char*) stopwordlist,
  142.     n*sizeof(char *))))
  143.     print_error(1,"realloc(%d)",n*sizeof(char *));
  144. }
  145.       }
  146.       fclose(f);
  147.       stopwordlist[i]=NULL;
  148.       break;
  149.     }
  150.   case 'q': silent=1; break;
  151.   case 'S': if (stopwordlist==ft_precompiled_stopwords) stopwordlist=NULL; break;
  152.   case '#':
  153.     DEBUGGER_ON;
  154.     DBUG_PUSH (argument);
  155.     break;
  156.   case 'V':
  157.   case '?':
  158.   case 'h':
  159.     usage();
  160.     exit(1);
  161.   }
  162.   return 0;
  163. }
  164. static void get_options(int argc, char *argv[])
  165. {
  166.   int ho_error;
  167.   if ((ho_error=handle_options(&argc, &argv, my_long_options, get_one_option)))
  168.     exit(ho_error);
  169.   if (!(d_file=argv[optind])) print_error(1,"No d_file");
  170.   if (!(df=fopen(d_file,"r")))
  171.     print_error(1,"fopen(%s)",d_file);
  172.   if (!(q_file=argv[optind+1])) print_error(1,"No q_file");
  173.   if (!(qf=fopen(q_file,"r")))
  174.     print_error(1,"fopen(%s)",q_file);
  175.   return;
  176. } /* get options */
  177. static int create_record(char *pos, FILE *file)
  178. {
  179.   uint tmp; char *ptr;
  180.   bzero((char *)pos,MAX_REC_LENGTH);
  181.   /* column 1 - VARCHAR */
  182.   if (!(fgets(pos+2,MAX_REC_LENGTH-32,file)))
  183.   {
  184.     if (feof(file))
  185.       return 0;
  186.     else
  187.       print_error(1,"fgets(docid) - 1");
  188.   }
  189.   tmp=(uint) strlen(pos+2)-1;
  190.   int2store(pos,tmp);
  191.   pos+=recinfo[0].length;
  192.   /* column 2 - BLOB */
  193.   if (!(fgets(blob_record,MAX_BLOB_LENGTH,file)))
  194.     print_error(1,"fgets(docid) - 2");
  195.   tmp=(uint) strlen(blob_record);
  196.   int4store(pos,tmp);
  197.   ptr=blob_record;
  198.   memcpy_fixed(pos+4,&ptr,sizeof(char*));
  199.   return 1;
  200. }
  201. /* VARARGS */
  202. static void print_error(int exit_code, const char *fmt,...)
  203. {
  204.   va_list args;
  205.   va_start(args,fmt);
  206.   fprintf(stderr,"%s: error: ",my_progname);
  207.   VOID(vfprintf(stderr, fmt, args));
  208.   VOID(fputc('n',stderr));
  209.   fflush(stderr);
  210.   va_end(args);
  211.   exit(exit_code);
  212. }
  213. static void usage()
  214. {
  215.   printf("%s [options]n", my_progname);
  216.   my_print_help(my_long_options);
  217.   my_print_variables(my_long_options);
  218. }