2 * Copyright (c) 2002-2004, Index Data.
3 * See the file LICENSE for details.
5 * $Id: yaz-z-cache.cpp,v 1.8 2004-01-24 21:32:31 adam Exp $
9 #include <yaz++/proxy.h>
11 struct Yaz_RecordCache_Entry {
13 Z_NamePlusRecord *m_record;
14 Z_RecordComposition *m_comp;
15 Yaz_RecordCache_Entry *m_next;
18 Yaz_RecordCache::Yaz_RecordCache ()
20 m_mem = nmem_create();
27 Yaz_RecordCache::~Yaz_RecordCache ()
32 void Yaz_RecordCache::set_max_size(int sz)
37 void Yaz_RecordCache::clear ()
40 m_mem = nmem_create();
46 void Yaz_RecordCache::copy_searchRequest(Z_SearchRequest *sr)
48 ODR encode = odr_createmem(ODR_ENCODE);
49 ODR decode = odr_createmem(ODR_DECODE);
53 int v = z_SearchRequest (encode, &sr, 1, 0);
57 char *buf = odr_getbuf(encode, &len, 0);
58 odr_setbuf(decode, buf, len, 0);
59 z_SearchRequest(decode, &m_searchRequest, 1, 0);
60 nmem_transfer(m_mem, decode->mem);
66 void Yaz_RecordCache::copy_presentRequest(Z_PresentRequest *pr)
68 ODR encode = odr_createmem(ODR_ENCODE);
69 ODR decode = odr_createmem(ODR_DECODE);
73 int v = z_PresentRequest (encode, &pr, 1, 0);
77 char *buf = odr_getbuf(encode, &len, 0);
78 odr_setbuf(decode, buf, len, 0);
79 z_PresentRequest(decode, &m_presentRequest, 1, 0);
80 nmem_transfer(m_mem, decode->mem);
86 void Yaz_RecordCache::add (ODR o, Z_NamePlusRecordList *npr, int start,
89 if (nmem_total(m_mem) > m_max_size)
91 // Build appropriate compspec for this response
92 Z_RecordComposition *comp = 0;
93 if (hits == -1 && m_presentRequest)
94 comp = m_presentRequest->recordComposition;
95 else if (hits > 0 && m_searchRequest)
97 Z_ElementSetNames *esn;
99 if (hits <= *m_searchRequest->smallSetUpperBound)
100 esn = m_searchRequest->smallSetElementSetNames;
102 esn = m_searchRequest->mediumSetElementSetNames;
103 comp = (Z_RecordComposition *) nmem_malloc(m_mem, sizeof(*comp));
104 comp->which = Z_RecordComp_simple;
105 comp->u.simple = esn;
108 // Z_NamePlusRecordList *npr to be owned by m_mem..
109 NMEM tmp_mem = odr_extract_mem(o);
110 nmem_transfer(m_mem, tmp_mem);
111 nmem_destroy(tmp_mem);
113 // Insert individual records in cache
115 for (i = 0; i<npr->num_records; i++)
117 Yaz_RecordCache_Entry *entry = (Yaz_RecordCache_Entry *)
118 nmem_malloc(m_mem, sizeof(*entry));
119 entry->m_record = (Z_NamePlusRecord *)
120 nmem_malloc(m_mem, sizeof(*entry->m_record));
121 entry->m_record->databaseName = npr->records[i]->databaseName;
122 entry->m_record->which = npr->records[i]->which;
123 entry->m_record->u.databaseRecord = npr->records[i]->u.databaseRecord;
124 entry->m_comp = comp;
125 entry->m_offset = i + start;
126 entry->m_next = m_entries;
131 int Yaz_RecordCache::match (Yaz_RecordCache_Entry *entry,
132 Odr_oid *syntax, int offset,
133 Z_RecordComposition *comp)
135 // See if our compspec match...
137 ODR o1 = odr_createmem(ODR_ENCODE);
138 ODR o2 = odr_createmem(ODR_ENCODE);
140 z_RecordComposition(o1, &comp, 1, 0);
141 z_RecordComposition(o2, &entry->m_comp, 1, 0);
144 char *buf1 = odr_getbuf(o1, &len1, 0);
146 char *buf2 = odr_getbuf(o2, &len2, 0);
148 if (buf1 && buf2 && len1 && len1 == len2 && !memcmp(buf1, buf2, len1))
150 else if (!buf1 && !buf2 && !len1 && !len2)
159 // See if offset, OID match..
160 if (entry->m_offset == offset &&
161 entry->m_record->which == Z_NamePlusRecord_databaseRecord &&
162 !oid_oidcmp(entry->m_record->u.databaseRecord->direct_reference,
167 oid_to_dotstring(entry->m_record->u.databaseRecord->direct_reference, mstr1);
169 oid_to_dotstring(syntax, mstr2);
170 yaz_log(LOG_LOG, "match fail 3 d=%s s=%s", mstr1, mstr2);
176 int Yaz_RecordCache::lookup (ODR o, Z_NamePlusRecordList **npr,
179 Z_RecordComposition *comp)
182 yaz_log(LOG_DEBUG, "cache lookup start=%d num=%d", start, num);
184 for (i = 0; i<num; i++)
186 Yaz_RecordCache_Entry *entry = m_entries;
187 for(; entry; entry = entry->m_next)
188 if (match(entry, syntax, start+i, comp))
193 *npr = (Z_NamePlusRecordList *) odr_malloc(o, sizeof(**npr));
194 (*npr)->num_records = num;
195 (*npr)->records = (Z_NamePlusRecord **)
196 odr_malloc(o, num * sizeof(Z_NamePlusRecord *));
197 for (i = 0; i<num; i++)
199 Yaz_RecordCache_Entry *entry = m_entries;
200 for(; entry; entry = entry->m_next)
201 if (match(entry, syntax, start+i, comp))
205 (*npr)->records[i] = (Z_NamePlusRecord *)
206 odr_malloc(o, sizeof(Z_NamePlusRecord));
207 (*npr)->records[i]->databaseName = entry->m_record->databaseName;
208 (*npr)->records[i]->which = entry->m_record->which;
209 (*npr)->records[i]->u.databaseRecord =
210 entry->m_record->u.databaseRecord;