root/gsdl/trunk/src/colservr/sqlbrowsefilter.cpp @ 15765

Revision 15765, 4.7 KB (checked in by mdewsnip, 12 years ago)

A couple of minor whitespace improvements.

Line 
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
30sqlbrowsefilterclass::sqlbrowsefilterclass ()
31{
32  sql_db_ptr = NULL;
33}
34
35
36sqlbrowsefilterclass::~sqlbrowsefilterclass ()
37{
38}
39
40
41void 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
52bool 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
61    logout << text_t2ascii
62       << "configuration error: sqlbrowsefilter contains a null sqldbclass\n\n";
63    return false;
64  }
65
66  if (indexstem.empty())
67  {
68    indexstem = collection;
69  }
70
71  // get the filename for the database and make sure it exists
72  sql_db_filename = filename_cat(dbhome, "collect", collection, "index", "text", indexstem);
73  sql_db_filename += sql_db_ptr->getfileextension();
74  if (!file_exists(sql_db_filename))
75  {
76    logout << text_t2ascii
77       << "warning: database \"" << sql_db_filename << "\" does not exist\n\n";
78    return false;
79  }
80
81  return true;
82}
83
84
85void sqlbrowsefilterclass::filter (const FilterRequest_t &request,
86                   FilterResponse_t &response,
87                   comerror_t &err, ostream &logout)
88{
89  outconvertclass text_t2ascii;
90
91  response.clear();
92  err = noError;
93
94  if (sql_db_ptr == NULL) {
95    // most likely a configuration problem
96    logout << text_t2ascii << "configuration error: sqlbrowsefilter contains a null sqldbclass\n\n";
97    err = configurationError;
98    return;
99  }
100
101  // open the database
102  sql_db_ptr->setlogout (&logout);
103  if (!sql_db_ptr->opendatabase (sql_db_filename, DB_READER, 100, false)) {
104    // most likely a system problem (we have already checked that the database exists)
105    logout << text_t2ascii << "system problem: open on database \"" << sql_db_filename << "\" failed\n\n";
106    err = systemProblem;
107    return;
108  }
109
110  // Request for the metadata values assigned to an element
111  if ((request.filterResultOptions & FRmetadataValues) && !request.fields.empty())
112  {
113    text_t metadata_element_name = *(request.fields.begin());
114    text_tarray metadata_values = sql_db_ptr->getmetadatavalues (metadata_element_name);
115
116    // Create a map from metadata value to ResultDocInfo_t, to remove duplicate values and obtain occurrence counts
117    map<text_t, ResultDocInfo_t> unique_metadata_values_map;
118    text_tarray::iterator metadata_value_iterator = metadata_values.begin();
119    while (metadata_value_iterator != metadata_values.end())
120    {
121      text_t metadata_value = *metadata_value_iterator;
122
123      // If no ResultDocInfo_t has already been created for this metadata value, create one now
124      if (unique_metadata_values_map.find(metadata_value) == unique_metadata_values_map.end())
125      {
126    ResultDocInfo_t metadata_value_result_doc;
127    metadata_value_result_doc.OID = metadata_value;
128    metadata_value_result_doc.result_num = 1;
129    unique_metadata_values_map[metadata_value] = metadata_value_result_doc;
130      }
131      // Otherwise we've seen this value before, so just update the occurrence count
132      else
133      {
134    unique_metadata_values_map[metadata_value].result_num++;
135      }
136
137      metadata_value_iterator++;
138    }
139
140    // Fill in response.docInfo with the ResultDocInfo_t objects we've created above
141    map<text_t, ResultDocInfo_t>::iterator unique_metadata_values_iterator = unique_metadata_values_map.begin();
142    while (unique_metadata_values_iterator != unique_metadata_values_map.end())
143    {
144      response.docInfo.push_back ((*unique_metadata_values_iterator).second);
145      unique_metadata_values_iterator++;
146    }
147  }
148
149  sql_db_ptr->closedatabase();  // Important that local library doesn't leave any files open
150}
Note: See TracBrowser for help on using the browser.