database.cc
No OneTemporary
Actions

Size

8 KB

Subscribers

None

database.cc
View Options

	/*
	* Copyright (C) 2014, Siemens AG
	* Author: Daniele Fognini, Johannes Najjar
	*
	* This program is free software; you can redistribute it and/or modify
	* it under the terms of the GNU General Public License version 2
	* as published by the Free Software Foundation.
	*
	* 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 "database.hpp"
	#include "identity.hpp"

	#include <iostream>
	#include <libfossUtils.hpp>

	using namespace fo;

	#define RETURN_IF_FALSE(query) \
	do {\
	if (!(query)) {\
	return false;\
	}\
	} while(0)

	DatabaseEntry::DatabaseEntry() :
	agent_fk(0),
	pfile_fk(0),
	content(""),
	hash(""),
	type(""),
	copy_startbyte(0),
	copy_endbyte(0)
	{
	};


	CopyrightDatabaseHandler CopyrightDatabaseHandler::spawn() const
	{
	DbManager spawnedDbMan(dbManager.spawn());
	return CopyrightDatabaseHandler(spawnedDbMan);
	}


	std::string CopyrightDatabaseHandler::getColumnListString(const CopyrightDatabaseHandler::ColumnDef in[], size_t size) const
	{
	std::string result;
	for (size_t i = 0; i < size; ++i)
	{
	if (i != 0)
	result += ", ";
	result += in[i].name;
	}
	return result;
	}

	std::string CopyrightDatabaseHandler::getColumnCreationString(const CopyrightDatabaseHandler::ColumnDef in[], size_t size) const
	{
	std::string result;
	for (size_t i = 0; i < size; ++i)
	{
	if (i != 0)
	result += ", ";
	result += in[i].name;
	result += " ";
	result += in[i].type;
	result += " ";
	result += in[i].creationFlags;
	}
	return result;
	}

	bool CopyrightDatabaseHandler::createTables() const
	{
	int failedCounter = 0;
	bool tablesChecked = false;

	dbManager.ignoreWarnings(true);
	while (!tablesChecked && failedCounter < MAX_TABLE_CREATION_RETRIES)
	{
	dbManager.begin();

	tablesChecked = createTableAgentFindings() && createTableClearing();

	if (tablesChecked)
	dbManager.commit();
	else
	{
	dbManager.rollback();
	++failedCounter;
	if (failedCounter < MAX_TABLE_CREATION_RETRIES)
	std::cout << "WARNING: table creation failed: trying again"
	" (" << failedCounter << "/" << MAX_TABLE_CREATION_RETRIES << ")"
	<< std::endl;
	}
	}
	if (tablesChecked && (failedCounter > 0))
	std::cout << "NOTICE: table creation succeded on try "
	<< failedCounter << "/" << MAX_TABLE_CREATION_RETRIES
	<< std::endl;

	dbManager.ignoreWarnings(false);
	return tablesChecked;
	}

	const CopyrightDatabaseHandler::ColumnDef CopyrightDatabaseHandler::columns[] =
	{
	#define SEQUENCE_NAME IDENTITY"_ct_pk_seq"
	{"ct_pk", "bigint", "PRIMARY KEY DEFAULT nextval('" SEQUENCE_NAME "'::regclass)"},
	{"agent_fk", "bigint", "NOT NULL"},
	{"pfile_fk", "bigint", "NOT NULL"},
	{"content", "text", ""},
	{"hash", "text", ""},
	{"type", "text", ""}, //TODO removed constrain: "CHECK (type in ('statement', 'email', 'url'))"},
	{"copy_startbyte", "integer", ""},
	{"copy_endbyte", "integer", ""},
	};

	bool CopyrightDatabaseHandler::createTableAgentFindings() const
	{
	if (!dbManager.sequenceExists(SEQUENCE_NAME))
	{
	RETURN_IF_FALSE(dbManager.queryPrintf("CREATE SEQUENCE "
	SEQUENCE_NAME
	" START WITH 1"
	" INCREMENT BY 1"
	" NO MAXVALUE"
	" NO MINVALUE"
	" CACHE 1"));
	}

	if (!dbManager.tableExists(IDENTITY))
	{
	size_t ncolumns = (sizeof(CopyrightDatabaseHandler::columns) / sizeof(CopyrightDatabaseHandler::ColumnDef));
	RETURN_IF_FALSE(dbManager.queryPrintf("CREATE table %s(%s)", IDENTITY,
	getColumnCreationString(CopyrightDatabaseHandler::columns, ncolumns).c_str()
	)
	);
	RETURN_IF_FALSE(dbManager.queryPrintf(
	"CREATE INDEX %s_agent_fk_index"
	" ON %s"
	" USING BTREE (agent_fk)",
	IDENTITY, IDENTITY
	));

	RETURN_IF_FALSE(dbManager.queryPrintf(
	"CREATE INDEX %s_hash_index"
	" ON %s"
	" USING BTREE (hash)",
	IDENTITY, IDENTITY
	));

	RETURN_IF_FALSE(dbManager.queryPrintf(
	"CREATE INDEX %s_pfile_fk_index"
	" ON %s"
	" USING BTREE (pfile_fk)",
	IDENTITY, IDENTITY
	));

	RETURN_IF_FALSE(dbManager.queryPrintf(
	"ALTER TABLE ONLY %s"
	" ADD CONSTRAINT agent_fk"
	" FOREIGN KEY (agent_fk)"
	" REFERENCES agent(agent_pk) ON DELETE CASCADE",
	IDENTITY
	));

	RETURN_IF_FALSE(dbManager.queryPrintf(
	"ALTER TABLE ONLY %s"
	" ADD CONSTRAINT pfile_fk"
	" FOREIGN KEY (pfile_fk)"
	" REFERENCES pfile(pfile_pk) ON DELETE CASCADE",
	IDENTITY
	));
	}
	return true;
	}


	const CopyrightDatabaseHandler::ColumnDef CopyrightDatabaseHandler::columnsDecision[] = {
	#define SEQUENCE_NAMEClearing IDENTITY"_decision_pk_seq"
	{"copyright_decision_pk", "bigint", "PRIMARY KEY DEFAULT nextval('" SEQUENCE_NAMEClearing "'::regclass)"},
	{"user_fk", "bigint", "NOT NULL"},
	{"pfile_fk", "bigint", "NOT NULL"},
	{"clearing_decision_type_fk", "bigint", "NOT NULL"},
	{"description", "text", ""},
	{"textFinding", "text", ""},
	{"comment", "text", ""}
	};

	bool CopyrightDatabaseHandler::createTableClearing() const
	{
	#define CLEARING_TABLE IDENTITY "_decision"

	if (!dbManager.sequenceExists(SEQUENCE_NAMEClearing))
	{
	RETURN_IF_FALSE(dbManager.queryPrintf("CREATE SEQUENCE "
	SEQUENCE_NAMEClearing
	" START WITH 1"
	" INCREMENT BY 1"
	" NO MAXVALUE"
	" NO MINVALUE"
	" CACHE 1"));
	}

	if (!dbManager.tableExists(CLEARING_TABLE))
	{
	size_t nDec = (sizeof(CopyrightDatabaseHandler::columnsDecision) / sizeof(CopyrightDatabaseHandler::ColumnDef));
	RETURN_IF_FALSE(dbManager.queryPrintf("CREATE table %s(%s)", CLEARING_TABLE,
	getColumnCreationString(CopyrightDatabaseHandler::columnsDecision, nDec).c_str()));

	RETURN_IF_FALSE(dbManager.queryPrintf(
	"CREATE INDEX %s_pfile_fk_index"
	" ON %s"
	" USING BTREE (pfile_fk)",
	CLEARING_TABLE, CLEARING_TABLE
	));

	RETURN_IF_FALSE(dbManager.queryPrintf(
	"CREATE INDEX %s_user_fk_index"
	" ON %s"
	" USING BTREE (user_fk)",
	CLEARING_TABLE, CLEARING_TABLE
	));

	RETURN_IF_FALSE(dbManager.queryPrintf(
	"CREATE INDEX %s_clearing_decision_type_fk_index"
	" ON %s"
	" USING BTREE (clearing_decision_type_fk)",
	CLEARING_TABLE, CLEARING_TABLE
	));

	RETURN_IF_FALSE(dbManager.queryPrintf(
	"ALTER TABLE ONLY %s"
	" ADD CONSTRAINT user_fk"
	" FOREIGN KEY (user_fk)"
	" REFERENCES users(user_pk) ON DELETE CASCADE",
	CLEARING_TABLE
	));

	RETURN_IF_FALSE(dbManager.queryPrintf(
	"ALTER TABLE ONLY %s"
	" ADD CONSTRAINT pfile_fk"
	" FOREIGN KEY (pfile_fk)"
	" REFERENCES pfile(pfile_pk) ON DELETE CASCADE",
	CLEARING_TABLE
	));
	}

	return true;
	}

	std::vector<unsigned long> CopyrightDatabaseHandler::queryFileIdsForUpload(int agentId, int uploadId)
	{
	std::string uploadTreeTableName = queryUploadTreeTableName(uploadId);

	QueryResult queryResult = dbManager.queryPrintf(
	"SELECT pfile_pk"
	" FROM ("
	" SELECT distinct(pfile_fk) AS PF"
	" FROM %s"
	" WHERE upload_fk = %d and (ufile_mode&x'3C000000'::int)=0"
	" ) AS SS "
	"left outer join %s on (PF = pfile_fk and agent_fk = %d) "
	"inner join pfile on (PF = pfile_pk) "
	"WHERE ct_pk IS null or agent_fk <> %d",
	uploadTreeTableName.c_str(),
	uploadId,
	IDENTITY,
	agentId,
	agentId
	);

	return queryResult.getSimpleResults<unsigned long>(0, fo::stringToUnsignedLong);
	}

	bool CopyrightDatabaseHandler::insertNoResultInDatabase(long int agentId, long int pFileId) const
	{
	return dbManager.execPrepared(
	fo_dbManager_PrepareStamement(
	dbManager.getStruct_dbManager(),
	"insertNoResultInDatabase",
	"INSERT INTO "
	IDENTITY
	"(agent_fk, pfile_fk) VALUES($1,$2)",
	long, long
	),
	agentId, pFileId
	);
	}

	bool CopyrightDatabaseHandler::insertInDatabase(DatabaseEntry& entry) const
	{
	return dbManager.execPrepared(
	fo_dbManager_PrepareStamement(
	dbManager.getStruct_dbManager(),
	"insertInDatabase",
	"INSERT INTO "
	IDENTITY
	"(agent_fk, pfile_fk, content, hash, type, copy_startbyte, copy_endbyte)"
	" VALUES($1,$2,$3,md5($3),$4,$5,$6)",
	long, long, char, char, int, int
	),
	entry.agent_fk, entry.pfile_fk,
	entry.content.c_str(),
	entry.type.c_str(),
	entry.copy_startbyte, entry.copy_endbyte
	);
	}

	CopyrightDatabaseHandler::CopyrightDatabaseHandler(DbManager manager) :
	AgentDatabaseHandler(manager)
	{

	}

File Metadata

Mime Type: text/x-c
Expires: Wed, Jun 4, 7:15 PM (1 w, 13 h ago)
Storage Engine: blob
Storage Format: Raw Data
Storage Handle: 3382309

database.ccNo OneTemporaryActions

database.ccView Options

File Metadata

Event Timeline

database.cc
No OneTemporary
Actions

database.cc
View Options