Line data Source code
1 : /**********************************************************************
2 : * PostgreSQL::InServer::Util
3 : *
4 : * src/pl/plperl/Util.xs
5 : *
6 : * Defines plperl interfaces for general-purpose utilities.
7 : * This module is bootstrapped as soon as an interpreter is initialized.
8 : * Currently doesn't define a PACKAGE= so all subs are in main:: to avoid
9 : * the need for explicit importing.
10 : *
11 : **********************************************************************/
12 :
13 : /* this must be first: */
14 : #include "postgres.h"
15 :
16 : #include "fmgr.h"
17 : #include "utils/builtins.h"
18 : #include "utils/bytea.h" /* for byteain & byteaout */
19 : #include "varatt.h"
20 :
21 : /* perl stuff */
22 : #define PG_NEED_PERL_XSUB_H
23 : #include "plperl.h"
24 :
25 :
26 : static text *
27 42 : sv2text(SV *sv)
28 : {
29 42 : char *str = sv2cstr(sv);
30 : text *text;
31 :
32 42 : text = cstring_to_text(str);
33 42 : pfree(str);
34 42 : return text;
35 : }
36 :
37 : MODULE = PostgreSQL::InServer::Util PREFIX = util_
38 :
39 : PROTOTYPES: ENABLE
40 : VERSIONCHECK: DISABLE
41 :
42 : int
43 : _aliased_constants()
44 : PROTOTYPE:
45 : ALIAS:
46 : DEBUG = DEBUG2
47 : LOG = LOG
48 : INFO = INFO
49 : NOTICE = NOTICE
50 : WARNING = WARNING
51 : ERROR = ERROR
52 : CODE:
53 : /* uses the ALIAS value as the return value */
54 186 : RETVAL = ix;
55 : OUTPUT:
56 : RETVAL
57 :
58 :
59 : void
60 : util_elog(level, msg)
61 : int level
62 : SV *msg
63 : CODE:
64 186 : if (level > ERROR) /* no PANIC allowed thanks */
65 0 : level = ERROR;
66 186 : if (level < DEBUG5)
67 0 : level = DEBUG5;
68 186 : plperl_util_elog(level, msg);
69 :
70 : SV *
71 : util_quote_literal(sv)
72 : SV *sv
73 : CODE:
74 7 : if (!sv || !SvOK(sv)) {
75 1 : RETVAL = &PL_sv_undef;
76 : }
77 : else {
78 6 : text *arg = sv2text(sv);
79 6 : text *quoted = DatumGetTextPP(DirectFunctionCall1(quote_literal, PointerGetDatum(arg)));
80 : char *str;
81 :
82 6 : pfree(arg);
83 6 : str = text_to_cstring(quoted);
84 6 : RETVAL = cstr2sv(str);
85 6 : pfree(str);
86 : }
87 : OUTPUT:
88 : RETVAL
89 :
90 : SV *
91 : util_quote_nullable(sv)
92 : SV *sv
93 : CODE:
94 30 : if (!sv || !SvOK(sv))
95 : {
96 2 : RETVAL = cstr2sv("NULL");
97 : }
98 : else
99 : {
100 28 : text *arg = sv2text(sv);
101 28 : text *quoted = DatumGetTextPP(DirectFunctionCall1(quote_nullable, PointerGetDatum(arg)));
102 : char *str;
103 :
104 28 : pfree(arg);
105 28 : str = text_to_cstring(quoted);
106 28 : RETVAL = cstr2sv(str);
107 28 : pfree(str);
108 : }
109 : OUTPUT:
110 : RETVAL
111 :
112 : SV *
113 : util_quote_ident(sv)
114 : SV *sv
115 : PREINIT:
116 : text *arg;
117 : text *quoted;
118 : char *str;
119 : CODE:
120 8 : arg = sv2text(sv);
121 8 : quoted = DatumGetTextPP(DirectFunctionCall1(quote_ident, PointerGetDatum(arg)));
122 :
123 8 : pfree(arg);
124 8 : str = text_to_cstring(quoted);
125 8 : RETVAL = cstr2sv(str);
126 8 : pfree(str);
127 : OUTPUT:
128 : RETVAL
129 :
130 : SV *
131 : util_decode_bytea(sv)
132 : SV *sv
133 : PREINIT:
134 : char *arg;
135 : text *ret;
136 : CODE:
137 4 : arg = SvPVbyte_nolen(sv);
138 4 : ret = DatumGetTextPP(DirectFunctionCall1(byteain, PointerGetDatum(arg)));
139 : /* not cstr2sv because this is raw bytes not utf8'able */
140 4 : RETVAL = newSVpvn(VARDATA_ANY(ret), VARSIZE_ANY_EXHDR(ret));
141 : OUTPUT:
142 : RETVAL
143 :
144 : SV *
145 : util_encode_bytea(sv)
146 : SV *sv
147 : PREINIT:
148 : text *arg;
149 : char *ret;
150 : STRLEN len;
151 : CODE:
152 : /* not sv2text because this is raw bytes not utf8'able */
153 5 : ret = SvPVbyte(sv, len);
154 5 : arg = cstring_to_text_with_len(ret, len);
155 5 : ret = DatumGetCString(DirectFunctionCall1(byteaout, PointerGetDatum(arg)));
156 5 : RETVAL = cstr2sv(ret);
157 : OUTPUT:
158 : RETVAL
159 :
160 : SV *
161 : looks_like_number(sv)
162 : SV *sv
163 : CODE:
164 11 : if (!SvOK(sv))
165 1 : RETVAL = &PL_sv_undef;
166 10 : else if ( looks_like_number(sv) )
167 6 : RETVAL = &PL_sv_yes;
168 : else
169 4 : RETVAL = &PL_sv_no;
170 : OUTPUT:
171 : RETVAL
172 :
173 : SV *
174 : encode_typed_literal(sv, typname)
175 : SV *sv
176 : char *typname;
177 : PREINIT:
178 : char *outstr;
179 : CODE:
180 16 : outstr = plperl_sv_to_literal(sv, typname);
181 15 : if (outstr == NULL)
182 1 : RETVAL = &PL_sv_undef;
183 : else
184 14 : RETVAL = cstr2sv(outstr);
185 : OUTPUT:
186 : RETVAL
187 :
188 : BOOT:
189 24 : items = 0; /* avoid 'unused variable' warning */
|
