summaryrefslogtreecommitdiffstats
path: root/reader/src/database/booksdb/runnables/LoadFileEntriesRunnable.cpp
blob: 3668b8397f794b9f29f8d97d62ff04bdf6276e2c (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
/*
 * Copyright (C) 2009-2012 Geometer Plus <contact@geometerplus.com>
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
 * 02110-1301, USA.
 */

#include "../DBRunnables.h"
#include "../../sqldb/implsqlite/SQLiteFactory.h"

LoadFileEntriesRunnable::LoadFileEntriesRunnable(DBConnection &connection) {
	myLoadFileEntries = SQLiteFactory::createCommand(BooksDBQuery::LOAD_FILE_ENTRIES, connection, "@file_id", DBValue::DBINT);

	myFindFileId = new FindFileIdRunnable(connection);
}

bool LoadFileEntriesRunnable::run() {
	DBCommand &cmd = *myLoadFileEntries;

	myFindFileId->setFileName(myFileName);
	if (!myFindFileId->run()) {
		return false;
	}
	((DBIntValue &) *cmd.parameter("@file_id").value()) = myFindFileId->fileId();

	shared_ptr<DBDataReader> reader = cmd.executeReader();

	if (reader.isNull()) {
		return false;
	}
	
	myEntries.clear();

	bool res = true;
	while (reader->next()) {
		if (reader->type(0) != DBValue::DBTEXT /* name */) {
			res = false;
			continue;
		}
		myEntries.push_back(
			myFileName + BooksDBQuery::ArchiveEntryDelimiter +
			reader->textValue(0, std::string())
		);
	}
	reader->close();
	return res;
}

void LoadFileEntriesRunnable::setFileName(const std::string &fileName) {
	myFileName = fileName;
}

void LoadFileEntriesRunnable::collectEntries(std::vector<std::string> &entries) {
	myEntries.swap(entries);
	myEntries.clear();
}