/*
 *	PROGRAM:	Object oriented API samples.
 *	MODULE:		12.batch_isc.cpp
 *	DESCRIPTION:	A sample of using Batch interface.
 *
 *					Example for the following interfaces:
 *					IUtil - get IStatement/ITransaction by handle
 *					IBatch - interface to work with FB batches
 *					IBatchCompletionState - contains result of batch execution
 *
 *	c++ 12.batch_isc.cpp -lfbclient
 *
 *  The contents of this file are subject to the Initial
 *  Developer's Public License Version 1.0 (the "License");
 *  you may not use this file except in compliance with the
 *  License. You may obtain a copy of the License at
 *  http://www.ibphoenix.com/main.nfs?a=ibphoenix&page=ibp_idpl.
 *
 *  Software distributed under the License is distributed AS IS,
 *  WITHOUT WARRANTY OF ANY KIND, either express or implied.
 *  See the License for the specific language governing rights
 *  and limitations under the License.
 *
 *  The Original Code was created by Alexander Peshkoff
 *  for the Firebird Open Source RDBMS project.
 *
 *  Copyright (c) 2018 Alexander Peshkoff <peshkoff@mail.ru>
 *  and all contributors signed below.
 *
 *  All Rights Reserved.
 *  Contributor(s): ______________________________________.
 */

#include "ifaceExamples.h"
#include <firebird/Message.h>

static IMaster* master = fb_get_master_interface();


// output error message to user

static void errPrint(IStatus* status)
{
	isc_print_status(status->getErrors());
}

static void raiseError(ThrowStatusWrapper& status, ISC_STATUS *vector)
{
	throw FbException(&status, vector);
}

// BatchCompletionState printer - prints all what we know about completed batch

static void print_cs(ThrowStatusWrapper& status, IBatchCompletionState* cs, IUtil* utl)
{
	unsigned p = 0;
	IStatus* s2 = NULL;
	bool pr1 = false, pr2 = false;

	// 1. Print per-message state info

	unsigned upcount = cs->getSize(&status);
	unsigned unk = 0, succ = 0;
	for (p = 0; p < upcount; ++p)
	{
		int s = cs->getState(&status, p);
		switch (s)
		{
		case IBatchCompletionState::EXECUTE_FAILED:
			if (!pr1)
			{
				printf("Message Status\n");
				pr1 = true;
			}
			printf("%5u   Execute failed\n", p);
			break;

		case IBatchCompletionState::SUCCESS_NO_INFO:
			++unk;
			break;

		default:
			if (!pr1)
			{
				printf("Message Status\n");
				pr1 = true;
			}
			printf("%5u   Updated %d record(s)\n", p, s);
			++succ;
			break;
		}
	}
	printf("Summary: total=%u success=%u success(but no update info)=%u\n", upcount, succ, unk);

	// 2. Print detailed errors (if exist) for messages 

	s2 = master->getStatus();
	for(p = 0; (p = cs->findError(&status, p)) != IBatchCompletionState::NO_MORE_ERRORS; ++p)
	{
		try
		{
			cs->getStatus(&status, s2, p);

			char text[1024];
			utl->formatStatus(text, sizeof(text) - 1, s2);
			text[sizeof(text) - 1] = 0;
			if (!pr2)
			{
				printf("\nDetailed errors status:\n");
				pr2 = true;
			}
			printf("Message %u: %s\n", p, text);
		}
		catch (const FbException& error)
		{
			// handle error
			fprintf(stderr, "\nError describing message %u\n", p);
			errPrint(error.getStatus());
			fprintf(stderr, "\n");
		}
	}

	if (s2)
		s2->dispose();
}

