| /********************************************************************** |
| // @@@ START COPYRIGHT @@@ |
| // |
| // Licensed to the Apache Software Foundation (ASF) under one |
| // or more contributor license agreements. See the NOTICE file |
| // distributed with this work for additional information |
| // regarding copyright ownership. The ASF licenses this file |
| // to you under the Apache License, Version 2.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.apache.org/licenses/LICENSE-2.0 |
| // |
| // Unless required by applicable law or agreed to in writing, |
| // software distributed under the License is distributed on an |
| // "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY |
| // KIND, either express or implied. See the License for the |
| // specific language governing permissions and limitations |
| // under the License. |
| // |
| // @@@ END COPYRIGHT @@@ |
| **********************************************************************/ |
| /* -*-C++-*- |
| **************************************************************************** |
| * |
| * File: clitest.cpp |
| * Description: Test driver useing exe util cli interface |
| * |
| * |
| * |
| * |
| **************************************************************************** |
| */ |
| #include "Platform.h" |
| |
| |
| #include <iostream> |
| #include <stdio.h> |
| #include <stdlib.h> |
| #include "BaseTypes.h" |
| #include "NAAssert.h" |
| #include "stdlib.h" |
| #include "stdio.h" |
| #include "sqlcli.h" |
| #include "ComDiags.h" |
| #include "ex_stdh.h" |
| #include "memorymonitor.h" |
| #include "ex_exe_stmt_globals.h" |
| #include "ex_esp_frag_dir.h" |
| #include "ComTdb.h" |
| #include "ex_tcb.h" |
| #include "ex_split_bottom.h" |
| #include "ex_send_bottom.h" |
| #include "NAExit.h" |
| #include "ExSqlComp.h" |
| #include "Globals.h" |
| #include "Int64.h" |
| #include "SqlStats.h" |
| #include "ComUser.h" |
| #include "ExpError.h" |
| #include "ComSqlId.h" |
| #include "ex_globals.h" |
| #include "ex_tcb.h" |
| #include "ExExeUtil.h" |
| #include "Globals.h" |
| #include "Context.h" |
| #include "blobtest.h" |
| |
| //DEFINE_DOVERS(clitestdriver) |
| // |
| int main(int argc, const char * argv[]) |
| { |
| |
| if ((argc < 2) || (argc > 2)) |
| { |
| cout << "Error - provide an option: " << endl; |
| cout << "Usage : clitestdriver <option number>" << endl; |
| cout << "TEST OPTION "<<endl; |
| cout << "-------------------------------------------"<<endl; |
| cout << "Blob test extract to buffer 1 "<< endl; |
| cout << "Blob test extract to file in chunks 2 "<< endl; |
| cout << "Blob test to insert to lob column 3 "<< endl; |
| cout << "Blob test to update lob column 4 "<< endl; |
| cout << "Blob test to append to lob column 5 "<< endl; |
| cout << "Blob test to update to lob lobhandle 6 "<< endl; |
| cout << "Blob test to append to lobhandle 7 "<< endl; |
| cout << "Blob test to truncate to lobhandle 8 "<< endl; |
| cout << "Blob test append lobhandle from a buffer in chunks from file 9 " << endl; |
| return 0; |
| } |
| Int32 retcode = 0; |
| SQLCTX_HANDLE defContext = 0; |
| Lng32 retCode = SQL_EXEC_CreateContext(&defContext, NULL, 0); |
| |
| int option = atoi(argv[1]); |
| |
| if(retcode != 0 ) |
| cout << "Error creating a CLI context error " << endl; |
| |
| |
| CliGlobals * cliGlob = GetCliGlobals(); |
| char tablename[50] = {'\0'}; |
| char columnname[50] = {'\0'}; |
| char filename[50] = {'\0'}; |
| switch (option) |
| { |
| case 1: |
| { |
| cout <<"***********" <<endl; |
| cout << " Blob test extract to user buffer " << endl; |
| cout <<"***********" <<endl; |
| cout << "Extract from a lob column in a lob table" << endl << endl; |
| cout << "Input lob table name :" << endl; |
| cin.getline(tablename,40); |
| cout << "Table name : " << tablename << endl; |
| cout << "Input lob column name to extract from :" << endl; |
| cin.getline(columnname,40); |
| cout << "Column Name : " << columnname << endl; |
| |
| //extract lob handle |
| |
| char *lobHandle = new char[1024]; |
| str_cpy_all(lobHandle," ",1024); |
| cout << "Extracting lob handle for column " << columnname << "..." << endl; |
| retcode = extractLobHandle(cliGlob, lobHandle, (char *)columnname,(char *)tablename); |
| if (retcode) |
| { |
| cout << "extractLobHandle returned " << retcode <<endl; |
| |
| delete lobHandle; |
| return retcode; |
| } |
| cout << "LOB handle for "<< columnname << ": " << lobHandle << endl; |
| //extract length of lob column from a table with 1 lob column. |
| |
| cout << "Extracting LOB data length for the above handle..." << endl; |
| Int64 lengthOfLob= 0; |
| retCode = extractLengthOfLobColumn(cliGlob, lobHandle, lengthOfLob,columnname,(char *)tablename); |
| if (retcode) |
| { |
| cout << "extractLengthOfLobColumn returned " << retcode <<endl; |
| |
| delete lobHandle; |
| return retcode; |
| } |
| cout << "LOB data length :" << lengthOfLob << endl; |
| //extract lob data to buffer |
| cout << "Extracting lob data into user buffer in a loop ..." << endl; |
| retCode = extractLobToBuffer(cliGlob,lobHandle, lengthOfLob,columnname,tablename); |
| if (retcode) |
| { |
| cout << "extractLobToBuffer returned " << retcode <<endl; |
| |
| delete lobHandle; |
| return retcode; |
| } |
| delete lobHandle; |
| return retcode; |
| } |
| |
| break; |
| |
| case 2: |
| { |
| cout <<"*************************************" <<endl; |
| cout << "Blob test extract to file in chunks " << endl; |
| cout <<"************************************" <<endl; |
| cout << "Extract from a lob column in a lob table" << endl << endl; |
| cout << "Input lob table name :" << endl; |
| cin.getline(tablename,40); |
| cout << "Table name : " << tablename << endl; |
| cout << "Input lob column name to extract from :" << endl; |
| cin.getline(columnname,40); |
| cout << "Column Name : " << columnname << endl; |
| cout <<"Input a filename to extract to : " << endl; |
| cin.getline(filename,40); |
| cout << "Output File Name : " << filename << endl; |
| |
| //extract lob handle |
| |
| char *lobHandle = new char[1024]; |
| str_cpy_all(lobHandle," ",1024); |
| cout << "Extracting lob handle for column " << columnname << "..." << endl; |
| retcode = extractLobHandle(cliGlob, lobHandle, (char *)columnname,tablename); |
| if (retcode) |
| { |
| cout << "extractLobHandle returned " << retcode <<endl; |
| |
| delete lobHandle; |
| return retcode; |
| } |
| cout << "LOB handle for "<< columnname << ": " << lobHandle << endl; |
| //extract length of lob column from a table with 1 lob column. |
| |
| cout << "Extracting LOB data length for the above handle..." << endl; |
| Int64 lengthOfLob= 0; |
| retCode = extractLengthOfLobColumn(cliGlob, lobHandle, lengthOfLob,columnname,tablename); |
| if (retcode) |
| { |
| cout << "extractLengthOfLobColumn returned " << retcode <<endl; |
| |
| delete lobHandle; |
| return retcode; |
| } |
| cout << "LOB data length :" << lengthOfLob << endl; |
| //extract lob data to buffer |
| cout << "Extracting lob data into file in chunks ..." << endl; |
| retCode = extractLobToFileInChunks(cliGlob,lobHandle, filename, lengthOfLob,columnname,tablename); |
| if (retcode) |
| { |
| cout << "extractLobToFileInChunks returned " << retcode <<endl; |
| |
| delete lobHandle; |
| return retcode; |
| } |
| delete lobHandle; |
| return retcode; |
| } |
| |
| break; |
| |
| case 3: |
| { |
| |
| cout <<"*************************************" <<endl; |
| cout << "Blob test insert lobdata from a buffer " << endl; |
| cout << "Input lob table name (1st int column, 2nd blob column:" << endl; |
| cin.getline(tablename,40); |
| cout << "Table name : " << tablename << endl; |
| retcode = insertBufferToLob(cliGlob,tablename); |
| return retcode; |
| } |
| break; |
| case 4: |
| { |
| |
| cout <<"*************************************" <<endl; |
| cout << "Blob test update lobdata from a buffer " << endl; |
| cout << "Input lob table name :" << endl; |
| cin.getline(tablename,40); |
| cout << "Table name (1st int column ,2nd blob column): " << tablename << endl; |
| cout << "Input lob column name to update :" << endl; |
| cin.getline(columnname,40); |
| cout << "Column Name : " << columnname << endl; |
| retcode = updateBufferToLob(cliGlob,tablename,columnname); |
| return retcode; |
| } |
| break; |
| case 5: |
| { |
| |
| cout <<"*************************************" <<endl; |
| cout << "Blob test update append lobdata from a buffer " << endl; |
| cout << "Input lob table name :" << endl; |
| cin.getline(tablename,40); |
| cout << "Table name (1st int column , 2nd blob column: " << tablename << endl; |
| cout << "Input lob column name to update :" << endl; |
| cin.getline(columnname,40); |
| cout << "Column Name : " << columnname << endl; |
| retcode = updateAppendBufferToLob(cliGlob,tablename,columnname); |
| return retcode; |
| } |
| break; |
| case 6: |
| { |
| |
| cout <<"*************************************" <<endl; |
| cout << "Blob test update lobhandle from a buffer " << endl; |
| cout << "Input lob table name :" << endl; |
| cin.getline(tablename,40); |
| cout << "Table name : " << tablename << endl; |
| cout << "Input lob column name to get handle from :" << endl; |
| cin.getline(columnname,40); |
| cout << "Column Name : " << columnname << endl; |
| |
| //extract lob handle |
| |
| char *lobHandle = new char[1024]; |
| str_cpy_all(lobHandle," ",1024); |
| cout << "Extracting lob handle for column " << columnname << "..." << endl; |
| retcode = extractLobHandle(cliGlob, lobHandle, (char *)columnname,tablename); |
| if (retcode) |
| { |
| cout << "extractLobHandle returned " << retcode <<endl; |
| |
| delete lobHandle; |
| return retcode; |
| } |
| cout << "LOB handle for "<< columnname << ": " << lobHandle << endl; |
| retcode = updateBufferToLobHandle(cliGlob,lobHandle); |
| return retcode; |
| } |
| break; |
| case 7: |
| { |
| |
| cout <<"*************************************" <<endl; |
| cout << "Blob test append lobhandle from a buffer " << endl; |
| cout << "Input lob table name :" << endl; |
| cin.getline(tablename,40); |
| cout << "Table name : " << tablename << endl; |
| cout << "Input lob column name to get handle from :" << endl; |
| cin.getline(columnname,40); |
| cout << "Column Name : " << columnname << endl; |
| |
| //extract lob handle |
| |
| char *lobHandle = new char[1024]; |
| str_cpy_all(lobHandle," ",1024); |
| cout << "Extracting lob handle for column " << columnname << "..." << endl; |
| retcode = extractLobHandle(cliGlob, lobHandle, (char *)columnname,tablename); |
| if (retcode) |
| { |
| cout << "extractLobHandle returned " << retcode <<endl; |
| |
| delete lobHandle; |
| return retcode; |
| } |
| cout << "LOB handle for "<< columnname << ": " << lobHandle << endl; |
| char *source= new char[300]; |
| strcpy(source,"zzzzzzzzzzzzzzzzzzzz"); |
| retcode = updateAppendBufferToLobHandle(cliGlob,lobHandle,20,(Int64)source); |
| return retcode; |
| } |
| break; |
| case 8: |
| { |
| |
| cout <<"*************************************" <<endl; |
| cout << "Blob test empty lobhandle from a buffer " << endl; |
| cout << "Input lob table name :" << endl; |
| cin.getline(tablename,40); |
| cout << "Table name : " << tablename << endl; |
| cout << "Input lob column name to get handle from :" << endl; |
| cin.getline(columnname,40); |
| cout << "Column Name : " << columnname << endl; |
| |
| //extract lob handle |
| |
| char *lobHandle = new char[1024]; |
| str_cpy_all(lobHandle," ",1024); |
| cout << "Extracting lob handle for column " << columnname << "..." << endl; |
| retcode = extractLobHandle(cliGlob, lobHandle, (char *)columnname,tablename); |
| if (retcode) |
| { |
| cout << "extractLobHandle returned " << retcode <<endl; |
| |
| delete lobHandle; |
| return retcode; |
| } |
| cout << "LOB handle for "<< columnname << ": " << lobHandle << endl; |
| retcode = updateTruncateLobHandle(cliGlob,lobHandle); |
| return retcode; |
| } |
| break; |
| |
| case 9: |
| { |
| |
| cout <<"*************************************" <<endl; |
| cout << "Blob test append lobhandle from a buffer from file " << endl; |
| cout << "Input lob table name :" << endl; |
| cin.getline(tablename,40); |
| cout << "Table name : " << tablename << endl; |
| cout << "Input lob column name to get handle from :" << endl; |
| cin.getline(columnname,40); |
| cout << "Column Name : " << columnname << endl; |
| cout << "Input source file name :" << endl; |
| cin.getline(filename,512); |
| cout << "Source name : " << filename << endl; |
| //extract lob handle |
| |
| char *lobHandle = new char[1024]; |
| str_cpy_all(lobHandle," ",1024); |
| cout << "Extracting lob handle for column " << columnname << "..." << endl; |
| retcode = extractLobHandle(cliGlob, lobHandle, (char *)columnname,tablename); |
| if (retcode) |
| { |
| cout << "extractLobHandle returned " << retcode <<endl; |
| |
| delete lobHandle; |
| return retcode; |
| } |
| cout << "LOB handle for "<< columnname << ": " << lobHandle << endl; |
| |
| |
| //read from input source file in chunks and append |
| int fdSourceFile = open(filename,O_RDONLY); |
| struct stat statbuf; |
| if (stat(filename, &statbuf) != 0) { |
| return -1; |
| } |
| |
| Int64 sourceEOF = statbuf.st_size; |
| char *fileInputData= new char[32*1024*1024]; //32 KB chunks |
| //char *fileInputData = new char[10000]; |
| Int64 lengthRead=0; |
| Int64 offset=0; |
| Int64 chunkLen = 32*1024*1024; |
| //Int64 chunkLen = 10000; |
| |
| while ((offset < sourceEOF) ) |
| { |
| if (lengthRead=pread(fdSourceFile, fileInputData, chunkLen, offset) == -1) { |
| close(fdSourceFile); |
| return -1; |
| } |
| chunkLen = min(chunkLen,sourceEOF-offset); |
| retcode = updateAppendBufferToLobHandle(cliGlob,lobHandle,chunkLen,(Int64)fileInputData); |
| offset +=chunkLen; |
| } |
| return retcode; |
| } |
| break; |
| }//end switch |
| |
| return 0; |
| |
| } |
| |
| |
