1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
|
/*
* pg_column_map.c - PG::ColumnMap class extension
* $Id$
*
*/
#include "pg.h"
#include "util.h"
#ifdef HAVE_INTTYPES_H
#include <inttypes.h>
#endif
VALUE rb_mPG_BinaryEncoder;
/*
* Document-class: PG::BinaryEncoder::Boolean < PG::SimpleEncoder
*
* This is the encoder class for the PostgreSQL boolean type.
*
* It accepts true and false. Other values will raise an exception.
*
*/
static int
pg_bin_enc_boolean(t_pg_coder *conv, VALUE value, char *out, VALUE *intermediate)
{
char mybool;
switch(value){
case Qtrue : mybool = 1; break;
case Qfalse : mybool = 0; break;
default :
rb_raise( rb_eTypeError, "wrong data for binary boolean converter" );
}
if(out) *out = mybool;
return 1;
}
/*
* Document-class: PG::BinaryEncoder::Int2 < PG::SimpleEncoder
*
* This is the encoder class for the PostgreSQL int2 type.
*
* Non-Number values are expected to have method +to_i+ defined.
*
*/
static int
pg_bin_enc_int2(t_pg_coder *conv, VALUE value, char *out, VALUE *intermediate)
{
if(out){
write_nbo16(NUM2INT(*intermediate), out);
}else{
*intermediate = pg_obj_to_i(value);
}
return 2;
}
/*
* Document-class: PG::BinaryEncoder::Int2 < PG::SimpleEncoder
*
* This is the encoder class for the PostgreSQL int4 type.
*
* Non-Number values are expected to have method +to_i+ defined.
*
*/
static int
pg_bin_enc_int4(t_pg_coder *conv, VALUE value, char *out, VALUE *intermediate)
{
if(out){
write_nbo32(NUM2LONG(*intermediate), out);
}else{
*intermediate = pg_obj_to_i(value);
}
return 4;
}
/*
* Document-class: PG::BinaryEncoder::Int2 < PG::SimpleEncoder
*
* This is the encoder class for the PostgreSQL int8 type.
*
* Non-Number values are expected to have method +to_i+ defined.
*
*/
static int
pg_bin_enc_int8(t_pg_coder *conv, VALUE value, char *out, VALUE *intermediate)
{
if(out){
write_nbo64(NUM2LL(*intermediate), out);
}else{
*intermediate = pg_obj_to_i(value);
}
return 8;
}
/*
* Document-class: PG::BinaryEncoder::FromBase64 < PG::CompositeEncoder
*
* This is an encoder class for conversion of base64 encoded data
* to it's binary representation.
*
*/
static int
pg_bin_enc_from_base64(t_pg_coder *conv, VALUE value, char *out, VALUE *intermediate)
{
int strlen;
VALUE subint;
t_pg_composite_coder *this = (t_pg_composite_coder *)conv;
t_pg_coder_enc_func enc_func = pg_coder_enc_func(this->elem);
if(out){
/* Second encoder pass, if required */
strlen = enc_func(this->elem, value, out, intermediate);
strlen = base64_decode( out, out, strlen );
return strlen;
} else {
/* First encoder pass */
strlen = enc_func(this->elem, value, NULL, &subint);
if( strlen == -1 ){
/* Encoded string is returned in subint */
VALUE out_str;
strlen = RSTRING_LENINT(subint);
out_str = rb_str_new(NULL, BASE64_DECODED_SIZE(strlen));
strlen = base64_decode( RSTRING_PTR(out_str), RSTRING_PTR(subint), strlen);
rb_str_set_len( out_str, strlen );
*intermediate = out_str;
return -1;
} else {
*intermediate = subint;
return BASE64_DECODED_SIZE(strlen);
}
}
}
void
init_pg_binary_encoder()
{
/* This module encapsulates all encoder classes with binary output format */
rb_mPG_BinaryEncoder = rb_define_module_under( rb_mPG, "BinaryEncoder" );
/* Make RDoc aware of the encoder classes... */
/* dummy = rb_define_class_under( rb_mPG_BinaryEncoder, "Boolean", rb_cPG_SimpleEncoder ); */
pg_define_coder( "Boolean", pg_bin_enc_boolean, rb_cPG_SimpleEncoder, rb_mPG_BinaryEncoder );
/* dummy = rb_define_class_under( rb_mPG_BinaryEncoder, "Int2", rb_cPG_SimpleEncoder ); */
pg_define_coder( "Int2", pg_bin_enc_int2, rb_cPG_SimpleEncoder, rb_mPG_BinaryEncoder );
/* dummy = rb_define_class_under( rb_mPG_BinaryEncoder, "Int4", rb_cPG_SimpleEncoder ); */
pg_define_coder( "Int4", pg_bin_enc_int4, rb_cPG_SimpleEncoder, rb_mPG_BinaryEncoder );
/* dummy = rb_define_class_under( rb_mPG_BinaryEncoder, "Int8", rb_cPG_SimpleEncoder ); */
pg_define_coder( "Int8", pg_bin_enc_int8, rb_cPG_SimpleEncoder, rb_mPG_BinaryEncoder );
/* dummy = rb_define_class_under( rb_mPG_BinaryEncoder, "String", rb_cPG_SimpleEncoder ); */
pg_define_coder( "String", pg_coder_enc_to_s, rb_cPG_SimpleEncoder, rb_mPG_BinaryEncoder );
/* dummy = rb_define_class_under( rb_mPG_BinaryEncoder, "Bytea", rb_cPG_SimpleEncoder ); */
pg_define_coder( "Bytea", pg_coder_enc_to_s, rb_cPG_SimpleEncoder, rb_mPG_BinaryEncoder );
/* dummy = rb_define_class_under( rb_mPG_BinaryEncoder, "FromBase64", rb_cPG_CompositeEncoder ); */
pg_define_coder( "FromBase64", pg_bin_enc_from_base64, rb_cPG_CompositeEncoder, rb_mPG_BinaryEncoder );
}
|
