rset/rsisamb.c

   1 /* $Id: rsisamb.c,v 1.34 2005-05-24 11:35:43 adam Exp $
   2    Copyright (C) 1995-2005
   3    Index Data ApS
   4
   5 This file is part of the Zebra server.
   6
   7 Zebra is free software; you can redistribute it and/or modify it under
   8 the terms of the GNU General Public License as published by the Free
   9 Software Foundation; either version 2, or (at your option) any later
  10 version.
  11
  12 Zebra is distributed in the hope that it will be useful, but WITHOUT ANY
  13 WARRANTY; without even the implied warranty of MERCHANTABILITY or
  14 FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
  15 for more details.
  16
  17 You should have received a copy of the GNU General Public License
  18 along with Zebra; see the file LICENSE.zebra.  If not, write to the
  19 Free Software Foundation, 59 Temple Place - Suite 330, Boston, MA
  20 02111-1307, USA.
  21 */
  22
  23 #include <stdio.h>
  24 #include <assert.h>
  25 #include <idzebra/util.h>
  26 #include <rset.h>
  27 #include <string.h>
  28
  29 static RSFD r_open(RSET ct, int flag);
  30 static void r_close(RSFD rfd);
  31 static void r_delete(RSET ct);
  32 static int r_forward(RSFD rfd, void *buf, TERMID *term, const void *untilbuf);
  33 static void r_pos(RSFD rfd, double *current, double *total);
  34 static int r_read(RSFD rfd, void *buf, TERMID *term);
  35 static int r_read_filter(RSFD rfd, void *buf, TERMID *term);
  36 static int r_write(RSFD rfd, const void *buf);
  37
  38 static const struct rset_control control =
  39 {
  40     "isamb",
  41     r_delete,
  42     rset_get_one_term,
  43     r_open,
  44     r_close,
  45     r_forward,
  46     r_pos,
  47     r_read,
  48     r_write,
  49 };
  50
  51 static const struct rset_control control_filter =
  52 {
  53     "isamb",
  54     r_delete,
  55     rset_get_one_term,
  56     r_open,
  57     r_close,
  58     r_forward,
  59     r_pos,
  60     r_read_filter,
  61     r_write,
  62 };
  63
  64 struct rfd_private {
  65     ISAMB_PP pt;
  66     void *buf;
  67 };
  68
  69 struct rset_private {
  70     ISAMB   is;
  71     ISAM_P pos;
  72 };
  73
  74 static int log_level = 0;
  75 static int log_level_initialized = 0;
  76
  77 RSET rsisamb_create(NMEM nmem, struct rset_key_control *kcontrol,
  78                     int scope,
  79                     ISAMB is, ISAM_P pos, TERMID term)
  80 {
  81     RSET rnew = rset_create_base(
  82         kcontrol->filter_func ? &control_filter : &control,
  83         nmem, kcontrol, scope, term, 0, 0);
  84     struct rset_private *info;
  85     if (!log_level_initialized)
  86     {
  87         log_level = yaz_log_module_level("rsisamb");
  88         log_level_initialized = 1;
  89     }
  90     info = (struct rset_private *) nmem_malloc(rnew->nmem, sizeof(*info));
  91     info->is = is;
  92     info->pos = pos;
  93     rnew->priv = info;
  94     yaz_log(log_level, "rsisamb_create");
  95     return rnew;
  96 }
  97
  98 static void r_delete(RSET ct)
  99 {
 100     yaz_log(log_level, "rsisamb_delete");
 101 }
 102
 103 RSFD r_open(RSET ct, int flag)
 104 {
 105     struct rset_private *info = (struct rset_private *) ct->priv;
 106     RSFD rfd;
 107     struct rfd_private *ptinfo;
 108
 109     if (flag & RSETF_WRITE)
 110     {
 111         yaz_log(YLOG_FATAL, "ISAMB set type is read-only");
 112         return NULL;
 113     }
 114     rfd = rfd_create_base(ct);
 115     if (rfd->priv)
 116         ptinfo = (struct rfd_private *) (rfd->priv);
 117     else {
 118         ptinfo = (struct rfd_private *) nmem_malloc(ct->nmem,sizeof(*ptinfo));
 119         ptinfo->buf = nmem_malloc(ct->nmem,ct->keycontrol->key_size);
 120         rfd->priv = ptinfo;
 121     }
 122     ptinfo->pt = isamb_pp_open(info->is, info->pos, ct->scope );
 123     yaz_log(log_level, "rsisamb_open");
 124     return rfd;
 125 }
 126
 127 static void r_close(RSFD rfd)
 128 {
 129     struct rfd_private *ptinfo = (struct rfd_private *)(rfd->priv);
 130     isamb_pp_close (ptinfo->pt);
 131     yaz_log(log_level, "rsisamb_close");
 132 }
 133
 134
 135 static int r_forward(RSFD rfd, void *buf, TERMID *term, const void *untilbuf)
 136 {
 137     struct rfd_private *pinfo = (struct rfd_private *)(rfd->priv);
 138     int rc;
 139     rc = isamb_pp_forward(pinfo->pt, buf, untilbuf);
 140     if (rc && term)
 141         *term = rfd->rset->term;
 142     yaz_log(log_level, "rsisamb_forward");
 143     return rc;
 144 }
 145
 146 static void r_pos(RSFD rfd, double *current, double *total)
 147 {
 148     struct rfd_private *pinfo = (struct rfd_private *)(rfd->priv);
 149     assert(rfd);
 150     isamb_pp_pos(pinfo->pt, current, total);
 151     yaz_log(log_level, "isamb.r_pos returning %0.1f/%0.1f",
 152               *current, *total);
 153 }
 154
 155 static int r_read(RSFD rfd, void *buf, TERMID *term)
 156 {
 157     struct rfd_private *pinfo = (struct rfd_private *)(rfd->priv);
 158     int rc;
 159     rc = isamb_pp_read(pinfo->pt, buf);
 160     if (rc && term)
 161         *term = rfd->rset->term;
 162     yaz_log(log_level, "isamb.r_read ");
 163     return rc;
 164 }
 165
 166 static int r_read_filter(RSFD rfd, void *buf, TERMID *term)
 167 {
 168     struct rfd_private *pinfo = (struct rfd_private *)rfd->priv;
 169     const struct rset_key_control *kctrl = rfd->rset->keycontrol;
 170     int rc;
 171     while((rc = isamb_pp_read(pinfo->pt, buf)))
 172     {
 173         int incl = (*kctrl->filter_func)(buf, kctrl->filter_data);
 174         if (incl)
 175             break;
 176     }
 177     if (rc && term)
 178         *term = rfd->rset->term;
 179     yaz_log(log_level, "isamb.r_read_filter");
 180     return rc;
 181 }
 182
 183 static int r_write(RSFD rfd, const void *buf)
 184 {
 185     yaz_log(YLOG_FATAL, "ISAMB set type is read-only");
 186     return -1;
 187 }