用otl写的oracle取数工具，执行传入在查询语句，把结果输出到文件

Posted 2020-08-12

项目中经常需要用到此类型的工具

 

  1 #pragma warning (disable:4786)
  2 #include <iostream>
  3 #include <map>
  4 #include <string>
  5 using namespace std;
  6 #include <stdio.h>
  7 #include <stdlib.h>
  8 #include <algorithm>
  9 #define OTL_ORA9I // Compile OTL 4.0/OCI9i
 10 #define OTL_STREAM_READ_ITERATOR_ON
 11 #define OTL_STL
 12 
 13 #ifdef WIN32
 14         #include <conio.h>
 15 #else
 16         //#define PASS_SECURITY
 17 #endif
 18 
 19 #ifdef PASS_SECURITY
 20         #include <pass_security/TDBCrypt.h> //包含密码解析头文件
 21 #else
 22         #define DB_ENV_LEN 100
 23 #endif
 24 
 25 #include "otlv4.h" // include the OTL 4 header file
 26 
 27 
 28 #define OTL_BUFFER_SIZE 2000 //otl缓冲行数
 29 #define BUFFER_SIZE 2000 //列内容缓冲
 30 
 31 
 32 
 33 int select(otl_connect& db, const string &sql, FILE *fp, const string &delemer = "|") //获得sql返回，并以|为分隔符输出到文件（第一行是列名）
 34 { 
 35 
 36         // create select stream
 37         otl_stream stream_i;
 38         //将输出类型全部转成string
 39         stream_i.set_all_column_types(otl_all_date2str | otl_all_num2str);
 40 
 41         stream_i.open(OTL_BUFFER_SIZE, // buffer size
 42                 sql.c_str(),
 43                 // SELECT statement
 44                 db // connect object
 45                 ); 
 46         
 47         //文件头
 48         int desc_len;
 49         otl_column_desc *desc = stream_i.describe_select(desc_len);
 50         for(int n = 0; n < desc_len; ++n){
 51                 fprintf(fp, "%s%s", desc[n].name, delemer.c_str());
 52         }
 53         fprintf(fp, "\n");
 54 
 55         //文件内容
 56         string content;
 57         int row_count = 0;
 58 
 59         while (!stream_i.eof())
 60         {
 61                 for(int n = 0; n < desc_len; ++n){
 62                         stream_i >> content;
 63                         fprintf(fp, "%s%s", content.c_str(), delemer.c_str());
 64                 }
 65                 fprintf(fp, "\n");
 66                 row_count++;
 67         }
 68 
 69         cerr << row_count << " rows loaded " << endl;
 70 
 71         return 0;
 72 }
 73 
 74 
 75 int get_passwd(string& io_dbname) //通过用户名获得 user/[email protected] 形式字符串
 76 {
 77         
 78         if (string::npos != io_dbname.find("/"))
 79         {
 80                 return 0;
 81         }
 82 
 83         
 84 
 85 #ifdef PASS_SECURITY
 86         std::string strDBConnect = io_dbname;
 87 
 88         char sDB[DB_ENV_LEN + 1],sUSR[DB_ENV_LEN + 1],sPWD[DB_ENV_LEN + 1];
 89         security::TDBCrypt Crypt;
 90         int iresult = Crypt.init();
 91         if (iresult) 
 92         {
 93                 cerr << "get passfile failed" << std::endl;
 94                 return 1;
 95         }
 96 
 97         if (Crypt.fetchUserPwd(io_dbname.c_str(), sUSR, sPWD, sDB)) 
 98         {
 99                 io_dbname = "";
100                 io_dbname = io_dbname + sUSR + "/" + sPWD + "@" + sDB;
101                 return 0;
102         }
103         else
104         {
105                 cerr << "get password failed" << std::endl;
106                 return 1;
107         }
108 #else
109 
110 #ifdef PASS_SECURITY
111         size_t nDBpos = io_dbname.find("@");
112 
113         string sUSR = io_dbname.substr(0, nDBpos);
114         string sDB = io_dbname.substr(nDBpos + 1, io_dbname.length() - 1);
115         char sPWD[DB_ENV_LEN + 1] = "";
116 
117 
118         cerr << "please input password of "<< io_dbname << " : " << std::endl;
119 
120         int ch;
121         size_t i = 0;
122         while (((ch = getch()) != ‘\r‘) && i < DB_ENV_LEN)
123         {
124                 if (ch != ‘\b‘)
125                 {
126                         sPWD[i++] = (char) ch;
127                         cerr << "*";
128                 }       
129                 else
130                 {
131                         if (i >= 1)
132                         {
133                                 i--;
134                                 cerr << "\b \b";
135                         }
136                         else 
137                         {
138                                 cerr << "\007";
139                         }
140                 }
141         }
142         sPWD[i] = ‘\0‘;
143         cerr << endl;
144 
145         io_dbname = sUSR + "/" + sPWD + "@" + sDB;
146 #endif
147 
148 #endif
149 
150         return 0;
151 }
152 
153 
154 int main(const int argc, const char * const argv[])
155 {
156         if (argc < 3)
157         {
158                 cerr << "参数错误！用法示例:\t" << endl;
159                 cerr << "dtload user/[email protected] sql filename" << endl;
160                 return 0;
161         }
162         
163         map<string, string> mpParam;
164         for (int i = 1; i < argc; i++)
165         {
166             if (*argv[i] == ‘-‘)
167             {
168                 if (i < argc)
169                 {
170                     mpParam[argv[i]] = argv[i + 1];
171                 }
172                 else
173                 {
174                     mpParam[argv[i]] = "";
175                 }
176             }
177         }
178 
179         otl_connect db;
180 
181         string dbname = argv[1];
182         string sql = argv[2];
183         string file = "";
184         FILE *fp = NULL;
185         
186         //如果输出文件名为空，则输出到屏幕
187         if (argc >= 4)
188         {
189                 
190                 file = argv[3];
191                 fp = fopen(file.c_str(), "w");
192                 
193                 if (fp == NULL)
194                 {
195                         cerr << "openfile " << file << " err! maybe can not writed\n ";
196                         return 0;
197                 } 
198         }
199         else
200         {
201                 fp = stdout;
202         }
203         
204         string delemer = "|";
205         if (argc >= 5)
206         {
207                 delemer = argv[4];
208         }
209 
210         //如果输入参数不包含密码，则从dbfile里面找密码
211         //get_passwd(dbname);
212 
213 
214         otl_connect::otl_initialize(); // initialize OCI environment
215 
216         try
217         {
218                 db.rlogon(dbname.c_str()); // connect to Oracle
219                 select(db, sql, fp, delemer); // select records from table              
220         }
221         catch(const otl_exception& p)
222         { // intercept OTL exceptions
223                 cerr<<p.msg<<endl; // print out error message
224                 cerr<<p.stm_text<<endl; // print out SQL that caused the error
225                 cerr<<p.var_info<<endl; // print out the variable that caused the error
226         }
227         
228         fclose(fp);
229         db.logoff(); // disconnect from Oracle
230         
231 }

 