int main()
{
	int rc = 0;

	// set default password if none specified in environment
	setenv("ISC_USER", "sysdba", 0);
	setenv("ISC_PASSWORD", "masterkey", 0);

	// With ThrowStatusWrapper passed as status interface FbException will be thrown on error
	ThrowStatusWrapper status(master->getStatus());

	ISC_STATUS_ARRAY st;
	isc_db_handle db = 0;
	isc_tr_handle tr = 0;
	isc_stmt_handle stmt = 0;
	isc_blob_handle blb = 0;

	// Declare pointers to required interfaces
	/*IProvider* prov = master->getDispatcher();
	IAttachment* att = NULL;*/
	IUtil* utl = master->getUtilInterface();
	IStatement* statemt = NULL;
	ITransaction* tra = NULL;
	IBatch* batch = NULL;
	IBatchCompletionState* cs = NULL;
	IXpbBuilder* pb = NULL;

	unsigned char streamBuf[10240];		// big enough for demo
	unsigned char* stream = NULL;

	try
	{
		// attach employee db
		if (isc_attach_database(st, 0, "employee", &db, 0, NULL))
			raiseError(status, st);

		isc_tr_handle tr = 0;
		if (isc_start_transaction(st, &tr, 1, &db, 0, NULL))
			raiseError(status, st);

		// cleanup
		const char* cleanSql = "delete from project where proj_id like 'BAT%'";
		if (isc_dsql_execute_immediate(st, &db, &tr, 0, cleanSql, 3, NULL))
			raiseError(status, st);

		if (isc_dsql_allocate_statement(st, &db, &stmt))
			raiseError(status, st);

		// get transaction interface
		if (fb_get_transaction_interface(st, &tra, &tr))
			raiseError(status, st);

		//
		printf("\nPart 1. BLOB created using IBlob interface.\n");
		//

		// prepare statement
		const char* sqlStmt1 = "insert into project(proj_id, proj_name) values(?, ?)";
		if (isc_dsql_prepare(st, &tr, &stmt, 0, sqlStmt1, 3, NULL))
			raiseError(status, st);
		// and get it's interface
		if (fb_get_statement_interface(st, &statemt, &stmt))
			raiseError(status, st);

		// Message to store in a table
		FB_MESSAGE(Msg1, ThrowStatusWrapper,
			(FB_VARCHAR(5), id)
			(FB_VARCHAR(10), name)
		) project1(&status, master);
		project1.clear();
		IMessageMetadata* meta = project1.getMetadata();

		// set batch parameters
		pb = utl->getXpbBuilder(&status, IXpbBuilder::BATCH, NULL, 0);
		// collect per-message statistics
		pb->insertInt(&status, IBatch::TAG_RECORD_COUNTS, 1);

		// create batch
		batch = statemt->createBatch(&status, meta,
			pb->getBufferLength(&status), pb->getBuffer(&status));

		// fill batch with data record by record
		project1->id.set("BAT11");
		project1->name.set("SNGL_REC");
		batch->add(&status, 1, project1.getData());

		project1->id.set("BAT12");
		project1->name.set("SNGL_REC2");
		batch->add(&status, 1, project1.getData());

		// execute it
		cs = batch->execute(&status, tra);
		print_cs(status, cs, utl);

		// close batch
		batch->release();
		batch = NULL;

		// unprepare statement
		statemt->release();
		statemt = NULL;
		if (isc_dsql_free_statement(st, &stmt, DSQL_unprepare))
			raiseError(status, st);

		//
		printf("\nPart 2. BLOB created using isc_create_blob.\n");
		//

		// prepare statement
		const char* sqlStmt2 = "insert into project(proj_id, proj_name, proj_desc) values(?, ?, ?)";
		if (isc_dsql_prepare(st, &tr, &stmt, 0, sqlStmt2, 3, NULL))
			raiseError(status, st);
		// and get it's interface
		if (fb_get_statement_interface(st, &statemt, &stmt))
			raiseError(status, st);

		// Message to store in a table
		FB_MESSAGE(Msg2, ThrowStatusWrapper,
			(FB_VARCHAR(5), id)
			(FB_VARCHAR(10), name)
			(FB_BLOB, desc)
		) project2(&status, master);
		project2.clear();
		meta = project2.getMetadata();

		// set batch parameters
		pb->clear(&status);
		// enable blobs processing - IDs generated by firebird engine
		pb->insertInt(&status, IBatch::TAG_BLOB_POLICY, IBatch::BLOB_ID_ENGINE);

		// create batch
		batch = statemt->createBatch(&status, meta,
			pb->getBufferLength(&status), pb->getBuffer(&status));

		// create blob
		ISC_QUAD realId;
		if (isc_create_blob(st, &db, &tr, &blb, &realId))
			raiseError(status, st);
		const char* text = "Blob created using traditional API";
		if (isc_put_segment(st, &blb, strlen(text), text))
			raiseError(status, st);
		if (isc_close_blob(st, &blb))
			raiseError(status, st);

		// add message
		project2->id.set("BAT38");
		project2->name.set("FRGN_BLB");
		batch->registerBlob(&status, &realId, &project2->desc);
		batch->add(&status, 1, project2.getData());

		// execute it
		cs = batch->execute(&status, tra);
		print_cs(status, cs, utl);

		// close batch
		batch->release();
		batch = NULL;

		// unprepare statement
		statemt->release();
		statemt = NULL;
		if (isc_dsql_free_statement(st, &stmt, DSQL_drop))
			raiseError(status, st);

		// cleanup
		tra->release();
		tra = NULL;

	    if (isc_commit_transaction (st, &tr))
	        raiseError(status, st);

	    if (isc_detach_database(st, &db))
	        raiseError(status, st);
	}
	catch (const FbException& error)
	{
		// handle error
		rc = 1;
		errPrint(error.getStatus());
	}

	// release interfaces after error caught
	if (cs)
		cs->dispose();
	if (batch)
		batch->release();
	if (tra)
		tra->release();
	if (statemt)
		statemt->release();

	// close handles if not closed
	if (blb)
		isc_cancel_blob(st, &blb);
	if (stmt)
		isc_dsql_free_statement(st, &stmt, DSQL_close);
	if (tr)
	    isc_rollback_transaction (st, &tr);
	if (db)
    	isc_detach_database(st, &db);

	// cleanup
	if (pb)
		pb->dispose();
	status.dispose();

	return rc;
}