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

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_test1.h"
  17. #include <my_getopt.h>
  18. static int key_field=FIELD_VARCHAR,extra_field=FIELD_SKIP_ENDSPACE;
  19. static uint key_length=200,extra_length=50;
  20. static int key_type=HA_KEYTYPE_TEXT;
  21. static int verbose=0,silent=0,skip_update=0,
  22.    no_keys=0,no_stopwords=0,no_search=0,no_fulltext=0;
  23. static int create_flag=0,error=0;
  24. #define MAX_REC_LENGTH 300
  25. static char record[MAX_REC_LENGTH],read_record[MAX_REC_LENGTH];
  26. static int run_test(const char *filename);
  27. static void get_options(int argc, char *argv[]);
  28. static void create_record(char *, int);
  29. static void usage();
  30. static struct my_option my_long_options[] =
  31. {
  32.   {"", 'v', "", 0, 0, 0, GET_NO_ARG, NO_ARG, 0, 0, 0, 0, 0, 0},
  33.   {"", '?', "", 0, 0, 0, GET_NO_ARG, NO_ARG, 0, 0, 0, 0, 0, 0},
  34.   {"", 'h', "", 0, 0, 0, GET_NO_ARG, NO_ARG, 0, 0, 0, 0, 0, 0},
  35.   {"", 'V', "", 0, 0, 0, GET_NO_ARG, NO_ARG, 0, 0, 0, 0, 0, 0},
  36.   {"", 'v', "", 0, 0, 0, GET_NO_ARG, NO_ARG, 0, 0, 0, 0, 0, 0},
  37.   {"", 's', "", 0, 0, 0, GET_NO_ARG, NO_ARG, 0, 0, 0, 0, 0, 0},
  38.   {"", 'N', "", 0, 0, 0, GET_NO_ARG, NO_ARG, 0, 0, 0, 0, 0, 0},
  39.   {"", 'S', "", 0, 0, 0, GET_NO_ARG, NO_ARG, 0, 0, 0, 0, 0, 0},
  40.   {"", 'K', "", 0, 0, 0, GET_NO_ARG, NO_ARG, 0, 0, 0, 0, 0, 0},
  41.   {"", 'F', "", 0, 0, 0, GET_NO_ARG, NO_ARG, 0, 0, 0, 0, 0, 0},
  42.   {"", 'U', "", 0, 0, 0, GET_NO_ARG, NO_ARG, 0, 0, 0, 0, 0, 0},
  43.   {"", '#', "", 0, 0, 0, GET_STR, OPT_ARG, 0, 0, 0, 0, 0, 0},
  44.   { 0, 0, 0, 0, 0, 0, GET_NO_ARG, NO_ARG, 0, 0, 0, 0, 0, 0}
  45. };
  46. int main(int argc, char *argv[])
  47. {
  48.   MY_INIT(argv[0]);
  49.   get_options(argc,argv);
  50.   exit(run_test("FT1"));
  51. }
  52. static MI_COLUMNDEF recinfo[3];
  53. static MI_KEYDEF keyinfo[2];
  54. static HA_KEYSEG keyseg[10];
  55. static int run_test(const char *filename)
  56. {
  57.   MI_INFO *file;
  58.   int i,j;
  59.   my_off_t pos;
  60.   bzero((char*) recinfo,sizeof(recinfo));
  61.   /* First define 2 columns */
  62.   recinfo[0].type=extra_field;
  63.   recinfo[0].length= (extra_field == FIELD_BLOB ? 4 + mi_portable_sizeof_char_ptr :
  64.       extra_length);
  65.   if (extra_field == FIELD_VARCHAR)
  66.     recinfo[0].length+=2;
  67.   recinfo[1].type=key_field;
  68.   recinfo[1].length= (key_field == FIELD_BLOB ? 4+mi_portable_sizeof_char_ptr :
  69.       key_length);
  70.   if (key_field == FIELD_VARCHAR)
  71.     recinfo[1].length+=2;
  72.   /* Define a key over the first column */
  73.   keyinfo[0].seg=keyseg;
  74.   keyinfo[0].keysegs=1;
  75.   keyinfo[0].seg[0].type= key_type;
  76.   keyinfo[0].seg[0].flag= (key_field == FIELD_BLOB)?HA_BLOB_PART:
  77.   (key_field == FIELD_VARCHAR)?HA_VAR_LENGTH:0;
  78.   keyinfo[0].seg[0].start=recinfo[0].length;
  79.   keyinfo[0].seg[0].length=key_length;
  80.   keyinfo[0].seg[0].null_bit= 0;
  81.   keyinfo[0].seg[0].null_pos=0;
  82.   keyinfo[0].seg[0].language=MY_CHARSET_CURRENT;
  83.   keyinfo[0].flag = (no_fulltext?HA_PACK_KEY:HA_FULLTEXT);
  84.   if (!silent)
  85.     printf("- Creating isam-filen");
  86.   if (mi_create(filename,(no_keys?0:1),keyinfo,2,recinfo,0,NULL,
  87. (MI_CREATE_INFO*) 0, create_flag))
  88.     goto err;
  89.   if (!(file=mi_open(filename,2,0)))
  90.     goto err;
  91.   if (!silent)
  92.     printf("- %s stopwordsn",no_stopwords?"Skipping":"Initializing");
  93.   ft_init_stopwords(no_stopwords?NULL:ft_precompiled_stopwords);
  94.   if (!silent)
  95.     printf("- Writing key:sn");
  96.   my_errno=0;
  97.   for (i=NUPD ; i<NDATAS; i++ )
  98.   {
  99.     create_record(record,i);
  100.     error=mi_write(file,record);
  101.     if (verbose || error)
  102.       printf("I= %2d  mi_write: %d  errno: %d, record: %sn",
  103. i,error,my_errno,data[i].f0);
  104.   }
  105.   if (!skip_update)
  106.   {
  107.     if (!silent)
  108.       printf("- Updating rowsn");
  109.     /* Read through all rows and update them */
  110.     pos=(ha_rows) 0;
  111.     i=0;
  112.     while ((error=mi_rrnd(file,read_record,pos)) == 0)
  113.     {
  114.       create_record(record,NUPD-i-1);
  115.       if (mi_update(file,read_record,record))
  116.       {
  117. printf("Can't update row: %.*s, error: %dn",
  118.        keyinfo[0].seg[0].length,record,my_errno);
  119.       }
  120.       if(++i == NUPD) break;
  121.       pos=HA_OFFSET_ERROR;
  122.     }
  123.     if (i != NUPD)
  124.       printf("Found %d of %d rowsn", i,NUPD);
  125.   }
  126.   if (mi_close(file)) goto err;
  127.   if(no_search) return 0;
  128.   if (!silent)
  129.     printf("- Reopening filen");
  130.   if (!(file=mi_open(filename,2,0))) goto err;
  131.   if (!silent)
  132.     printf("- Reading rows with keyn");
  133.   for (i=0 ; i < NQUERIES ; i++)
  134.   { FT_DOCLIST *result;
  135.     result=ft_nlq_init_search(file,0,(char*) query[i],strlen(query[i]),1);
  136.     if(!result) {
  137.       printf("Query %d: `%s' failed with errno %3dn",i,query[i],my_errno);
  138.       continue;
  139.     }
  140.     printf("Query %d: `%s'. Found: %d. Top five documents:n",
  141.     i,query[i],result->ndocs);
  142.     for(j=0;j<5;j++) { double w; int err;
  143. err=ft_nlq_read_next(result, read_record);
  144. if(err==HA_ERR_END_OF_FILE) {
  145.     printf("No more matches!n");
  146.     break;
  147. } else if (err) {
  148.     printf("ft_read_next %d failed with errno %3dn",j,my_errno);
  149.     break;
  150. }
  151.         w=ft_nlq_get_relevance(result);
  152. if(key_field == FIELD_VARCHAR) {
  153.     uint l;
  154.     char *p;
  155.     p=recinfo[0].length+read_record;
  156.     l=uint2korr(p);
  157.     printf("%10.7f: %.*sn",w,(int) l,p+2);
  158. } else
  159.     printf("%10.7f: %.*sn",w,recinfo[1].length,
  160.   recinfo[0].length+read_record);
  161.     }
  162.     ft_nlq_close_search(result);
  163.   }
  164.   if (mi_close(file)) goto err;
  165.   my_end(MY_CHECK_ERROR);
  166.   return (0);
  167. err:
  168.   printf("got error: %3d when using myisam-databasen",my_errno);
  169.   return 1; /* skip warning */
  170. }
  171. static char blob_key[MAX_REC_LENGTH];
  172. /* static char blob_record[MAX_REC_LENGTH+20*20]; */
  173. void create_record(char *pos, int n)
  174. {
  175.   bzero((char*) pos,MAX_REC_LENGTH);
  176.   if (recinfo[0].type == FIELD_BLOB)
  177.   {
  178.     uint tmp;
  179.     char *ptr;
  180.     strnmov(blob_key,data[n].f0,keyinfo[0].seg[0].length);
  181.     tmp=strlen(blob_key);
  182.     int4store(pos,tmp);
  183.     ptr=blob_key;
  184.     memcpy_fixed(pos+4,&ptr,sizeof(char*));
  185.     pos+=recinfo[0].length;
  186.   }
  187.   else if (recinfo[0].type == FIELD_VARCHAR)
  188.   {
  189.     uint tmp;
  190.     strnmov(pos+2,data[n].f0,keyinfo[0].seg[0].length);
  191.     tmp=strlen(pos+2);
  192.     int2store(pos,tmp);
  193.     pos+=recinfo[0].length;
  194.   }
  195.   else
  196.   {
  197.     strnmov(pos,data[n].f0,keyinfo[0].seg[0].length);
  198.     pos+=recinfo[0].length;
  199.   }
  200.   if (recinfo[1].type == FIELD_BLOB)
  201.   {
  202.     uint tmp;
  203.     char *ptr;
  204.     strnmov(blob_key,data[n].f2,keyinfo[0].seg[0].length);
  205.     tmp=strlen(blob_key);
  206.     int4store(pos,tmp);
  207.     ptr=blob_key;
  208.     memcpy_fixed(pos+4,&ptr,sizeof(char*));
  209.     pos+=recinfo[1].length;
  210.   }
  211.   else if (recinfo[1].type == FIELD_VARCHAR)
  212.   {
  213.     uint tmp;
  214.     strnmov(pos+2,data[n].f2,keyinfo[0].seg[0].length);
  215.     tmp=strlen(pos+2);
  216.     int2store(pos,tmp);
  217.     pos+=recinfo[1].length;
  218.   }
  219.   else
  220.   {
  221.     strnmov(pos,data[n].f2,keyinfo[0].seg[0].length);
  222.     pos+=recinfo[1].length;
  223.   }
  224. }
  225. static my_bool
  226. get_one_option(int optid, const struct my_option *opt __attribute__((unused)),
  227.        char *argument)
  228. {
  229.   switch(optid) {
  230.   case 'v': verbose=1; break;
  231.   case 's': silent=1; break;
  232.   case 'F': no_fulltext=1; no_search=1;
  233.   case 'U': skip_update=1; break;
  234.   case 'K': no_keys=no_search=1; break;
  235.   case 'N': no_search=1; break;
  236.   case 'S': no_stopwords=1; break;
  237.   case '#':
  238.     DEBUGGER_ON;
  239.     DBUG_PUSH (argument);
  240.     break;
  241.   case 'V':
  242.   case '?':
  243.   case 'h':
  244.     usage();
  245.     exit(1);
  246.   }
  247.   return 0;
  248. }
  249. /* Read options */
  250. static void get_options(int argc,char *argv[])
  251. {
  252.   int ho_error;
  253.   if ((ho_error=handle_options(&argc, &argv, my_long_options, get_one_option)))
  254.     exit(ho_error);
  255.   return;
  256. } /* get options */
  257. static void usage()
  258. {
  259.   printf("%s [options]n", my_progname);
  260.   my_print_help(my_long_options);
  261.   my_print_variables(my_long_options);
  262. }