Makefile

 1 ## -*- Makefile -*-
 2 ## by hch
 3 
 4 top_srcdir      = ../
 5 
 6 DISPATCH = $(bindir)/Dispatch
 7 
 8 TARGETS    =    tbasload
 9 
10 include $(top_srcdir)/MKFILE/Make.rules
11 
12 
13 OBJS        = tbasload.o
14 
15 CPPFLAGS        := 16         $(ORACLE_INCLUDE) 17         -I. 18         $(CPPFLAGS)
19 
20 LDFLAGS        := 21         $(LDFLAGS) 22         $(ORACLE_LDFLAGS) 
23 
24 tbasload: $(OBJS)
25     @echo "Creating executable -*- [email protected] -*- ..."
26     @-rm -f [email protected]
27     @$(CXX) -o [email protected] $(LDFLAGS) $(ORACLE_LIBS) $^ 

 

Make.rules

 1 # -*-Make.rules-*-
 2 # by szh
 3 
 4 buildbits ?= 32
 5 debug ?= 1
 6 
 7 STATICLIBS    = yes
 8 
 9 
10 VERSION        = 1.0.0
11 SOVERSION    = 10
12 ifeq ($(buildbits),64)
13     bindir        = $(top_srcdir)/bin
14     libdir        = $(top_srcdir)/lib
15 else
16     bindir        = $(top_srcdir)/bin32
17     libdir        = $(top_srcdir)/lib32
18 endif
19 includedir    = $(top_srcdir)/include/common
20 
21 UNAME                   := $(shell uname)
22 
23 include  $(top_srcdir)/MKFILE/Make.rules.$(UNAME)
24 include  $(top_srcdir)/MKFILE/Make.rules.Customized
25 
26 CPPFLAGS    = -I$(includedir) $(ORACLE_INCLUDE)
27 LDFLAGS        := $(LDFLAGS) $(CXXFLAGS) -L$(libdir)
28 
29 ifeq ($(mklibfilename),)
30     mklibfilename       = $(if $(2),lib$(1).so.$(2),lib$(1).so)
31 endif
32 
33 ifeq ($(mksoname),)
34     mksoname            = $(if $(2),lib$(1).so.$(2),lib$(1).so)
35 endif
36 
37 ifeq ($(mklibname),)
38     ifeq ($(STATICLIBS),yes)
39         mklibname       = lib$(1).a
40     else
41         mklibname       = lib$(1).so
42     endif
43 endif
44 
45 ifndef mklibtargets
46     ifeq ($(STATICLIBS),yes)
47         mklibtargets    = $(3)
48     else
49         mklibtargets    = $(1) $(2) $(3)
50     endif
51 endif
52 
53 ifeq ($(mkshlib),)
54     $(error You need to define mkshlib in Make.rules.$(UNAME))
55 endif
56 
57 EVERYTHING    =    all clean
58 
59 .SUFFIXES:
60 
61 %.o: %.cpp
62     @echo "  $<"
63     @$(CXX) -o [email protected] -c $(CPPFLAGS) $(CXXFLAGS) $<
64 
65 %.o: %.c
66     @echo "  $<"
67     @$(CC) -o [email protected] -c $(CPPFLAGS) $(CFLAGS) $<
68 
69 all:: warning $(SRCS) $(TARGETS)
70 
71 warning::
72     @echo 
73     @echo "CONFIGURATION: buildbits=$(buildbits) debug=$(debug) STATICLIBS=$(STATICLIBS)"
74     @echo 
75     
76 
77 clean::
78     @-rm -f $(TARGETS) $(OBJS)
79     @-rm -f core *.o version.h