| <!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Transitional//EN" "http://www.w3.org/TR/xhtml1/DTD/xhtml1-transitional.dtd"> |
| <html xmlns="http://www.w3.org/1999/xhtml"> |
| <head> |
| <meta http-equiv="Content-Type" content="text/xhtml;charset=UTF-8"/> |
| <title>MADlib: dt_utility.sql_in Source File</title> |
| |
| <link href="tabs.css" rel="stylesheet" type="text/css"/> |
| <link href="doxygen.css" rel="stylesheet" type="text/css" /> |
| <link href="navtree.css" rel="stylesheet" type="text/css"/> |
| <script type="text/javascript" src="jquery.js"></script> |
| <script type="text/javascript" src="resize.js"></script> |
| <script type="text/javascript" src="navtree.js"></script> |
| <script type="text/javascript"> |
| $(document).ready(initResizable); |
| </script> |
| <link href="search/search.css" rel="stylesheet" type="text/css"/> |
| <script type="text/javascript" src="search/search.js"></script> |
| <script type="text/javascript"> |
| $(document).ready(function() { searchBox.OnSelectItem(0); }); |
| </script> |
| <script src="../mathjax/MathJax.js"> |
| MathJax.Hub.Config({ |
| extensions: ["tex2jax.js", "TeX/AMSmath.js", "TeX/AMSsymbols.js"], |
| jax: ["input/TeX","output/HTML-CSS"], |
| }); |
| </script> |
| </head> |
| <body> |
| <div id="top"><!-- do not remove this div! --> |
| |
| |
| <div id="titlearea"> |
| <table cellspacing="0" cellpadding="0"> |
| <tbody> |
| <tr style="height: 56px;"> |
| |
| |
| <td style="padding-left: 0.5em;"> |
| <div id="projectname">MADlib |
|  <span id="projectnumber">0.6</span> <span style="font-size:10pt; font-style:italic"><a href="../latest/./dt__utility_8sql__in_source.html"> A newer version is available</a></span> |
| </div> |
| <div id="projectbrief">User Documentation</div> |
| </td> |
| |
| |
| |
| </tr> |
| </tbody> |
| </table> |
| </div> |
| |
| <!-- Generated by Doxygen 1.7.5.1 --> |
| <script type="text/javascript"> |
| var searchBox = new SearchBox("searchBox", "search",false,'Search'); |
| </script> |
| <script type="text/javascript" src="dynsections.js"></script> |
| <div id="navrow1" class="tabs"> |
| <ul class="tablist"> |
| <li><a href="index.html"><span>Main Page</span></a></li> |
| <li><a href="modules.html"><span>Modules</span></a></li> |
| <li class="current"><a href="files.html"><span>Files</span></a></li> |
| <li> |
| <div id="MSearchBox" class="MSearchBoxInactive"> |
| <span class="left"> |
| <img id="MSearchSelect" src="search/mag_sel.png" |
| onmouseover="return searchBox.OnSearchSelectShow()" |
| onmouseout="return searchBox.OnSearchSelectHide()" |
| alt=""/> |
| <input type="text" id="MSearchField" value="Search" accesskey="S" |
| onfocus="searchBox.OnSearchFieldFocus(true)" |
| onblur="searchBox.OnSearchFieldFocus(false)" |
| onkeyup="searchBox.OnSearchFieldChange(event)"/> |
| </span><span class="right"> |
| <a id="MSearchClose" href="javascript:searchBox.CloseResultsWindow()"><img id="MSearchCloseImg" border="0" src="search/close.png" alt=""/></a> |
| </span> |
| </div> |
| </li> |
| </ul> |
| </div> |
| <div id="navrow2" class="tabs2"> |
| <ul class="tablist"> |
| <li><a href="files.html"><span>File List</span></a></li> |
| <li><a href="globals.html"><span>File Members</span></a></li> |
| </ul> |
| </div> |
| </div> |
| <div id="side-nav" class="ui-resizable side-nav-resizable"> |
| <div id="nav-tree"> |
| <div id="nav-tree-contents"> |
| </div> |
| </div> |
| <div id="splitbar" style="-moz-user-select:none;" |
| class="ui-resizable-handle"> |
| </div> |
| </div> |
| <script type="text/javascript"> |
| initNavTree('dt__utility_8sql__in.html',''); |
| </script> |
| <div id="doc-content"> |
| <div class="header"> |
| <div class="headertitle"> |
| <div class="title">dt_utility.sql_in</div> </div> |
| </div> |
| <div class="contents"> |
| <a href="dt__utility_8sql__in.html">Go to the documentation of this file.</a><div class="fragment"><pre class="fragment"><a name="l00001"></a>00001 <span class="comment">/* ----------------------------------------------------------------------- */</span><span class="comment">/** </span> |
| <a name="l00002"></a>00002 <span class="comment"> *</span> |
| <a name="l00003"></a>00003 <span class="comment"> * @file dt_utility.sql_in</span> |
| <a name="l00004"></a>00004 <span class="comment"> *</span> |
| <a name="l00005"></a>00005 <span class="comment"> * @brief Utility functions widely used in C4.5 and random forest.</span> |
| <a name="l00006"></a>00006 <span class="comment"> * @date April 5, 2012</span> |
| <a name="l00007"></a>00007 <span class="comment"> *</span> |
| <a name="l00008"></a>00008 <span class="comment"> */</span><span class="comment">/* ----------------------------------------------------------------------- */</span> |
| <a name="l00009"></a>00009 |
| <a name="l00010"></a>00010 |
| <a name="l00011"></a>00011 <span class="comment">/*</span> |
| <a name="l00012"></a>00012 <span class="comment"> * @brief Cast any value to text.</span> |
| <a name="l00013"></a>00013 <span class="comment"> *</span> |
| <a name="l00014"></a>00014 <span class="comment"> * @param val A value with any specific type.</span> |
| <a name="l00015"></a>00015 <span class="comment"> *</span> |
| <a name="l00016"></a>00016 <span class="comment"> * @return The text format string for the value.</span> |
| <a name="l00017"></a>00017 <span class="comment"> *</span> |
| <a name="l00018"></a>00018 <span class="comment"> * @note Greenplum doesn't support boolean to text casting.</span> |
| <a name="l00019"></a>00019 <span class="comment"> *</span> |
| <a name="l00020"></a>00020 <span class="comment"> */</span> |
| <a name="l00021"></a>00021 CREATE OR REPLACE FUNCTION MADLIB_SCHEMA.__to_char |
| <a name="l00022"></a>00022 ( |
| <a name="l00023"></a>00023 val anyelement |
| <a name="l00024"></a>00024 ) |
| <a name="l00025"></a>00025 RETURNS TEXT |
| <a name="l00026"></a>00026 AS <span class="stringliteral">'MODULE_PATHNAME'</span>, <span class="stringliteral">'dt_to_text'</span> |
| <a name="l00027"></a>00027 LANGUAGE C STRICT IMMUTABLE; |
| <a name="l00028"></a>00028 |
| <a name="l00029"></a>00029 |
| <a name="l00030"></a>00030 <span class="comment">/*</span> |
| <a name="l00031"></a>00031 <span class="comment"> * @brief Cast regclass to text. we will not create a cast, </span> |
| <a name="l00032"></a>00032 <span class="comment"> * since it may override the existing cast. </span> |
| <a name="l00033"></a>00033 <span class="comment"> * Although there is no cast for regclass to text,</span> |
| <a name="l00034"></a>00034 <span class="comment"> * PL/PGSQL can coerce it to text automatically.</span> |
| <a name="l00035"></a>00035 <span class="comment"> * Another implementation can use sql function:</span> |
| <a name="l00036"></a>00036 <span class="comment"> * select textin(regclassout('pg_class'::regclass));</span> |
| <a name="l00037"></a>00037 <span class="comment"> * </span> |
| <a name="l00038"></a>00038 <span class="comment"> * @param rc The regclass of the table.</span> |
| <a name="l00039"></a>00039 <span class="comment"> *</span> |
| <a name="l00040"></a>00040 <span class="comment"> * @return The text representation for the regclass.</span> |
| <a name="l00041"></a>00041 <span class="comment"> *</span> |
| <a name="l00042"></a>00042 <span class="comment"> */</span> |
| <a name="l00043"></a>00043 CREATE or replace FUNCTION MADLIB_SCHEMA.__regclass_to_text |
| <a name="l00044"></a>00044 ( |
| <a name="l00045"></a>00045 rc regclass |
| <a name="l00046"></a>00046 ) |
| <a name="l00047"></a>00047 RETURNS TEXT |
| <a name="l00048"></a>00048 AS $$ |
| <a name="l00049"></a>00049 BEGIN |
| <a name="l00050"></a>00050 RETURN rc; |
| <a name="l00051"></a>00051 END |
| <a name="l00052"></a>00052 $$ LANGUAGE PLPGSQL IMMUTABLE; |
| <a name="l00053"></a>00053 |
| <a name="l00054"></a>00054 |
| <a name="l00055"></a>00055 <span class="comment">/*</span> |
| <a name="l00056"></a>00056 <span class="comment"> * @brief Format a string with the value in args array.</span> |
| <a name="l00057"></a>00057 <span class="comment"> *</span> |
| <a name="l00058"></a>00058 <span class="comment"> * @param fmt The format string.</span> |
| <a name="l00059"></a>00059 <span class="comment"> * @param args The specified elements in format string.</span> |
| <a name="l00060"></a>00060 <span class="comment"> *</span> |
| <a name="l00061"></a>00061 <span class="comment"> * @return The formated string.</span> |
| <a name="l00062"></a>00062 <span class="comment"> *</span> |
| <a name="l00063"></a>00063 <span class="comment"> * @note Each '%' in fmt will be replaced with the corresponding value of args.</span> |
| <a name="l00064"></a>00064 <span class="comment"> * The number of '%'s should equal to the length of array args.</span> |
| <a name="l00065"></a>00065 <span class="comment"> *</span> |
| <a name="l00066"></a>00066 <span class="comment"> */</span> |
| <a name="l00067"></a>00067 CREATE OR REPLACE FUNCTION MADLIB_SCHEMA.__format |
| <a name="l00068"></a>00068 ( |
| <a name="l00069"></a>00069 fmt TEXT, |
| <a name="l00070"></a>00070 args TEXT[] |
| <a name="l00071"></a>00071 ) |
| <a name="l00072"></a>00072 RETURNS TEXT |
| <a name="l00073"></a>00073 AS <span class="stringliteral">'MODULE_PATHNAME'</span>, <span class="stringliteral">'dt_text_format'</span> |
| <a name="l00074"></a>00074 LANGUAGE C IMMUTABLE; |
| <a name="l00075"></a>00075 |
| <a name="l00076"></a>00076 |
| <a name="l00077"></a>00077 <span class="comment">/*</span> |
| <a name="l00078"></a>00078 <span class="comment"> * @brief Short form to format a string with four parameters.</span> |
| <a name="l00079"></a>00079 <span class="comment"> *</span> |
| <a name="l00080"></a>00080 <span class="comment"> * @param arg1 The first argument.</span> |
| <a name="l00081"></a>00081 <span class="comment"> * @param arg2 The second argument.</span> |
| <a name="l00082"></a>00082 <span class="comment"> * @param arg3 The third argument.</span> |
| <a name="l00083"></a>00083 <span class="comment"> * @param arg4 The fouth argument.</span> |
| <a name="l00084"></a>00084 <span class="comment"> * </span> |
| <a name="l00085"></a>00085 <span class="comment"> * @return The formated string.</span> |
| <a name="l00086"></a>00086 <span class="comment"> *</span> |
| <a name="l00087"></a>00087 <span class="comment"> */</span> |
| <a name="l00088"></a>00088 CREATE OR REPLACE FUNCTION MADLIB_SCHEMA.__format |
| <a name="l00089"></a>00089 ( |
| <a name="l00090"></a>00090 fmt TEXT, |
| <a name="l00091"></a>00091 arg1 TEXT, |
| <a name="l00092"></a>00092 arg2 TEXT, |
| <a name="l00093"></a>00093 arg3 TEXT, |
| <a name="l00094"></a>00094 arg4 TEXT |
| <a name="l00095"></a>00095 ) |
| <a name="l00096"></a>00096 RETURNS TEXT AS $$ |
| <a name="l00097"></a>00097 SELECT MADLIB_SCHEMA.__format($1, ARRAY[$2, $3, $4, $5]); |
| <a name="l00098"></a>00098 $$ LANGUAGE sql IMMUTABLE; |
| <a name="l00099"></a>00099 |
| <a name="l00100"></a>00100 |
| <a name="l00101"></a>00101 <span class="comment">/*</span> |
| <a name="l00102"></a>00102 <span class="comment"> * @brief Short form to format a string with three parameters.</span> |
| <a name="l00103"></a>00103 <span class="comment"> *</span> |
| <a name="l00104"></a>00104 <span class="comment"> * @param arg1 The first argument.</span> |
| <a name="l00105"></a>00105 <span class="comment"> * @param arg2 The second argument.</span> |
| <a name="l00106"></a>00106 <span class="comment"> * @param arg3 The third argument.</span> |
| <a name="l00107"></a>00107 <span class="comment"> *</span> |
| <a name="l00108"></a>00108 <span class="comment"> * @return The formated string.</span> |
| <a name="l00109"></a>00109 <span class="comment"> *</span> |
| <a name="l00110"></a>00110 <span class="comment"> */</span> |
| <a name="l00111"></a>00111 CREATE OR REPLACE FUNCTION MADLIB_SCHEMA.__format |
| <a name="l00112"></a>00112 ( |
| <a name="l00113"></a>00113 fmt TEXT, |
| <a name="l00114"></a>00114 arg1 TEXT, |
| <a name="l00115"></a>00115 arg2 TEXT, |
| <a name="l00116"></a>00116 arg3 TEXT |
| <a name="l00117"></a>00117 ) |
| <a name="l00118"></a>00118 RETURNS TEXT AS $$ |
| <a name="l00119"></a>00119 SELECT MADLIB_SCHEMA.__format($1, ARRAY[$2, $3, $4]); |
| <a name="l00120"></a>00120 $$ LANGUAGE sql IMMUTABLE; |
| <a name="l00121"></a>00121 |
| <a name="l00122"></a>00122 |
| <a name="l00123"></a>00123 <span class="comment">/*</span> |
| <a name="l00124"></a>00124 <span class="comment"> * @brief Short form to format a string with two parameters.</span> |
| <a name="l00125"></a>00125 <span class="comment"> *</span> |
| <a name="l00126"></a>00126 <span class="comment"> * @param arg1 The first argument.</span> |
| <a name="l00127"></a>00127 <span class="comment"> * @param arg2 The second argument.</span> |
| <a name="l00128"></a>00128 <span class="comment"> *</span> |
| <a name="l00129"></a>00129 <span class="comment"> * @return The formated string.</span> |
| <a name="l00130"></a>00130 <span class="comment"> *</span> |
| <a name="l00131"></a>00131 <span class="comment"> */</span> |
| <a name="l00132"></a>00132 CREATE OR REPLACE FUNCTION MADLIB_SCHEMA.__format |
| <a name="l00133"></a>00133 ( |
| <a name="l00134"></a>00134 fmt TEXT, |
| <a name="l00135"></a>00135 arg1 TEXT, |
| <a name="l00136"></a>00136 arg2 TEXT |
| <a name="l00137"></a>00137 ) |
| <a name="l00138"></a>00138 RETURNS TEXT AS $$ |
| <a name="l00139"></a>00139 SELECT MADLIB_SCHEMA.__format($1, ARRAY[$2, $3]); |
| <a name="l00140"></a>00140 $$ LANGUAGE sql IMMUTABLE; |
| <a name="l00141"></a>00141 |
| <a name="l00142"></a>00142 |
| <a name="l00143"></a>00143 <span class="comment">/*</span> |
| <a name="l00144"></a>00144 <span class="comment"> * @brief Short form to format a string with a parameter.</span> |
| <a name="l00145"></a>00145 <span class="comment"> *</span> |
| <a name="l00146"></a>00146 <span class="comment"> * @param arg1 The first argument.</span> |
| <a name="l00147"></a>00147 <span class="comment"> *</span> |
| <a name="l00148"></a>00148 <span class="comment"> * @return The formated string.</span> |
| <a name="l00149"></a>00149 <span class="comment"> *</span> |
| <a name="l00150"></a>00150 <span class="comment"> */</span> |
| <a name="l00151"></a>00151 CREATE OR REPLACE FUNCTION MADLIB_SCHEMA.__format |
| <a name="l00152"></a>00152 ( |
| <a name="l00153"></a>00153 fmt TEXT, |
| <a name="l00154"></a>00154 arg1 TEXT |
| <a name="l00155"></a>00155 ) |
| <a name="l00156"></a>00156 RETURNS TEXT AS $$ |
| <a name="l00157"></a>00157 SELECT MADLIB_SCHEMA.__format($1, ARRAY[$2]); |
| <a name="l00158"></a>00158 $$ LANGUAGE sql IMMUTABLE; |
| <a name="l00159"></a>00159 |
| <a name="l00160"></a>00160 |
| <a name="l00161"></a>00161 <span class="comment">/*</span> |
| <a name="l00162"></a>00162 <span class="comment"> * @brief Raise exception if the condition is false.</span> |
| <a name="l00163"></a>00163 <span class="comment"> *</span> |
| <a name="l00164"></a>00164 <span class="comment"> * @param condition The assert condition.</span> |
| <a name="l00165"></a>00165 <span class="comment"> * @param reason The reason string displayed when assert failure.</span> |
| <a name="l00166"></a>00166 <span class="comment"> *</span> |
| <a name="l00167"></a>00167 <span class="comment"> */</span> |
| <a name="l00168"></a>00168 CREATE OR REPLACE FUNCTION MADLIB_SCHEMA.__assert |
| <a name="l00169"></a>00169 ( |
| <a name="l00170"></a>00170 condition BOOLEAN, |
| <a name="l00171"></a>00171 reason TEXT |
| <a name="l00172"></a>00172 ) |
| <a name="l00173"></a>00173 RETURNS <span class="keywordtype">void</span> AS $$ |
| <a name="l00174"></a>00174 BEGIN |
| <a name="l00175"></a>00175 IF (NOT condition) THEN |
| <a name="l00176"></a>00176 RAISE EXCEPTION <span class="stringliteral">'ERROR: %'</span>, reason; |
| <a name="l00177"></a>00177 END IF; |
| <a name="l00178"></a>00178 END |
| <a name="l00179"></a>00179 $$ LANGUAGE PLPGSQL IMMUTABLE; |
| <a name="l00180"></a>00180 |
| <a name="l00181"></a>00181 |
| <a name="l00182"></a>00182 <span class="comment">/*</span> |
| <a name="l00183"></a>00183 <span class="comment"> * @brief Test if the specified table exists or not.</span> |
| <a name="l00184"></a>00184 <span class="comment"> *</span> |
| <a name="l00185"></a>00185 <span class="comment"> * @param full_table_name The full table name.</span> |
| <a name="l00186"></a>00186 <span class="comment"> *</span> |
| <a name="l00187"></a>00187 <span class="comment"> */</span> |
| <a name="l00188"></a>00188 CREATE OR REPLACE FUNCTION MADLIB_SCHEMA.__table_exists |
| <a name="l00189"></a>00189 ( |
| <a name="l00190"></a>00190 full_table_name TEXT |
| <a name="l00191"></a>00191 ) |
| <a name="l00192"></a>00192 RETURNS BOOLEAN AS |
| <a name="l00193"></a>00193 <span class="stringliteral">'MODULE_PATHNAME'</span>, <span class="stringliteral">'table_exists'</span> |
| <a name="l00194"></a>00194 LANGUAGE C IMMUTABLE; |
| <a name="l00195"></a>00195 |
| <a name="l00196"></a>00196 |
| <a name="l00197"></a>00197 <span class="comment">/*</span> |
| <a name="l00198"></a>00198 <span class="comment"> * @brief Test if the specified column exists or not.</span> |
| <a name="l00199"></a>00199 <span class="comment"> *</span> |
| <a name="l00200"></a>00200 <span class="comment"> * @param full_table_name The full table name.</span> |
| <a name="l00201"></a>00201 <span class="comment"> *</span> |
| <a name="l00202"></a>00202 <span class="comment"> * @return True if the column exists, otherwise return false.</span> |
| <a name="l00203"></a>00203 <span class="comment"> *</span> |
| <a name="l00204"></a>00204 <span class="comment"> */</span> |
| <a name="l00205"></a>00205 CREATE OR REPLACE FUNCTION MADLIB_SCHEMA.__column_exists |
| <a name="l00206"></a>00206 ( |
| <a name="l00207"></a>00207 full_table_name TEXT, |
| <a name="l00208"></a>00208 column_name TEXT |
| <a name="l00209"></a>00209 ) |
| <a name="l00210"></a>00210 RETURNS BOOLEAN AS $$ |
| <a name="l00211"></a>00211 DECLARE |
| <a name="l00212"></a>00212 curstmt TEXT := <span class="stringliteral">''</span>; |
| <a name="l00213"></a>00213 result INT := 0; |
| <a name="l00214"></a>00214 BEGIN |
| <a name="l00215"></a>00215 PERFORM MADLIB_SCHEMA.__assert |
| <a name="l00216"></a>00216 ( |
| <a name="l00217"></a>00217 (full_table_name IS NOT NULL) AND (column_name IS NOT NULL), |
| <a name="l00218"></a>00218 <span class="stringliteral">'the table name and column name must not be null'</span> |
| <a name="l00219"></a>00219 ); |
| <a name="l00220"></a>00220 |
| <a name="l00221"></a>00221 IF (MADLIB_SCHEMA.__table_exists(full_table_name)) THEN |
| <a name="l00222"></a>00222 SELECT MADLIB_SCHEMA.__format |
| <a name="l00223"></a>00223 ( |
| <a name="l00224"></a>00224 <span class="stringliteral">'SELECT COUNT(*) </span> |
| <a name="l00225"></a>00225 <span class="stringliteral"> FROM pg_catalog.pg_attribute </span> |
| <a name="l00226"></a>00226 <span class="stringliteral"> WHERE attnum > 0 AND </span> |
| <a name="l00227"></a>00227 <span class="stringliteral"> (NOT attisdropped) AND</span> |
| <a name="l00228"></a>00228 <span class="stringliteral"> attname = '</span><span class="charliteral">'%'</span><span class="stringliteral">' AND</span> |
| <a name="l00229"></a>00229 <span class="stringliteral"> attrelid = '</span><span class="charliteral">'%'</span><span class="stringliteral">'::regclass'</span>, |
| <a name="l00230"></a>00230 ARRAY[ |
| <a name="l00231"></a>00231 column_name, |
| <a name="l00232"></a>00232 full_table_name |
| <a name="l00233"></a>00233 ] |
| <a name="l00234"></a>00234 ) INTO curstmt; |
| <a name="l00235"></a>00235 |
| <a name="l00236"></a>00236 EXECUTE curstmt INTO result; |
| <a name="l00237"></a>00237 |
| <a name="l00238"></a>00238 RETURN result >= 1; |
| <a name="l00239"></a>00239 END IF; |
| <a name="l00240"></a>00240 |
| <a name="l00241"></a>00241 RETURN <span class="charliteral">'f'</span>; |
| <a name="l00242"></a>00242 END |
| <a name="l00243"></a>00243 $$ LANGUAGE PLPGSQL STABLE; |
| <a name="l00244"></a>00244 |
| <a name="l00245"></a>00245 |
| <a name="l00246"></a>00246 <span class="comment">/*</span> |
| <a name="l00247"></a>00247 <span class="comment"> * @brief Assert if the specified table exists or not.</span> |
| <a name="l00248"></a>00248 <span class="comment"> *</span> |
| <a name="l00249"></a>00249 <span class="comment"> * @param full_table_name The full table name.</span> |
| <a name="l00250"></a>00250 <span class="comment"> *</span> |
| <a name="l00251"></a>00251 <span class="comment"> */</span> |
| <a name="l00252"></a>00252 CREATE OR REPLACE FUNCTION MADLIB_SCHEMA.__assert_table |
| <a name="l00253"></a>00253 ( |
| <a name="l00254"></a>00254 full_table_name TEXT, |
| <a name="l00255"></a>00255 existence BOOLEAN |
| <a name="l00256"></a>00256 ) |
| <a name="l00257"></a>00257 RETURNS <span class="keywordtype">void</span> AS $$ |
| <a name="l00258"></a>00258 DECLARE |
| <a name="l00259"></a>00259 err_msg TEXT; |
| <a name="l00260"></a>00260 BEGIN |
| <a name="l00261"></a>00261 IF (existence) THEN |
| <a name="l00262"></a>00262 err_msg = 'assertion failure. Table: ''' || full_table_name || |
| <a name="l00263"></a>00263 ''' does not exist'; |
| <a name="l00264"></a>00264 ELSE |
| <a name="l00265"></a>00265 err_msg = 'assertion failure. Table: ''' || full_table_name || |
| <a name="l00266"></a>00266 ''' already exists'; |
| <a name="l00267"></a>00267 END IF; |
| <a name="l00268"></a>00268 |
| <a name="l00269"></a>00269 PERFORM MADLIB_SCHEMA.__assert |
| <a name="l00270"></a>00270 ( |
| <a name="l00271"></a>00271 MADLIB_SCHEMA.__table_exists(full_table_name) = existence, |
| <a name="l00272"></a>00272 err_msg |
| <a name="l00273"></a>00273 ); |
| <a name="l00274"></a>00274 END |
| <a name="l00275"></a>00275 $$ LANGUAGE PLPGSQL STABLE; |
| <a name="l00276"></a>00276 |
| <a name="l00277"></a>00277 |
| <a name="l00278"></a>00278 <span class="comment">/*</span> |
| <a name="l00279"></a>00279 <span class="comment"> * @brief Strip the schema name from the full table name.</span> |
| <a name="l00280"></a>00280 <span class="comment"> *</span> |
| <a name="l00281"></a>00281 <span class="comment"> * @param full_table_name The full table name. </span> |
| <a name="l00282"></a>00282 <span class="comment"> *</span> |
| <a name="l00283"></a>00283 <span class="comment"> * @return The table name without schema name.</span> |
| <a name="l00284"></a>00284 <span class="comment"> *</span> |
| <a name="l00285"></a>00285 <span class="comment"> */</span> |
| <a name="l00286"></a>00286 CREATE OR REPLACE FUNCTION MADLIB_SCHEMA.__strip_schema_name |
| <a name="l00287"></a>00287 ( |
| <a name="l00288"></a>00288 full_table_name TEXT |
| <a name="l00289"></a>00289 ) |
| <a name="l00290"></a>00290 RETURNS TEXT AS $$ |
| <a name="l00291"></a>00291 DECLARE |
| <a name="l00292"></a>00292 str_val TEXT; |
| <a name="l00293"></a>00293 BEGIN |
| <a name="l00294"></a>00294 PERFORM MADLIB_SCHEMA.__assert |
| <a name="l00295"></a>00295 ( |
| <a name="l00296"></a>00296 full_table_name IS NOT NULL, |
| <a name="l00297"></a>00297 'table name should not be null' |
| <a name="l00298"></a>00298 ); |
| <a name="l00299"></a>00299 |
| <a name="l00300"></a>00300 |
| <a name="l00301"></a>00301 str_val = trim(both ' ' FROM split_part(full_table_name, '.', 2)); |
| <a name="l00302"></a>00302 |
| <a name="l00303"></a>00303 IF (length(str_val) = 0) THEN |
| <a name="l00304"></a>00304 str_val = btrim(full_table_name, ' '); |
| <a name="l00305"></a>00305 END IF; |
| <a name="l00306"></a>00306 |
| <a name="l00307"></a>00307 RETURN lower(str_val); |
| <a name="l00308"></a>00308 end |
| <a name="l00309"></a>00309 $$ LANGUAGE PLPGSQL IMMUTABLE; |
| <a name="l00310"></a>00310 |
| <a name="l00311"></a>00311 |
| <a name="l00312"></a>00312 <span class="comment">/*</span> |
| <a name="l00313"></a>00313 <span class="comment"> * @brief Get the schema name from a full table name.</span> |
| <a name="l00314"></a>00314 <span class="comment"> * if there is no schema name in the full table name, then</span> |
| <a name="l00315"></a>00315 <span class="comment"> * if the table exists, we return the schema name from catalog</span> |
| <a name="l00316"></a>00316 <span class="comment"> * else the current schema name,</span> |
| <a name="l00317"></a>00317 <span class="comment"> * else return the schema name from the full table name directly.</span> |
| <a name="l00318"></a>00318 <span class="comment"> *</span> |
| <a name="l00319"></a>00319 <span class="comment"> * @param full_table_name The full table name.</span> |
| <a name="l00320"></a>00320 <span class="comment"> *</span> |
| <a name="l00321"></a>00321 <span class="comment"> * @return The schema name of the table.</span> |
| <a name="l00322"></a>00322 <span class="comment"> *</span> |
| <a name="l00323"></a>00323 <span class="comment"> * @note This function should be VOLATILE, since we may get schema name </span> |
| <a name="l00324"></a>00324 <span class="comment"> * from a new created table in the same transaction.</span> |
| <a name="l00325"></a>00325 <span class="comment"> *</span> |
| <a name="l00326"></a>00326 <span class="comment"> */</span> |
| <a name="l00327"></a>00327 CREATE OR REPLACE FUNCTION MADLIB_SCHEMA.__get_schema_name |
| <a name="l00328"></a>00328 ( |
| <a name="l00329"></a>00329 full_table_name TEXT |
| <a name="l00330"></a>00330 ) |
| <a name="l00331"></a>00331 RETURNS TEXT AS $$ |
| <a name="l00332"></a>00332 DECLARE |
| <a name="l00333"></a>00333 table_name TEXT; |
| <a name="l00334"></a>00334 temp TEXT[]; |
| <a name="l00335"></a>00335 len INT; |
| <a name="l00336"></a>00336 curstmt TEXT; |
| <a name="l00337"></a>00337 schema_name TEXT; |
| <a name="l00338"></a>00338 BEGIN |
| <a name="l00339"></a>00339 PERFORM MADLIB_SCHEMA.__assert |
| <a name="l00340"></a>00340 ( |
| <a name="l00341"></a>00341 full_table_name IS NOT NULL, |
| <a name="l00342"></a>00342 'table name should not be null' |
| <a name="l00343"></a>00343 ); |
| <a name="l00344"></a>00344 |
| <a name="l00345"></a>00345 temp = string_to_array(full_table_name, '.'); |
| <a name="l00346"></a>00346 len = array_upper(temp, 1); |
| <a name="l00347"></a>00347 |
| <a name="l00348"></a>00348 IF (1 = len) THEN |
| <a name="l00349"></a>00349 -- if table exists, return the schema name from catalog |
| <a name="l00350"></a>00350 IF (MADLIB_SCHEMA.__table_exists(full_table_name)) THEN |
| <a name="l00351"></a>00351 SELECT nspname |
| <a name="l00352"></a>00352 FROM pg_catalog.pg_namespace n |
| <a name="l00353"></a>00353 WHERE n.oid = |
| <a name="l00354"></a>00354 ( |
| <a name="l00355"></a>00355 SELECT relnamespace FROM pg_catalog.pg_class |
| <a name="l00356"></a>00356 WHERE oid= full_table_name::regclass |
| <a name="l00357"></a>00357 ) |
| <a name="l00358"></a>00358 |
| <a name="l00359"></a>00359 INTO schema_name; |
| <a name="l00360"></a>00360 ELSE |
| <a name="l00361"></a>00361 -- get the current schema name |
| <a name="l00362"></a>00362 schema_name = current_schema(); |
| <a name="l00363"></a>00363 END IF; |
| <a name="l00364"></a>00364 ELSE |
| <a name="l00365"></a>00365 PERFORM MADLIB_SCHEMA.__assert |
| <a name="l00366"></a>00366 ( |
| <a name="l00367"></a>00367 len = 2, |
| <a name="l00368"></a>00368 'wrong full table name<' || full_table_name || '>' |
| <a name="l00369"></a>00369 ); |
| <a name="l00370"></a>00370 -- get the shema name directly |
| <a name="l00371"></a>00371 schema_name = lower(btrim(temp[1], ' ')); |
| <a name="l00372"></a>00372 END IF; |
| <a name="l00373"></a>00373 |
| <a name="l00374"></a>00374 RETURN schema_name; |
| <a name="l00375"></a>00375 end |
| <a name="l00376"></a>00376 $$ LANGUAGE PLPGSQL; |
| <a name="l00377"></a>00377 |
| <a name="l00378"></a>00378 |
| <a name="l00379"></a>00379 <span class="comment">/*</span> |
| <a name="l00380"></a>00380 <span class="comment"> * @brief Test if the given element is in the specified array or not.</span> |
| <a name="l00381"></a>00381 <span class="comment"> *</span> |
| <a name="l00382"></a>00382 <span class="comment"> * @param find The element to be found.</span> |
| <a name="l00383"></a>00383 <span class="comment"> * @param arr The array containing the elements.</span> |
| <a name="l00384"></a>00384 <span class="comment"> * </span> |
| <a name="l00385"></a>00385 <span class="comment"> * @return True the element is in the array. Otherwise returns false.</span> |
| <a name="l00386"></a>00386 <span class="comment"> *</span> |
| <a name="l00387"></a>00387 <span class="comment"> */</span> |
| <a name="l00388"></a>00388 CREATE OR REPLACE FUNCTION MADLIB_SCHEMA.__array_search |
| <a name="l00389"></a>00389 ( |
| <a name="l00390"></a>00390 find ANYELEMENT, |
| <a name="l00391"></a>00391 arr ANYARRAY |
| <a name="l00392"></a>00392 ) |
| <a name="l00393"></a>00393 RETURNS BOOLEAN AS $$ |
| <a name="l00394"></a>00394 SELECT count(*) = 1 |
| <a name="l00395"></a>00395 FROM |
| <a name="l00396"></a>00396 ( |
| <a name="l00397"></a>00397 SELECT unnest($2) as elem |
| <a name="l00398"></a>00398 ) t |
| <a name="l00399"></a>00399 WHERE $1 IS NOT DISTINCT FROM elem |
| <a name="l00400"></a>00400 $$ LANGUAGE sql IMMUTABLE; |
| <a name="l00401"></a>00401 |
| <a name="l00402"></a>00402 |
| <a name="l00403"></a>00403 <span class="comment">/*</span> |
| <a name="l00404"></a>00404 <span class="comment"> * @brief Test if each element in the given array is a column of the table.</span> |
| <a name="l00405"></a>00405 <span class="comment"> *</span> |
| <a name="l00406"></a>00406 <span class="comment"> * @param column_names The array containing the columns to be tested.</span> |
| <a name="l00407"></a>00407 <span class="comment"> * @param table_name The full table name.</span> |
| <a name="l00408"></a>00408 <span class="comment"> *</span> |
| <a name="l00409"></a>00409 <span class="comment"> * @return True if each element of column_names is a column of the table.</span> |
| <a name="l00410"></a>00410 <span class="comment"> *</span> |
| <a name="l00411"></a>00411 <span class="comment"> */</span> |
| <a name="l00412"></a>00412 CREATE OR REPLACE FUNCTION MADLIB_SCHEMA.__columns_in_table |
| <a name="l00413"></a>00413 ( |
| <a name="l00414"></a>00414 column_names TEXT[], |
| <a name="l00415"></a>00415 table_name TEXT |
| <a name="l00416"></a>00416 ) |
| <a name="l00417"></a>00417 RETURNS BOOLEAN AS $$ |
| <a name="l00418"></a>00418 SELECT count(*) = 0 |
| <a name="l00419"></a>00419 FROM |
| <a name="l00420"></a>00420 ( |
| <a name="l00421"></a>00421 SELECT unnest($1) |
| <a name="l00422"></a>00422 EXCEPT |
| <a name="l00423"></a>00423 SELECT quote_ident(attname) |
| <a name="l00424"></a>00424 FROM pg_attribute |
| <a name="l00425"></a>00425 WHERE attrelid = $2::regclass AND |
| <a name="l00426"></a>00426 attnum > 0 AND |
| <a name="l00427"></a>00427 NOT attisdropped |
| <a name="l00428"></a>00428 ) t; |
| <a name="l00429"></a>00429 $$ LANGUAGE sql STABLE; |
| <a name="l00430"></a>00430 |
| <a name="l00431"></a>00431 |
| <a name="l00432"></a>00432 <span class="comment">/*</span> |
| <a name="l00433"></a>00433 <span class="comment"> * @brief Get the number of columns for a given table.</span> |
| <a name="l00434"></a>00434 <span class="comment"> *</span> |
| <a name="l00435"></a>00435 <span class="comment"> * @param table_name The full table name.</span> |
| <a name="l00436"></a>00436 <span class="comment"> *</span> |
| <a name="l00437"></a>00437 <span class="comment"> * @return The number of columns in the given table.</span> |
| <a name="l00438"></a>00438 <span class="comment"> *</span> |
| <a name="l00439"></a>00439 <span class="comment"> */</span> |
| <a name="l00440"></a>00440 CREATE OR REPLACE FUNCTION MADLIB_SCHEMA.__num_of_columns |
| <a name="l00441"></a>00441 ( |
| <a name="l00442"></a>00442 table_name TEXT |
| <a name="l00443"></a>00443 ) |
| <a name="l00444"></a>00444 RETURNS INT AS $$ |
| <a name="l00445"></a>00445 SELECT count(attname)::INT |
| <a name="l00446"></a>00446 FROM pg_attribute |
| <a name="l00447"></a>00447 WHERE attrelid = $1::regclass AND |
| <a name="l00448"></a>00448 attnum > 0 AND |
| <a name="l00449"></a>00449 NOT attisdropped |
| <a name="l00450"></a>00450 $$ LANGUAGE sql STABLE; |
| <a name="l00451"></a>00451 |
| <a name="l00452"></a>00452 |
| <a name="l00453"></a>00453 <span class="comment">/*</span> |
| <a name="l00454"></a>00454 <span class="comment"> * @brief Convert a string with delimiter ',' to an array. </span> |
| <a name="l00455"></a>00455 <span class="comment"> * Each element in the array is trimed from the start</span> |
| <a name="l00456"></a>00456 <span class="comment"> * and end using a space.</span> |
| <a name="l00457"></a>00457 <span class="comment"> *</span> |
| <a name="l00458"></a>00458 <span class="comment"> * @param csv_str The string with elements delimited by ','.</span> |
| <a name="l00459"></a>00459 <span class="comment"> *</span> |
| <a name="l00460"></a>00460 <span class="comment"> * @return The splitting string array.</span> |
| <a name="l00461"></a>00461 <span class="comment"> *</span> |
| <a name="l00462"></a>00462 <span class="comment"> * @note If the input string is NULL or an empty string</span> |
| <a name="l00463"></a>00463 <span class="comment"> * after trimmed with ' ', then NULL will be returned.</span> |
| <a name="l00464"></a>00464 <span class="comment"> *</span> |
| <a name="l00465"></a>00465 <span class="comment"> */</span> |
| <a name="l00466"></a>00466 CREATE OR REPLACE FUNCTION MADLIB_SCHEMA.__csvstr_to_array |
| <a name="l00467"></a>00467 ( |
| <a name="l00468"></a>00468 csv_str TEXT |
| <a name="l00469"></a>00469 ) |
| <a name="l00470"></a>00470 RETURNS TEXT[] AS $$ |
| <a name="l00471"></a>00471 DECLARE |
| <a name="l00472"></a>00472 ret TEXT[]; |
| <a name="l00473"></a>00473 str_val TEXT; |
| <a name="l00474"></a>00474 index INTEGER; |
| <a name="l00475"></a>00475 BEGIN |
| <a name="l00476"></a>00476 ret = string_to_array(lower(btrim(csv_str, ' ')), ','); |
| <a name="l00477"></a>00477 |
| <a name="l00478"></a>00478 -- if the input array is NULL or an empty one, |
| <a name="l00479"></a>00479 -- then we return NULL directly |
| <a name="l00480"></a>00480 -- (array_upper will return non-NULL otherwise) |
| <a name="l00481"></a>00481 IF (array_upper(ret, 1) IS NULL) THEN |
| <a name="l00482"></a>00482 RETURN NULL; |
| <a name="l00483"></a>00483 END IF; |
| <a name="l00484"></a>00484 |
| <a name="l00485"></a>00485 -- null or empty array will be filtered |
| <a name="l00486"></a>00486 FOR index IN 1..array_upper(ret, 1) LOOP |
| <a name="l00487"></a>00487 ret[index] = quote_ident(btrim(ret[index], ' ')); |
| <a name="l00488"></a>00488 END LOOP; |
| <a name="l00489"></a>00489 |
| <a name="l00490"></a>00490 RETURN ret; |
| <a name="l00491"></a>00491 END |
| <a name="l00492"></a>00492 $$ LANGUAGE PLPGSQL IMMUTABLE; |
| <a name="l00493"></a>00493 |
| <a name="l00494"></a>00494 |
| <a name="l00495"></a>00495 <span class="comment">/*</span> |
| <a name="l00496"></a>00496 <span class="comment"> * @brief Retrieve a BOOL array. The ith element in the array</span> |
| <a name="l00497"></a>00497 <span class="comment"> * indicate whether arr2[i] is in arr1 or not. The size</span> |
| <a name="l00498"></a>00498 <span class="comment"> * of the BOOL array is the same as arr2. For example,</span> |
| <a name="l00499"></a>00499 <span class="comment"> * arr1 = ['aa', 'bb', 'dd'],</span> |
| <a name="l00500"></a>00500 <span class="comment"> * arr2 = ['aa', 'dd', 'bb', 'ee', 'ccc']</span> |
| <a name="l00501"></a>00501 <span class="comment"> * then the BOOL array is:</span> |
| <a name="l00502"></a>00502 <span class="comment"> * ['t', 't', 't', 'f', 'f']</span> |
| <a name="l00503"></a>00503 <span class="comment"> *</span> |
| <a name="l00504"></a>00504 <span class="comment"> * @param src_arr The source array.</span> |
| <a name="l00505"></a>00505 <span class="comment"> * @param tst_arr The array to be tested.</span> |
| <a name="l00506"></a>00506 <span class="comment"> *</span> |
| <a name="l00507"></a>00507 <span class="comment"> * @return A BOOL array. </span> |
| <a name="l00508"></a>00508 <span class="comment"> *</span> |
| <a name="l00509"></a>00509 <span class="comment"> */</span> |
| <a name="l00510"></a>00510 CREATE OR REPLACE FUNCTION MADLIB_SCHEMA.__array_elem_in |
| <a name="l00511"></a>00511 ( |
| <a name="l00512"></a>00512 src_arr ANYARRAY, |
| <a name="l00513"></a>00513 tst_arr ANYARRAY |
| <a name="l00514"></a>00514 ) |
| <a name="l00515"></a>00515 RETURNS BOOLEAN[] AS $$ |
| <a name="l00516"></a>00516 SELECT <a class="code" href="array__ops_8sql__in.html#af9f60293134ba4ce05b1f97f6faba822" title="ARRAY_AGG aggregate for compatibility with GPDB < 4.1 and Postgres < 9.0 This is a slower solution th...">array_agg</a>(elem in (SELECT unnest($1))) |
| <a name="l00517"></a>00517 FROM |
| <a name="l00518"></a>00518 ( |
| <a name="l00519"></a>00519 SELECT unnest($2) as elem |
| <a name="l00520"></a>00520 ) t |
| <a name="l00521"></a>00521 $$ LANGUAGE sql IMMUTABLE; |
| <a name="l00522"></a>00522 |
| <a name="l00523"></a>00523 |
| <a name="l00524"></a>00524 <span class="comment">/*</span> |
| <a name="l00525"></a>00525 <span class="comment"> * @brief Sort the given array.</span> |
| <a name="l00526"></a>00526 <span class="comment"> *</span> |
| <a name="l00527"></a>00527 <span class="comment"> * @param arr The array to be sorted.</span> |
| <a name="l00528"></a>00528 <span class="comment"> *</span> |
| <a name="l00529"></a>00529 <span class="comment"> * @return The sorted array. </span> |
| <a name="l00530"></a>00530 <span class="comment"> *</span> |
| <a name="l00531"></a>00531 <span class="comment"> */</span> |
| <a name="l00532"></a>00532 |
| <a name="l00533"></a>00533 CREATE OR REPLACE FUNCTION MADLIB_SCHEMA.__array_sort |
| <a name="l00534"></a>00534 ( |
| <a name="l00535"></a>00535 arr ANYARRAY |
| <a name="l00536"></a>00536 ) |
| <a name="l00537"></a>00537 RETURNS ANYARRAY AS $$ |
| <a name="l00538"></a>00538 SELECT ARRAY |
| <a name="l00539"></a>00539 (SELECT elem FROM unnest($1) elem ORDER BY elem LIMIT ALL) |
| <a name="l00540"></a>00540 $$ LANGUAGE sql IMMUTABLE; |
| <a name="l00541"></a>00541 |
| </pre></div></div> |
| </div> |
| <div id="nav-path" class="navpath"> |
| <ul> |
| <li class="navelem"><a class="el" href="dt__utility_8sql__in.html">dt_utility.sql_in</a> </li> |
| <!-- window showing the filter options --> |
| <div id="MSearchSelectWindow" |
| onmouseover="return searchBox.OnSearchSelectShow()" |
| onmouseout="return searchBox.OnSearchSelectHide()" |
| onkeydown="return searchBox.OnSearchSelectKey(event)"> |
| <a class="SelectItem" href="javascript:void(0)" onclick="searchBox.OnSelectItem(0)"><span class="SelectionMark"> </span>All</a><a class="SelectItem" href="javascript:void(0)" onclick="searchBox.OnSelectItem(1)"><span class="SelectionMark"> </span>Files</a><a class="SelectItem" href="javascript:void(0)" onclick="searchBox.OnSelectItem(2)"><span class="SelectionMark"> </span>Functions</a></div> |
| |
| <!-- iframe showing the search results (closed by default) --> |
| <div id="MSearchResultsWindow"> |
| <iframe src="javascript:void(0)" frameborder="0" |
| name="MSearchResults" id="MSearchResults"> |
| </iframe> |
| </div> |
| |
| |
| <li class="footer">Generated on Tue Apr 2 2013 14:57:03 for MADlib by |
| <a href="http://www.doxygen.org/index.html"> |
| <img class="footer" src="doxygen.png" alt="doxygen"/></a> 1.7.5.1 </li> |
| </ul> |
| </div> |
| |
| |
| </body> |
| </html> |