[15757] | 1 | /**********************************************************************
|
---|
| 2 | *
|
---|
| 3 | * sqlbrowsefilter.cpp --
|
---|
| 4 | * Copyright (C) 2008 DL Consulting Ltd
|
---|
| 5 | *
|
---|
| 6 | * A component of the Greenstone digital library software
|
---|
| 7 | * from the New Zealand Digital Library Project at the
|
---|
| 8 | * University of Waikato, New Zealand.
|
---|
| 9 | *
|
---|
| 10 | * This program is free software; you can redistribute it and/or modify
|
---|
| 11 | * it under the terms of the GNU General Public License as published by
|
---|
| 12 | * the Free Software Foundation; either version 2 of the License, or
|
---|
| 13 | * (at your option) any later version.
|
---|
| 14 | *
|
---|
| 15 | * This program is distributed in the hope that it will be useful,
|
---|
| 16 | * but WITHOUT ANY WARRANTY; without even the implied warranty of
|
---|
| 17 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
---|
| 18 | * GNU General Public License for more details.
|
---|
| 19 | *
|
---|
| 20 | * You should have received a copy of the GNU General Public License
|
---|
| 21 | * along with this program; if not, write to the Free Software
|
---|
| 22 | * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
|
---|
| 23 | *
|
---|
| 24 | *********************************************************************/
|
---|
| 25 |
|
---|
| 26 | #include "sqlbrowsefilter.h"
|
---|
| 27 | #include "fileutil.h"
|
---|
| 28 |
|
---|
| 29 |
|
---|
| 30 | sqlbrowsefilterclass::sqlbrowsefilterclass ()
|
---|
| 31 | {
|
---|
| 32 | sql_db_ptr = NULL;
|
---|
| 33 | }
|
---|
| 34 |
|
---|
| 35 |
|
---|
| 36 | sqlbrowsefilterclass::~sqlbrowsefilterclass ()
|
---|
| 37 | {
|
---|
| 38 | }
|
---|
| 39 |
|
---|
| 40 |
|
---|
| 41 | void sqlbrowsefilterclass::configure (const text_t &key, const text_tarray &cfgline)
|
---|
| 42 | {
|
---|
| 43 | filterclass::configure (key, cfgline);
|
---|
| 44 |
|
---|
| 45 | if (key == "indexstem")
|
---|
| 46 | {
|
---|
| 47 | indexstem = cfgline[0];
|
---|
| 48 | }
|
---|
| 49 | }
|
---|
| 50 |
|
---|
| 51 |
|
---|
| 52 | bool sqlbrowsefilterclass::init (ostream &logout)
|
---|
| 53 | {
|
---|
| 54 | outconvertclass text_t2ascii;
|
---|
| 55 |
|
---|
| 56 | if (!filterclass::init(logout)) return false;
|
---|
| 57 |
|
---|
| 58 | if (sql_db_ptr == NULL)
|
---|
| 59 | {
|
---|
| 60 | // most likely a configuration problem
|
---|
[15766] | 61 | logout << text_t2ascii << "configuration error: sqlbrowsefilter contains a null sqldbclass\n\n";
|
---|
[15757] | 62 | return false;
|
---|
| 63 | }
|
---|
| 64 |
|
---|
| 65 | if (indexstem.empty())
|
---|
| 66 | {
|
---|
| 67 | indexstem = collection;
|
---|
| 68 | }
|
---|
| 69 |
|
---|
| 70 | // get the filename for the database and make sure it exists
|
---|
| 71 | sql_db_filename = filename_cat(dbhome, "collect", collection, "index", "text", indexstem);
|
---|
| 72 | sql_db_filename += sql_db_ptr->getfileextension();
|
---|
| 73 | if (!file_exists(sql_db_filename))
|
---|
| 74 | {
|
---|
[15766] | 75 | logout << text_t2ascii << "warning: database \"" << sql_db_filename << "\" does not exist\n\n";
|
---|
[15757] | 76 | return false;
|
---|
| 77 | }
|
---|
| 78 |
|
---|
| 79 | return true;
|
---|
| 80 | }
|
---|
| 81 |
|
---|
| 82 |
|
---|
| 83 | void sqlbrowsefilterclass::filter (const FilterRequest_t &request,
|
---|
| 84 | FilterResponse_t &response,
|
---|
| 85 | comerror_t &err, ostream &logout)
|
---|
| 86 | {
|
---|
| 87 | outconvertclass text_t2ascii;
|
---|
| 88 |
|
---|
| 89 | response.clear();
|
---|
| 90 | err = noError;
|
---|
| 91 |
|
---|
| 92 | if (sql_db_ptr == NULL) {
|
---|
| 93 | // most likely a configuration problem
|
---|
[15765] | 94 | logout << text_t2ascii << "configuration error: sqlbrowsefilter contains a null sqldbclass\n\n";
|
---|
[15757] | 95 | err = configurationError;
|
---|
| 96 | return;
|
---|
| 97 | }
|
---|
| 98 |
|
---|
| 99 | // open the database
|
---|
| 100 | sql_db_ptr->setlogout (&logout);
|
---|
| 101 | if (!sql_db_ptr->opendatabase (sql_db_filename, DB_READER, 100, false)) {
|
---|
| 102 | // most likely a system problem (we have already checked that the database exists)
|
---|
[15765] | 103 | logout << text_t2ascii << "system problem: open on database \"" << sql_db_filename << "\" failed\n\n";
|
---|
[15757] | 104 | err = systemProblem;
|
---|
| 105 | return;
|
---|
| 106 | }
|
---|
| 107 |
|
---|
[15764] | 108 | // Request for the metadata values assigned to an element
|
---|
[15757] | 109 | if ((request.filterResultOptions & FRmetadataValues) && !request.fields.empty())
|
---|
| 110 | {
|
---|
| 111 | text_t metadata_element_name = *(request.fields.begin());
|
---|
| 112 | text_tarray metadata_values = sql_db_ptr->getmetadatavalues (metadata_element_name);
|
---|
[15764] | 113 |
|
---|
| 114 | // Create a map from metadata value to ResultDocInfo_t, to remove duplicate values and obtain occurrence counts
|
---|
| 115 | map<text_t, ResultDocInfo_t> unique_metadata_values_map;
|
---|
[15757] | 116 | text_tarray::iterator metadata_value_iterator = metadata_values.begin();
|
---|
| 117 | while (metadata_value_iterator != metadata_values.end())
|
---|
| 118 | {
|
---|
[15764] | 119 | text_t metadata_value = *metadata_value_iterator;
|
---|
[15757] | 120 |
|
---|
[15764] | 121 | // If no ResultDocInfo_t has already been created for this metadata value, create one now
|
---|
| 122 | if (unique_metadata_values_map.find(metadata_value) == unique_metadata_values_map.end())
|
---|
| 123 | {
|
---|
| 124 | ResultDocInfo_t metadata_value_result_doc;
|
---|
| 125 | metadata_value_result_doc.OID = metadata_value;
|
---|
| 126 | metadata_value_result_doc.result_num = 1;
|
---|
| 127 | unique_metadata_values_map[metadata_value] = metadata_value_result_doc;
|
---|
| 128 | }
|
---|
| 129 | // Otherwise we've seen this value before, so just update the occurrence count
|
---|
| 130 | else
|
---|
| 131 | {
|
---|
| 132 | unique_metadata_values_map[metadata_value].result_num++;
|
---|
| 133 | }
|
---|
| 134 |
|
---|
[15757] | 135 | metadata_value_iterator++;
|
---|
| 136 | }
|
---|
[15764] | 137 |
|
---|
| 138 | // Fill in response.docInfo with the ResultDocInfo_t objects we've created above
|
---|
| 139 | map<text_t, ResultDocInfo_t>::iterator unique_metadata_values_iterator = unique_metadata_values_map.begin();
|
---|
| 140 | while (unique_metadata_values_iterator != unique_metadata_values_map.end())
|
---|
| 141 | {
|
---|
| 142 | response.docInfo.push_back ((*unique_metadata_values_iterator).second);
|
---|
| 143 | unique_metadata_values_iterator++;
|
---|
| 144 | }
|
---|
[15757] | 145 | }
|
---|
| 146 |
|
---|
| 147 | sql_db_ptr->closedatabase(); // Important that local library doesn't leave any files open
|
---|
| 148 | }
|
---|