//-----------------------------------------------------------------------------
// Copyright (c) 2016, 2017 Oracle and/or its affiliates.  All rights reserved.
// This program is free software: you can modify it and/or redistribute it
// under the terms of:
//
// (i)  the Universal Permissive License v 1.0 or at your option, any
//      later version (http://oss.oracle.com/licenses/upl); and/or
//
// (ii) the Apache License v 2.0. (http://www.apache.org/licenses/LICENSE-2.0)
//-----------------------------------------------------------------------------

//-----------------------------------------------------------------------------
// TestDMLReturning.c
//   Tests DML returning clause which returns multiple rows.
//-----------------------------------------------------------------------------

#include "Test.h"
#define SQL_TEXT_1          "truncate table TestTempTable"
#define SQL_TEXT_2          "begin " \
                            "    for i in 1..10 loop " \
                            "        insert into TestTempTable values (i, " \
                            "                'Test String ' || i); " \
                            "    end loop; " \
                            "end;"
#define SQL_TEXT_3          "delete from TestTempTable " \
                            "where IntCol <= :inIntCol " \
                            "returning IntCol, StringCol " \
                            "into :outIntCol, :outStrCol"

//-----------------------------------------------------------------------------
// main()
//-----------------------------------------------------------------------------
int main(int argc, char **argv)
{
    dpiData *inIntColData, *outIntColData, *outStrColData;
    dpiVar *inIntColVar, *outIntColVar, *outStrColVar;
    uint32_t numQueryColumns, numRows, i;
    dpiStmt *stmt;
    dpiConn *conn;

    // connect to database
    conn = GetConnection(1, NULL);
    if (!conn)
        return -1;

    // truncate table
    printf("Truncating table...\n");
    if (dpiConn_prepareStmt(conn, 0, SQL_TEXT_1, strlen(SQL_TEXT_1), NULL, 0,
            &stmt) < 0)
        return ShowError();
    if (dpiStmt_execute(stmt, 0, &numQueryColumns) < 0)
        return ShowError();
    dpiStmt_release(stmt);

    // populate table
    printf("Populating table...\n");
    if (dpiConn_prepareStmt(conn, 0, SQL_TEXT_2, strlen(SQL_TEXT_2), NULL, 0,
            &stmt) < 0)
        return ShowError();
    if (dpiStmt_execute(stmt, 0, &numQueryColumns) < 0)
        return ShowError();
    dpiStmt_release(stmt);

    // create variables
    if (dpiConn_newVar(conn, DPI_ORACLE_TYPE_NUMBER, DPI_NATIVE_TYPE_INT64, 1,
            0, 0, 0, NULL, &inIntColVar, &inIntColData) < 0)
        return ShowError();
    if (dpiConn_newVar(conn, DPI_ORACLE_TYPE_NUMBER, DPI_NATIVE_TYPE_INT64, 1,
            0, 0, 0, NULL, &outIntColVar, &outIntColData) < 0)
        return ShowError();
    if (dpiConn_newVar(conn, DPI_ORACLE_TYPE_VARCHAR, DPI_NATIVE_TYPE_BYTES, 1,
            100, 0, 0, NULL, &outStrColVar, &outStrColData) < 0)
        return ShowError();

    // prepare and execute delete statement with DML returning clause
    printf("Deleting rows with DML returning...\n");
    if (dpiConn_prepareStmt(conn, 0, SQL_TEXT_3, strlen(SQL_TEXT_3), NULL, 0,
            &stmt) < 0)
        return ShowError();
    inIntColData->isNull = 0;
    inIntColData->value.asInt64 = 4;
    if (dpiStmt_bindByPos(stmt, 1, inIntColVar) < 0)
        return ShowError();
    if (dpiStmt_bindByPos(stmt, 2, outIntColVar) < 0)
        return ShowError();
    if (dpiStmt_bindByPos(stmt, 3, outStrColVar) < 0)
        return ShowError();
    if (dpiStmt_execute(stmt, 0, &numQueryColumns) < 0)
        return ShowError();

    // get the data and number of rows returned
    if (dpiVar_getData(outIntColVar, &numRows, &outIntColData) < 0)
        return ShowError();
    if (dpiVar_getData(outStrColVar, &numRows, &outStrColData) < 0)
        return ShowError();
    printf("%d rows returned.\n", numRows);

    // display the output from the rows
    for (i = 0; i < numRows; i++)
        printf("IntCol = %" PRId64 ", StrCol = %.*s\n",
                outIntColData[i].value.asInt64,
                outStrColData[i].value.asBytes.length,
                outStrColData[i].value.asBytes.ptr);

    // clean up
    dpiVar_release(inIntColVar);
    dpiVar_release(outIntColVar);
    dpiVar_release(outStrColVar);
    dpiStmt_release(stmt);
    dpiConn_release(conn);

    printf("Done.\n");
    return 0;
}