X-Git-Url: http://git.indexdata.com/?p=yazpp-moved-to-github.git;a=blobdiff_plain;f=src%2Fyaz-z-server-sr.cpp;h=17158107cdac885a7ae0fde128e959a2f721fe06;hp=396757f2bd358289c13031ccfe6332a635964e1e;hb=a0042915b20e2a1d15efa254c35e22cf09e0796a;hpb=4c758111f1ff5a815888f70a4c4335ab149f5608 diff --git a/src/yaz-z-server-sr.cpp b/src/yaz-z-server-sr.cpp index 396757f..1715810 100644 --- a/src/yaz-z-server-sr.cpp +++ b/src/yaz-z-server-sr.cpp @@ -1,28 +1,27 @@ -/* - * Copyright (c) 2000-2001, Index Data. +/* This file is part of the yazpp toolkit. + * Copyright (C) 1998-2008 Index Data and Mike Taylor * See the file LICENSE for details. - * - * $Id: yaz-z-server-sr.cpp,v 1.7 2004-11-30 21:10:31 adam Exp $ - * */ -#include -#include +#include +#include + +using namespace yazpp_1; -Z_Records *Yaz_Facility_Retrieval::pack_records (Yaz_Z_Server *s, - const char *resultSetName, - int start, int xnum, - Z_RecordComposition *comp, - int *next, int *pres, - int *format) +Z_Records *Yaz_Facility_Retrieval::pack_records (Z_Server *s, + const char *resultSetName, + int start, int xnum, + Z_RecordComposition *comp, + int *next, int *pres, + Odr_oid *format) { int recno, total_length = 0, toget = xnum, dumped_records = 0; Z_Records *records = - (Z_Records *) odr_malloc (odr_encode(), sizeof(*records)); + (Z_Records *) odr_malloc (odr_encode(), sizeof(*records)); Z_NamePlusRecordList *reclist = - (Z_NamePlusRecordList *) odr_malloc (odr_encode(), sizeof(*reclist)); + (Z_NamePlusRecordList *) odr_malloc (odr_encode(), sizeof(*reclist)); Z_NamePlusRecord **list = - (Z_NamePlusRecord **) odr_malloc (odr_encode(), sizeof(*list) * toget); + (Z_NamePlusRecord **) odr_malloc (odr_encode(), sizeof(*list) * toget); records->which = Z_Records_DBOSD; records->u.databaseOrSurDiagnostics = reclist; @@ -33,79 +32,79 @@ Z_Records *Yaz_Facility_Retrieval::pack_records (Yaz_Z_Server *s, for (recno = start; reclist->num_records < toget; recno++) { - Z_NamePlusRecord *this_rec = - (Z_NamePlusRecord *) odr_malloc (odr_encode(), sizeof(*this_rec)); - this_rec->databaseName = 0; - this_rec->which = Z_NamePlusRecord_databaseRecord; - this_rec->u.databaseRecord = 0; - - int this_length = 0; - - sr_record (resultSetName, recno, format, comp, this_rec, records); - - if (records->which != Z_Records_DBOSD) - { - *pres = Z_PRES_FAILURE; - return records; - } - - if (this_rec->which == Z_NamePlusRecord_databaseRecord && - this_rec->u.databaseRecord == 0) - { // handler did not return a record.. - create_surrogateDiagnostics(odr_encode(), this_rec, 0, 14, 0); - } - /* - * we get the number of bytes allocated on the stream before any - * allocation done by the backend - this should give us a reasonable - * idea of the total size of the data so far. - */ - total_length = odr_total(odr_encode()) - dumped_records; - this_length = odr_total(odr_encode()) - total_length; - if (this_length + total_length > m_preferredMessageSize) - { - /* record is small enough, really */ - if (this_length <= m_preferredMessageSize) - { - *pres = Z_PRES_PARTIAL_2; - break; - } - if (this_length >= m_maximumRecordSize) - { /* too big entirely */ - reclist->records[reclist->num_records] = this_rec; - create_surrogateDiagnostics(odr_encode(), this_rec, - this_rec->databaseName, 17, 0); - reclist->num_records++; - *next = recno + 1; - dumped_records += this_length; - continue; - } - else /* record can only be fetched by itself */ - { - if (toget > 1) - { - yaz_log(YLOG_DEBUG, " Dropped it"); - reclist->records[reclist->num_records] = this_rec; - create_surrogateDiagnostics(odr_encode(), this_rec, - this_rec->databaseName, - 16, 0); - reclist->num_records++; - // *next = freq.last_in_set ? 0 : recno + 1; - *next = recno + 1; - dumped_records += this_length; - continue; - } - } - } - reclist->records[reclist->num_records] = this_rec; - reclist->num_records++; - *next = recno + 1; + Z_NamePlusRecord *this_rec = + (Z_NamePlusRecord *) odr_malloc (odr_encode(), sizeof(*this_rec)); + this_rec->databaseName = 0; + this_rec->which = Z_NamePlusRecord_databaseRecord; + this_rec->u.databaseRecord = 0; + + int this_length = 0; + + sr_record (resultSetName, recno, format, comp, this_rec, records); + + if (records->which != Z_Records_DBOSD) + { + *pres = Z_PRES_FAILURE; + return records; + } + + if (this_rec->which == Z_NamePlusRecord_databaseRecord && + this_rec->u.databaseRecord == 0) + { // handler did not return a record.. + create_surrogateDiagnostics(odr_encode(), this_rec, 0, 14, 0); + } + /* + * we get the number of bytes allocated on the stream before any + * allocation done by the backend - this should give us a reasonable + * idea of the total size of the data so far. + */ + total_length = odr_total(odr_encode()) - dumped_records; + this_length = odr_total(odr_encode()) - total_length; + if (this_length + total_length > m_preferredMessageSize) + { + /* record is small enough, really */ + if (this_length <= m_preferredMessageSize) + { + *pres = Z_PRES_PARTIAL_2; + break; + } + if (this_length >= m_maximumRecordSize) + { /* too big entirely */ + reclist->records[reclist->num_records] = this_rec; + create_surrogateDiagnostics(odr_encode(), this_rec, + this_rec->databaseName, 17, 0); + reclist->num_records++; + *next = recno + 1; + dumped_records += this_length; + continue; + } + else /* record can only be fetched by itself */ + { + if (toget > 1) + { + yaz_log(YLOG_DEBUG, " Dropped it"); + reclist->records[reclist->num_records] = this_rec; + create_surrogateDiagnostics(odr_encode(), this_rec, + this_rec->databaseName, + 16, 0); + reclist->num_records++; + // *next = freq.last_in_set ? 0 : recno + 1; + *next = recno + 1; + dumped_records += this_length; + continue; + } + } + } + reclist->records[reclist->num_records] = this_rec; + reclist->num_records++; + *next = recno + 1; } return records; } -void Yaz_Facility_Retrieval::fetch_via_piggyback (Yaz_Z_Server *s, - Z_SearchRequest *req, - Z_SearchResponse *res) +void Yaz_Facility_Retrieval::fetch_via_piggyback (Z_Server *s, + Z_SearchRequest *req, + Z_SearchResponse *res) { bool_t *sr = (bool_t *)odr_malloc (odr_encode(), sizeof(*sr)); *sr = 1; @@ -122,73 +121,73 @@ void Yaz_Facility_Retrieval::fetch_via_piggyback (Yaz_Z_Server *s, /* how many records does the user agent want, then? */ if (hits <= *req->smallSetUpperBound) { - toget = hits; - if ((comp.u.simple = req->smallSetElementSetNames)) - compp = ∁ + toget = hits; + if ((comp.u.simple = req->smallSetElementSetNames)) + compp = ∁ } else if (hits < *req->largeSetLowerBound) { - toget = *req->mediumSetPresentNumber; - if (toget > hits) - toget = hits; - if ((comp.u.simple = req->mediumSetElementSetNames)) - compp = ∁ + toget = *req->mediumSetPresentNumber; + if (toget > hits) + toget = hits; + if ((comp.u.simple = req->mediumSetElementSetNames)) + compp = ∁ } if (toget && !res->records) { - res->presentStatus = (int *) odr_malloc (odr_encode(), sizeof(int)); - *res->presentStatus = Z_PRES_SUCCESS; - res->records = - pack_records(s, req->resultSetName, 1, toget, compp, - res->nextResultSetPosition, - res->presentStatus, - req->preferredRecordSyntax); - if (!res->records) - return; - if (res->records->which == Z_Records_DBOSD) - *res->numberOfRecordsReturned = - res->records->u.databaseOrSurDiagnostics->num_records; - res->searchStatus = sr; - res->resultSetStatus = 0; + res->presentStatus = (int *) odr_malloc (odr_encode(), sizeof(int)); + *res->presentStatus = Z_PRES_SUCCESS; + res->records = + pack_records(s, req->resultSetName, 1, toget, compp, + res->nextResultSetPosition, + res->presentStatus, + req->preferredRecordSyntax); + if (!res->records) + return; + if (res->records->which == Z_Records_DBOSD) + *res->numberOfRecordsReturned = + res->records->u.databaseOrSurDiagnostics->num_records; + res->searchStatus = sr; + res->resultSetStatus = 0; } else { - if (hits) - *res->nextResultSetPosition = 1; - res->numberOfRecordsReturned = nulint; - res->searchStatus = sr; - res->resultSetStatus = 0; - res->presentStatus = 0; + if (hits) + *res->nextResultSetPosition = 1; + res->numberOfRecordsReturned = nulint; + res->searchStatus = sr; + res->resultSetStatus = 0; + res->presentStatus = 0; } } -void Yaz_Facility_Retrieval::fetch_via_present (Yaz_Z_Server *s, - Z_PresentRequest *req, - Z_PresentResponse *res) +void Yaz_Facility_Retrieval::fetch_via_present (Z_Server *s, + Z_PresentRequest *req, + Z_PresentResponse *res) { res->records = - pack_records (s, req->resultSetId,*req->resultSetStartPoint, - *req->numberOfRecordsRequested, - req->recordComposition, - res->nextResultSetPosition, - res->presentStatus, - req->preferredRecordSyntax); + pack_records (s, req->resultSetId,*req->resultSetStartPoint, + *req->numberOfRecordsRequested, + req->recordComposition, + res->nextResultSetPosition, + res->presentStatus, + req->preferredRecordSyntax); if (res->records->which == Z_Records_DBOSD) - *res->numberOfRecordsReturned = - res->records->u.databaseOrSurDiagnostics->num_records; + *res->numberOfRecordsReturned = + res->records->u.databaseOrSurDiagnostics->num_records; } -int Yaz_Facility_Retrieval::init(Yaz_Z_Server *s, Z_InitRequest *initRequest, - Z_InitResponse *initResponse) +int Yaz_Facility_Retrieval::init(Z_Server *s, Z_InitRequest *initRequest, + Z_InitResponse *initResponse) { Z_Options *req = initRequest->options; Z_Options *res = initResponse->options; if (ODR_MASK_GET(req, Z_Options_search)) - ODR_MASK_SET(res, Z_Options_search); + ODR_MASK_SET(res, Z_Options_search); if (ODR_MASK_GET(req, Z_Options_present)) - ODR_MASK_SET(res, Z_Options_present); + ODR_MASK_SET(res, Z_Options_present); m_preferredMessageSize = *initRequest->preferredMessageSize; m_maximumRecordSize = *initRequest->maximumRecordSize; return sr_init (initRequest, initResponse); @@ -204,7 +203,7 @@ ODR Yaz_Facility_Retrieval::odr_decode() return m_odr_decode; } -int Yaz_Facility_Retrieval::recv(Yaz_Z_Server *s, Z_APDU *apdu_request) +int Yaz_Facility_Retrieval::recv(Z_Server *s, Z_APDU *apdu_request) { Z_APDU *apdu_response; m_odr_encode = s->odr_encode(); @@ -212,27 +211,35 @@ int Yaz_Facility_Retrieval::recv(Yaz_Z_Server *s, Z_APDU *apdu_request) switch (apdu_request->which) { case Z_APDU_searchRequest: - apdu_response = s->create_Z_PDU(Z_APDU_searchResponse); - s->transfer_referenceId(apdu_request, apdu_response); - sr_search (apdu_request->u.searchRequest, - apdu_response->u.searchResponse); - if (!apdu_response->u.searchResponse->records) - { - fetch_via_piggyback(s, apdu_request->u.searchRequest, - apdu_response->u.searchResponse); - } - s->send_Z_PDU(apdu_response, 0); - return 1; + apdu_response = s->create_Z_PDU(Z_APDU_searchResponse); + s->transfer_referenceId(apdu_request, apdu_response); + sr_search (apdu_request->u.searchRequest, + apdu_response->u.searchResponse); + if (!apdu_response->u.searchResponse->records) + { + fetch_via_piggyback(s, apdu_request->u.searchRequest, + apdu_response->u.searchResponse); + } + s->send_Z_PDU(apdu_response, 0); + return 1; case Z_APDU_presentRequest: - apdu_response = s->create_Z_PDU(Z_APDU_presentResponse); - s->transfer_referenceId(apdu_request, apdu_response); - sr_present (apdu_request->u.presentRequest, - apdu_response->u.presentResponse); - if (!apdu_response->u.presentResponse->records) - fetch_via_present(s, apdu_request->u.presentRequest, - apdu_response->u.presentResponse); - s->send_Z_PDU(apdu_response, 0); - return 1; + apdu_response = s->create_Z_PDU(Z_APDU_presentResponse); + s->transfer_referenceId(apdu_request, apdu_response); + sr_present (apdu_request->u.presentRequest, + apdu_response->u.presentResponse); + if (!apdu_response->u.presentResponse->records) + fetch_via_present(s, apdu_request->u.presentRequest, + apdu_response->u.presentResponse); + s->send_Z_PDU(apdu_response, 0); + return 1; } return 0; } +/* + * Local variables: + * c-basic-offset: 4 + * indent-tabs-mode: nil + * End: + * vim: shiftwidth=4 tabstop=8 expandtab + */ +