xapian-core/tests/harness/backendmanager_multi.cc

   1 /** @file backendmanager_multi.cc
   2  * @brief BackendManager subclass for multi databases.
   3  */
   4 /* Copyright (C) 2007,2008,2009,2011,2012,2013,2015,2017,2018 Olly Betts
   5  * Copyright (C) 2008 Lemur Consulting Ltd
   6  *
   7  * This program is free software; you can redistribute it and/or
   8  * modify it under the terms of the GNU General Public License as
   9  * published by the Free Software Foundation; either version 2 of the
  10  * License, or (at your option) any later version.
  11  *
  12  * This program is distributed in the hope that it will be useful,
  13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15  * GNU General Public License for more details.
  16  *
  17  * You should have received a copy of the GNU General Public License
  18  * along with this program; if not, write to the Free Software
  19  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301 USA
  20  */
  21
  22 #include <config.h>
  23
  24 #include "backendmanager_multi.h"
  25
  26 #include "filetests.h"
  27 #include "index_utils.h"
  28 #include "str.h"
  29
  30 #include <cerrno>
  31 #include <cstdio> // For rename().
  32 #include <cstring>
  33
  34 using namespace std;
  35
  36 BackendManagerMulti::BackendManagerMulti(const std::string& datadir_,
  37                                          BackendManager* sub_manager_)
  38     : BackendManager(datadir_),
  39       sub_manager(sub_manager_),
  40       cachedir(".multi" + sub_manager_->get_dbtype())
  41 {
  42     // Ensure the directory we store cached test databases in exists.
  43     (void)create_dir_if_needed(cachedir);
  44 }
  45
  46 std::string
  47 BackendManagerMulti::get_dbtype() const
  48 {
  49     return "multi_" + sub_manager->get_dbtype();
  50 }
  51
  52 #define NUMBER_OF_SUB_DBS 2
  53
  54 string
  55 BackendManagerMulti::createdb_multi(const string& name,
  56                                     const vector<string>& files)
  57 {
  58     string dbname;
  59     if (!name.empty()) {
  60         dbname = name;
  61     } else {
  62         dbname = "db";
  63         for (const string& file : files) {
  64             dbname += "__";
  65             dbname += file;
  66         }
  67     }
  68
  69     string db_path = cachedir;
  70     db_path += '/';
  71     db_path += dbname;
  72
  73     if (!name.empty()) {
  74         remove(db_path.c_str());
  75     } else {
  76         if (file_exists(db_path)) return db_path;
  77     }
  78
  79     string tmpfile = db_path + ".tmp";
  80     ofstream out(tmpfile.c_str());
  81     if (!out.is_open()) {
  82         string msg = "Couldn't create file '";
  83         msg += tmpfile;
  84         msg += "' (";
  85         msg += strerror(errno);
  86         msg += ')';
  87         throw msg;
  88     }
  89
  90     // Open NUMBER_OF_SUB_DBS databases and index files to them alternately so
  91     // a multi-db combining them contains the documents in the expected order.
  92     Xapian::WritableDatabase dbs;
  93     const string& subtype = sub_manager->get_dbtype();
  94     int flags = Xapian::DB_CREATE_OR_OVERWRITE;
  95     if (subtype == "glass") {
  96         flags |= Xapian::DB_BACKEND_GLASS;
  97     } else {
  98         string msg = "Unknown multidb subtype: ";
  99         msg += subtype;
 100         throw msg;
 101     }
 102     string dbbase = db_path;
 103     dbbase += "___";
 104     size_t dbbase_len = dbbase.size();
 105     string line = subtype;
 106     line += ' ';
 107     line += dbname;
 108     line += "___";
 109     for (size_t n = 0; n < NUMBER_OF_SUB_DBS; ++n) {
 110         dbbase += str(n);
 111         dbs.add_database(Xapian::WritableDatabase(dbbase, flags));
 112         dbbase.resize(dbbase_len);
 113         out << line << n << '\n';
 114     }
 115     out.close();
 116
 117     FileIndexer(get_datadir(), files).index_to(dbs);
 118     dbs.close();
 119
 120     if (rename(tmpfile.c_str(), db_path.c_str()) < 0) {
 121         throw Xapian::Database("rename failed", errno);
 122     }
 123
 124     last_wdb_path = db_path;
 125     return db_path;
 126 }
 127
 128 string
 129 BackendManagerMulti::do_get_database_path(const vector<string> & files)
 130 {
 131     return createdb_multi(string(), files);
 132 }
 133
 134 Xapian::WritableDatabase
 135 BackendManagerMulti::get_writable_database(const string& name, const string& file)
 136 {
 137     vector<string> files;
 138     if (!file.empty()) files.push_back(file);
 139     return Xapian::WritableDatabase(createdb_multi(name, files));
 140 }
 141
 142 string
 143 BackendManagerMulti::get_writable_database_path(const std::string& name)
 144 {
 145     return cachedir + "/" + name;
 146 }
 147
 148 string
 149 BackendManagerMulti::get_compaction_output_path(const string& name)
 150 {
 151     return cachedir + "/" + name;
 152 }
 153
 154 Xapian::WritableDatabase
 155 BackendManagerMulti::get_writable_database_again()
 156 {
 157     return Xapian::WritableDatabase(last_wdb_path);
 158 }
 159
 160 string
 161 BackendManagerMulti::get_writable_database_path_again()
 162 {
 163     return last_wdb_path;
 164 }