<!DOCTYPE HTML PUBLIC "-//W3C//DTD HTML 4.01 Transitional//EN">
<html><head><meta http-equiv="Content-Type" content="text/html;charset=UTF-8">
<title>DTN Reference Implementation: PostgresSQLImplementation.cc Source File</title>
<link href="tabs.css" rel="stylesheet" type="text/css">
<link href="doxygen.css" rel="stylesheet" type="text/css">
</head><body>
<!-- Generated by Doxygen 1.5.8 -->
<div class="navigation" id="top">
<div class="tabs">
<ul>
<li><a href="main.html"><span>Main Page</span></a></li>
<li><a href="namespaces.html"><span>Namespaces</span></a></li>
<li><a href="annotated.html"><span>Classes</span></a></li>
<li class="current"><a href="files.html"><span>Files</span></a></li>
</ul>
</div>
<div class="tabs">
<ul>
<li><a href="files.html"><span>File List</span></a></li>
<li><a href="globals.html"><span>File Members</span></a></li>
</ul>
</div>
<h1>PostgresSQLImplementation.cc</h1><a href="PostgresSQLImplementation_8cc.html">Go to the documentation of this file.</a><div class="fragment"><pre class="fragment"><a name="l00001"></a>00001 <span class="comment">/*</span>
<a name="l00002"></a>00002 <span class="comment"> * Copyright 2004-2006 Intel Corporation</span>
<a name="l00003"></a>00003 <span class="comment"> * </span>
<a name="l00004"></a>00004 <span class="comment"> * Licensed under the Apache License, Version 2.0 (the "License");</span>
<a name="l00005"></a>00005 <span class="comment"> * you may not use this file except in compliance with the License.</span>
<a name="l00006"></a>00006 <span class="comment"> * You may obtain a copy of the License at</span>
<a name="l00007"></a>00007 <span class="comment"> * </span>
<a name="l00008"></a>00008 <span class="comment"> * http://www.apache.org/licenses/LICENSE-2.0</span>
<a name="l00009"></a>00009 <span class="comment"> * </span>
<a name="l00010"></a>00010 <span class="comment"> * Unless required by applicable law or agreed to in writing, software</span>
<a name="l00011"></a>00011 <span class="comment"> * distributed under the License is distributed on an "AS IS" BASIS,</span>
<a name="l00012"></a>00012 <span class="comment"> * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.</span>
<a name="l00013"></a>00013 <span class="comment"> * See the License for the specific language governing permissions and</span>
<a name="l00014"></a>00014 <span class="comment"> * limitations under the License.</span>
<a name="l00015"></a>00015 <span class="comment"> */</span>
<a name="l00016"></a>00016
<a name="l00017"></a>00017 <span class="preprocessor">#ifdef HAVE_CONFIG_H</span>
<a name="l00018"></a>00018 <span class="preprocessor"></span><span class="preprocessor"># include <<a class="code" href="dtn-config_8h.html">dtn-config.h</a>></span>
<a name="l00019"></a>00019 <span class="preprocessor">#endif</span>
<a name="l00020"></a>00020 <span class="preprocessor"></span>
<a name="l00021"></a>00021 <span class="preprocessor">#if POSTGRES_ENABLED</span>
<a name="l00022"></a>00022 <span class="preprocessor"></span>
<a name="l00023"></a>00023 <span class="preprocessor">#include <string.h></span>
<a name="l00024"></a>00024 <span class="preprocessor">#include <oasys/debug/DebugUtils.h></span>
<a name="l00025"></a>00025 <span class="preprocessor">#include <oasys/util/StringBuffer.h></span>
<a name="l00026"></a>00026 <span class="preprocessor">#include "<a class="code" href="PostgresSQLImplementation_8h.html">PostgresSQLImplementation.h</a>"</span>
<a name="l00027"></a>00027
<a name="l00028"></a>00028 <span class="keyword">namespace </span>dtn {
<a name="l00029"></a>00029
<a name="l00030"></a>00030 <a class="code" href="classdtn_1_1PostgresSQLImplementation.html#319849abe65b627c99f76acfcffc2dc0">PostgresSQLImplementation::PostgresSQLImplementation</a>()
<a name="l00031"></a>00031 : SQLImplementation(<span class="stringliteral">"BYTEA"</span>, <span class="stringliteral">"BOOLEAN"</span>),
<a name="l00032"></a>00032 <a class="code" href="classoasys_1_1Logger.html">Logger</a>(<span class="stringliteral">"/storage/postgresql"</span>)
<a name="l00033"></a>00033 {
<a name="l00034"></a>00034 query_result_ = NULL;
<a name="l00035"></a>00035 }
<a name="l00036"></a>00036
<a name="l00037"></a>00037 <span class="keywordtype">int</span>
<a name="l00038"></a>00038 PostgresSQLImplementation::connect(<span class="keyword">const</span> <span class="keywordtype">char</span>* dbName)
<a name="l00039"></a>00039 {
<a name="l00040"></a>00040 <span class="keywordtype">char</span> *pghost;
<a name="l00041"></a>00041 <span class="keywordtype">char</span> *pgport;
<a name="l00042"></a>00042 <span class="keywordtype">char</span> *pgoptions;
<a name="l00043"></a>00043 <span class="keywordtype">char</span> *pgtty;
<a name="l00044"></a>00044
<a name="l00045"></a>00045 log_debug(<span class="stringliteral">"connecting to database %s"</span>, dbName);
<a name="l00046"></a>00046
<a name="l00047"></a>00047 <span class="comment">/*</span>
<a name="l00048"></a>00048 <span class="comment"> * begin, by setting the parameters for a backend connection if</span>
<a name="l00049"></a>00049 <span class="comment"> * the parameters are null, then the system will try to use</span>
<a name="l00050"></a>00050 <span class="comment"> * reasonable defaults by looking up environment variables or,</span>
<a name="l00051"></a>00051 <span class="comment"> * failing that, using hardwired constants</span>
<a name="l00052"></a>00052 <span class="comment"> */</span>
<a name="l00053"></a>00053
<a name="l00054"></a>00054 pghost = NULL; <span class="comment">/* host name of the backend server */</span>
<a name="l00055"></a>00055 pgport = NULL; <span class="comment">/* port of the backend server */</span>
<a name="l00056"></a>00056 pgoptions = NULL; <span class="comment">/* special options to start up the backend</span>
<a name="l00057"></a>00057 <span class="comment"> * server */</span>
<a name="l00058"></a>00058 pgtty = NULL; <span class="comment">/* debugging tty for the backend server */</span>
<a name="l00059"></a>00059
<a name="l00065"></a>00065 db_ = PQsetdb(pghost, pgport, pgoptions, pgtty, dbName);
<a name="l00066"></a>00066
<a name="l00070"></a>00070 <span class="keywordflow">if</span> (PQstatus(db_) == CONNECTION_BAD)
<a name="l00071"></a>00071 {
<a name="l00072"></a>00072 log_err(<span class="stringliteral">"connection to database '%s' failed: %s"</span>,
<a name="l00073"></a>00073 dbName, PQerrorMessage(db_));
<a name="l00074"></a>00074 <span class="keywordflow">return</span> -1;
<a name="l00075"></a>00075 }
<a name="l00076"></a>00076
<a name="l00077"></a>00077 <span class="keywordflow">return</span> 0;
<a name="l00078"></a>00078 }
<a name="l00079"></a>00079
<a name="l00080"></a>00080 <span class="keywordtype">int</span>
<a name="l00081"></a>00081 PostgresSQLImplementation::close()
<a name="l00082"></a>00082 {
<a name="l00083"></a>00083 PQfinish(db_);
<a name="l00084"></a>00084 <span class="keywordflow">return</span> 0;
<a name="l00085"></a>00085 }
<a name="l00086"></a>00086
<a name="l00087"></a>00087 <span class="comment">/*</span>
<a name="l00088"></a>00088 <span class="comment">size_t</span>
<a name="l00089"></a>00089 <span class="comment">PostgresSQLImplementation::get_value_length(int tuple_no, int field_no)</span>
<a name="l00090"></a>00090 <span class="comment">{</span>
<a name="l00091"></a>00091 <span class="comment"> size_t retval;</span>
<a name="l00092"></a>00092 <span class="comment"> const char* val = get_value(tuple_no,field_no);</span>
<a name="l00093"></a>00093 <span class="comment"> unescape_binary((const u_char*)val,&retval);</span>
<a name="l00094"></a>00094 <span class="comment"> return retval;</span>
<a name="l00095"></a>00095 <span class="comment">}</span>
<a name="l00096"></a>00096 <span class="comment">*/</span>
<a name="l00097"></a>00097
<a name="l00098"></a>00098 <span class="keyword">const</span> <span class="keywordtype">char</span>*
<a name="l00099"></a>00099 PostgresSQLImplementation::get_value(<span class="keywordtype">int</span> tuple_no, <span class="keywordtype">int</span> field_no)
<a name="l00100"></a>00100 {
<a name="l00101"></a>00101 <span class="keyword">const</span> <span class="keywordtype">char</span>* ret ;
<a name="l00102"></a>00102 <a class="code" href="SDNV_8cc.html#ca68c0d4ac8df0838e209fb5300f7be3">ASSERT</a>(query_result_);
<a name="l00103"></a>00103 ret = PQgetvalue(query_result_, tuple_no, field_no);
<a name="l00104"></a>00104 <span class="keywordflow">return</span> ret;
<a name="l00105"></a>00105 }
<a name="l00106"></a>00106
<a name="l00107"></a>00107 <span class="keywordtype">bool</span>
<a name="l00108"></a>00108 PostgresSQLImplementation::has_table(<span class="keyword">const</span> <span class="keywordtype">char</span>* tablename)
<a name="l00109"></a>00109 {
<a name="l00110"></a>00110 <span class="keywordtype">bool</span> retval = 0;
<a name="l00111"></a>00111 oasys::StringBuffer query;
<a name="l00112"></a>00112
<a name="l00113"></a>00113 query.appendf(<span class="stringliteral">"select * from pg_tables where tablename = '%s'"</span>, tablename);
<a name="l00114"></a>00114 <span class="keywordtype">int</span> ret = exec_query(query.c_str());
<a name="l00115"></a>00115 <a class="code" href="SDNV_8cc.html#ca68c0d4ac8df0838e209fb5300f7be3">ASSERT</a>(ret == 0);
<a name="l00116"></a>00116 <span class="keywordflow">if</span> (num_tuples() == 1) retval = 1;
<a name="l00117"></a>00117
<a name="l00118"></a>00118 <span class="keywordflow">return</span> retval;
<a name="l00119"></a>00119 }
<a name="l00120"></a>00120
<a name="l00121"></a>00121 <span class="keywordtype">int</span>
<a name="l00122"></a>00122 PostgresSQLImplementation::num_tuples()
<a name="l00123"></a>00123 {
<a name="l00124"></a>00124 <span class="keywordtype">int</span> ret = -1;
<a name="l00125"></a>00125 <a class="code" href="SDNV_8cc.html#ca68c0d4ac8df0838e209fb5300f7be3">ASSERT</a>(query_result_);
<a name="l00126"></a>00126 ret = PQntuples(query_result_);
<a name="l00127"></a>00127 <span class="keywordflow">return</span> ret;
<a name="l00128"></a>00128 }
<a name="l00129"></a>00129
<a name="l00130"></a>00130 <span class="keyword">static</span> <span class="keywordtype">int</span>
<a name="l00131"></a>00131 status_to_int(ExecStatusType t)
<a name="l00132"></a>00132 {
<a name="l00133"></a>00133 <span class="keywordflow">if</span> (t == PGRES_COMMAND_OK) <span class="keywordflow">return</span> 0;
<a name="l00134"></a>00134 <span class="keywordflow">if</span> (t == PGRES_TUPLES_OK) <span class="keywordflow">return</span> 0;
<a name="l00135"></a>00135 <span class="keywordflow">if</span> (t == PGRES_EMPTY_QUERY) <span class="keywordflow">return</span> 0;
<a name="l00136"></a>00136 <span class="keywordflow">return</span> -1;
<a name="l00137"></a>00137 }
<a name="l00138"></a>00138
<a name="l00139"></a>00139 <span class="keywordtype">int</span>
<a name="l00140"></a>00140 PostgresSQLImplementation::exec_query(<span class="keyword">const</span> <span class="keywordtype">char</span>* query)
<a name="l00141"></a>00141 {
<a name="l00142"></a>00142 <span class="keywordtype">int</span> ret = -1;
<a name="l00143"></a>00143
<a name="l00144"></a>00144 <span class="keywordflow">if</span> (query_result_ != NULL) {
<a name="l00145"></a>00145 PQclear(query_result_);
<a name="l00146"></a>00146 query_result_ = NULL;
<a name="l00147"></a>00147 }
<a name="l00148"></a>00148
<a name="l00149"></a>00149 query_result_ = PQexec(db_, query);
<a name="l00150"></a>00150 <a class="code" href="SDNV_8cc.html#ca68c0d4ac8df0838e209fb5300f7be3">ASSERT</a>(query_result_);
<a name="l00151"></a>00151 ExecStatusType t = PQresultStatus(query_result_);
<a name="l00152"></a>00152 ret = status_to_int(t);
<a name="l00153"></a>00153
<a name="l00154"></a>00154 <span class="keywordflow">return</span> ret;
<a name="l00155"></a>00155 }
<a name="l00156"></a>00156
<a name="l00157"></a>00157 <span class="keyword">const</span> <span class="keywordtype">char</span>*
<a name="l00158"></a>00158 PostgresSQLImplementation::escape_string(<span class="keyword">const</span> <span class="keywordtype">char</span>* from)
<a name="l00159"></a>00159 {
<a name="l00160"></a>00160 <span class="keywordtype">int</span> length = strlen(from);
<a name="l00161"></a>00161 <span class="keywordtype">char</span>* to = (<span class="keywordtype">char</span> *) malloc(2*length+1);
<a name="l00162"></a>00162 PQescapeString (to,from,length);
<a name="l00163"></a>00163 <span class="keywordflow">return</span> to;
<a name="l00164"></a>00164 }
<a name="l00165"></a>00165
<a name="l00166"></a>00166 <span class="keyword">const</span> u_char*
<a name="l00167"></a>00167 PostgresSQLImplementation::escape_binary(<span class="keyword">const</span> u_char* from, <span class="keywordtype">int</span> from_length)
<a name="l00168"></a>00168 {
<a name="l00169"></a>00169 <span class="keywordtype">size_t</span> to_length;
<a name="l00170"></a>00170 u_char* from1 = (u_char *) from ;
<a name="l00171"></a>00171 u_char* to = PQescapeBytea(from1,from_length,&to_length);
<a name="l00172"></a>00172 <span class="keywordflow">return</span> to;
<a name="l00173"></a>00173 }
<a name="l00174"></a>00174
<a name="l00175"></a>00175 <span class="keyword">const</span> u_char*
<a name="l00176"></a>00176 PostgresSQLImplementation::unescape_binary(<span class="keyword">const</span> u_char* from)
<a name="l00177"></a>00177 {
<a name="l00178"></a>00178 u_char* from1 = (u_char *) from ;
<a name="l00179"></a>00179 <span class="keywordtype">size_t</span> to_length ;
<a name="l00180"></a>00180 <span class="keyword">const</span> u_char* to = PQunescapeBytea(from1,&to_length);
<a name="l00181"></a>00181 <span class="keywordflow">return</span> to;
<a name="l00182"></a>00182 }
<a name="l00183"></a>00183
<a name="l00184"></a>00184 } <span class="comment">// namespace dtn</span>
<a name="l00185"></a>00185
<a name="l00186"></a>00186 <span class="preprocessor">#endif </span><span class="comment">/* POSTGRES_ENABLED */</span>
</pre></div></div>
<hr size="1"><address style="text-align: right;"><small>Generated on Fri Jan 30 09:26:57 2009 for DTN Reference Implementation by
<a href="http://www.doxygen.org/index.html">
<img src="doxygen.png" alt="doxygen" align="middle" border="0"></a> 1.5.8 </small></address>
</body>
</html>
distrib
>
Mandriva
>
2010.0
>
i586
>
media
>
contrib-release
>
by-pkgid
>
29c369053a983cc5d8c880dc6e0a2237
>
files
>
466
