pg 0.18.2 → 1.5.3

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (139) hide show
  1. checksums.yaml +5 -5
  2. checksums.yaml.gz.sig +0 -0
  3. data/.appveyor.yml +42 -0
  4. data/.gems +6 -0
  5. data/.github/workflows/binary-gems.yml +117 -0
  6. data/.github/workflows/source-gem.yml +137 -0
  7. data/.gitignore +22 -0
  8. data/.hgsigs +34 -0
  9. data/.hgtags +41 -0
  10. data/.irbrc +23 -0
  11. data/.pryrc +23 -0
  12. data/.tm_properties +21 -0
  13. data/.travis.yml +49 -0
  14. data/BSDL +2 -2
  15. data/Gemfile +14 -0
  16. data/History.md +876 -0
  17. data/Manifest.txt +8 -21
  18. data/README-Windows.rdoc +17 -28
  19. data/README.ja.md +276 -0
  20. data/README.md +286 -0
  21. data/Rakefile +40 -131
  22. data/Rakefile.cross +88 -70
  23. data/certs/ged.pem +24 -0
  24. data/certs/larskanis-2022.pem +26 -0
  25. data/certs/larskanis-2023.pem +24 -0
  26. data/ext/errorcodes.def +113 -0
  27. data/ext/errorcodes.rb +1 -1
  28. data/ext/errorcodes.txt +36 -2
  29. data/ext/extconf.rb +120 -54
  30. data/ext/gvl_wrappers.c +8 -0
  31. data/ext/gvl_wrappers.h +44 -33
  32. data/ext/pg.c +226 -200
  33. data/ext/pg.h +99 -99
  34. data/ext/pg_binary_decoder.c +164 -16
  35. data/ext/pg_binary_encoder.c +249 -22
  36. data/ext/pg_coder.c +189 -44
  37. data/ext/pg_connection.c +1866 -1173
  38. data/ext/pg_copy_coder.c +398 -42
  39. data/ext/pg_errors.c +1 -1
  40. data/ext/pg_record_coder.c +522 -0
  41. data/ext/pg_result.c +727 -232
  42. data/ext/pg_text_decoder.c +629 -43
  43. data/ext/pg_text_encoder.c +269 -102
  44. data/ext/pg_tuple.c +572 -0
  45. data/ext/pg_type_map.c +64 -23
  46. data/ext/pg_type_map_all_strings.c +21 -7
  47. data/ext/pg_type_map_by_class.c +59 -27
  48. data/ext/pg_type_map_by_column.c +86 -43
  49. data/ext/pg_type_map_by_mri_type.c +49 -20
  50. data/ext/pg_type_map_by_oid.c +62 -29
  51. data/ext/pg_type_map_in_ruby.c +56 -22
  52. data/ext/{util.c → pg_util.c} +12 -12
  53. data/ext/{util.h → pg_util.h} +2 -2
  54. data/lib/pg/basic_type_map_based_on_result.rb +67 -0
  55. data/lib/pg/basic_type_map_for_queries.rb +198 -0
  56. data/lib/pg/basic_type_map_for_results.rb +104 -0
  57. data/lib/pg/basic_type_registry.rb +299 -0
  58. data/lib/pg/binary_decoder/date.rb +9 -0
  59. data/lib/pg/binary_decoder/timestamp.rb +26 -0
  60. data/lib/pg/binary_encoder/timestamp.rb +20 -0
  61. data/lib/pg/coder.rb +36 -13
  62. data/lib/pg/connection.rb +797 -77
  63. data/lib/pg/exceptions.rb +16 -2
  64. data/lib/pg/result.rb +24 -7
  65. data/lib/pg/text_decoder/date.rb +18 -0
  66. data/lib/pg/text_decoder/inet.rb +9 -0
  67. data/lib/pg/text_decoder/json.rb +14 -0
  68. data/lib/pg/text_decoder/numeric.rb +9 -0
  69. data/lib/pg/text_decoder/timestamp.rb +30 -0
  70. data/lib/pg/text_encoder/date.rb +12 -0
  71. data/lib/pg/text_encoder/inet.rb +28 -0
  72. data/lib/pg/text_encoder/json.rb +14 -0
  73. data/lib/pg/text_encoder/numeric.rb +9 -0
  74. data/lib/pg/text_encoder/timestamp.rb +24 -0
  75. data/lib/pg/tuple.rb +30 -0
  76. data/lib/pg/type_map_by_column.rb +3 -2
  77. data/lib/pg/version.rb +4 -0
  78. data/lib/pg.rb +106 -41
  79. data/misc/openssl-pg-segfault.rb +31 -0
  80. data/misc/postgres/History.txt +9 -0
  81. data/misc/postgres/Manifest.txt +5 -0
  82. data/misc/postgres/README.txt +21 -0
  83. data/misc/postgres/Rakefile +21 -0
  84. data/misc/postgres/lib/postgres.rb +16 -0
  85. data/misc/ruby-pg/History.txt +9 -0
  86. data/misc/ruby-pg/Manifest.txt +5 -0
  87. data/misc/ruby-pg/README.txt +21 -0
  88. data/misc/ruby-pg/Rakefile +21 -0
  89. data/misc/ruby-pg/lib/ruby/pg.rb +16 -0
  90. data/pg.gemspec +34 -0
  91. data/rakelib/task_extension.rb +46 -0
  92. data/sample/array_insert.rb +1 -1
  93. data/sample/async_api.rb +4 -8
  94. data/sample/async_copyto.rb +1 -1
  95. data/sample/async_mixed.rb +1 -1
  96. data/sample/check_conn.rb +1 -1
  97. data/sample/copydata.rb +71 -0
  98. data/sample/copyfrom.rb +1 -1
  99. data/sample/copyto.rb +1 -1
  100. data/sample/cursor.rb +1 -1
  101. data/sample/disk_usage_report.rb +6 -15
  102. data/sample/issue-119.rb +2 -2
  103. data/sample/losample.rb +1 -1
  104. data/sample/minimal-testcase.rb +2 -2
  105. data/sample/notify_wait.rb +1 -1
  106. data/sample/pg_statistics.rb +6 -15
  107. data/sample/replication_monitor.rb +9 -18
  108. data/sample/test_binary_values.rb +1 -1
  109. data/sample/wal_shipper.rb +2 -2
  110. data/sample/warehouse_partitions.rb +8 -17
  111. data/translation/.po4a-version +7 -0
  112. data/translation/po/all.pot +910 -0
  113. data/translation/po/ja.po +1047 -0
  114. data/translation/po4a.cfg +12 -0
  115. data.tar.gz.sig +0 -0
  116. metadata +137 -204
  117. metadata.gz.sig +0 -0
  118. data/ChangeLog +0 -5545
  119. data/History.rdoc +0 -313
  120. data/README.ja.rdoc +0 -14
  121. data/README.rdoc +0 -161
  122. data/lib/pg/basic_type_mapping.rb +0 -399
  123. data/lib/pg/constants.rb +0 -11
  124. data/lib/pg/text_decoder.rb +0 -42
  125. data/lib/pg/text_encoder.rb +0 -27
  126. data/spec/data/expected_trace.out +0 -26
  127. data/spec/data/random_binary_data +0 -0
  128. data/spec/helpers.rb +0 -355
  129. data/spec/pg/basic_type_mapping_spec.rb +0 -251
  130. data/spec/pg/connection_spec.rb +0 -1535
  131. data/spec/pg/result_spec.rb +0 -449
  132. data/spec/pg/type_map_by_class_spec.rb +0 -138
  133. data/spec/pg/type_map_by_column_spec.rb +0 -222
  134. data/spec/pg/type_map_by_mri_type_spec.rb +0 -136
  135. data/spec/pg/type_map_by_oid_spec.rb +0 -149
  136. data/spec/pg/type_map_in_ruby_spec.rb +0 -164
  137. data/spec/pg/type_map_spec.rb +0 -22
  138. data/spec/pg/type_spec.rb +0 -688
  139. data/spec/pg_spec.rb +0 -50
data/ext/pg.h CHANGED
@@ -9,95 +9,28 @@
9
9
  #include <stdio.h>
10
10
  #include <stdlib.h>
11
11
  #include <sys/types.h>
12
+ #if !defined(_WIN32)
13
+ # include <sys/time.h>
14
+ # include <sys/socket.h>
15
+ #endif
12
16
  #if defined(HAVE_UNISTD_H) && !defined(_WIN32)
13
17
  # include <unistd.h>
14
18
  #endif /* HAVE_UNISTD_H */
15
19
 
16
20
  /* Ruby headers */
17
21
  #include "ruby.h"
18
- #ifdef HAVE_RUBY_ST_H
19
- # include "ruby/st.h"
20
- #elif HAVE_ST_H
21
- # include "st.h"
22
- #endif
22
+ #include "ruby/st.h"
23
+ #include "ruby/encoding.h"
23
24
 
24
- #if defined(HAVE_RUBY_ENCODING_H) && HAVE_RUBY_ENCODING_H
25
- # include "ruby/encoding.h"
26
- # define M17N_SUPPORTED
27
- # ifdef HAVE_RB_ENCDB_ALIAS
28
- extern int rb_encdb_alias(const char *, const char *);
29
- # define ENC_ALIAS(name, orig) rb_encdb_alias((name), (orig))
30
- # elif HAVE_RB_ENC_ALIAS
31
- extern int rb_enc_alias(const char *, const char *);
32
- # define ENC_ALIAS(name, orig) rb_enc_alias((name), (orig))
33
- # else
34
- extern int rb_enc_alias(const char *alias, const char *orig); /* declaration missing in Ruby 1.9.1 */
35
- # define ENC_ALIAS(name, orig) rb_enc_alias((name), (orig))
36
- # endif
37
-
38
-
39
- # if !defined(ENCODING_SET_INLINED)
40
- /* Rubinius doesn't define ENCODING_SET_INLINED, so we fall back to the more
41
- * portable version.
42
- */
43
- # define PG_ENCODING_SET_NOCHECK(obj,i) \
44
- do { \
45
- rb_enc_set_index((obj), (i)); \
46
- } while(0)
47
- # else
48
- # define PG_ENCODING_SET_NOCHECK(obj,i) \
25
+ #define PG_ENCODING_SET_NOCHECK(obj,i) \
49
26
  do { \
50
27
  if ((i) < ENCODING_INLINE_MAX) \
51
28
  ENCODING_SET_INLINED((obj), (i)); \
52
29
  else \
53
30
  rb_enc_set_index((obj), (i)); \
54
31
  } while(0)
55
- # endif
56
-
57
- #else
58
- # define PG_ENCODING_SET_NOCHECK(obj,i) /* nothing */
59
- #endif
60
-
61
- #if RUBY_VM != 1
62
- # define RUBY_18_COMPAT
63
- #endif
64
-
65
- #ifndef RARRAY_LEN
66
- # define RARRAY_LEN(x) RARRAY((x))->len
67
- #endif /* RARRAY_LEN */
68
-
69
- #ifndef RSTRING_LEN
70
- # define RSTRING_LEN(x) RSTRING((x))->len
71
- #endif /* RSTRING_LEN */
72
32
 
73
- #ifndef RSTRING_PTR
74
- # define RSTRING_PTR(x) RSTRING((x))->ptr
75
- #endif /* RSTRING_PTR */
76
-
77
- #ifndef StringValuePtr
78
- # define StringValuePtr(x) STR2CSTR(x)
79
- #endif /* StringValuePtr */
80
-
81
- #ifdef RUBY_18_COMPAT
82
- # define rb_io_stdio_file GetWriteFile
83
- # include "rubyio.h"
84
- #else
85
- # include "ruby/io.h"
86
- #endif
87
-
88
- #ifdef RUBINIUS
89
- /* Workaround for wrong FIXNUM_MAX definition */
90
- typedef intptr_t native_int;
91
- #endif
92
-
93
- #ifndef RETURN_SIZED_ENUMERATOR
94
- #define RETURN_SIZED_ENUMERATOR(obj, argc, argv, size_fn) RETURN_ENUMERATOR((obj), (argc), (argv))
95
- #endif
96
-
97
- #ifndef HAVE_RB_HASH_DUP
98
- /* Rubinius doesn't define rb_hash_dup() */
99
- #define rb_hash_dup(tuple) rb_funcall((tuple), rb_intern("dup"), 0)
100
- #endif
33
+ #include "ruby/io.h"
101
34
 
102
35
  #ifndef timeradd
103
36
  #define timeradd(a, b, result) \
@@ -124,6 +57,7 @@
124
57
  #endif
125
58
 
126
59
  /* PostgreSQL headers */
60
+ #include "pg_config.h"
127
61
  #include "libpq-fe.h"
128
62
  #include "libpq/libpq-fs.h" /* large-object interface */
129
63
  #include "pg_config_manual.h"
@@ -142,12 +76,32 @@ typedef long suseconds_t;
142
76
  #define PG_MAX_COLUMNS 4000
143
77
  #endif
144
78
 
79
+ #ifdef HAVE_RB_GC_MARK_MOVABLE
80
+ #define pg_compact_callback(x) (x)
81
+ #define pg_gc_location(x) x = rb_gc_location(x)
82
+ #else
83
+ #define rb_gc_mark_movable(x) rb_gc_mark(x)
84
+ #define pg_compact_callback(x) {(x)}
85
+ #define pg_gc_location(x) UNUSED(x)
86
+ #endif
87
+
88
+ /* For compatibility with ruby < 3.0 */
89
+ #ifndef RUBY_TYPED_FROZEN_SHAREABLE
90
+ #define PG_RUBY_TYPED_FROZEN_SHAREABLE 0
91
+ #else
92
+ #define PG_RUBY_TYPED_FROZEN_SHAREABLE RUBY_TYPED_FROZEN_SHAREABLE
93
+ #endif
94
+ #define PG_ENC_IDX_BITS 28
95
+
145
96
  /* The data behind each PG::Connection object */
146
97
  typedef struct {
147
98
  PGconn *pgconn;
148
99
 
149
100
  /* Cached IO object for the socket descriptor */
150
101
  VALUE socket_io;
102
+ /* function pointers of the original libpq notice receivers */
103
+ PQnoticeReceiver default_notice_receiver;
104
+ PQnoticeProcessor default_notice_processor;
151
105
  /* Proc object that receives notices as PG::Result objects */
152
106
  VALUE notice_receiver;
153
107
  /* Proc object that receives notices as String objects */
@@ -158,13 +112,21 @@ typedef struct {
158
112
  VALUE type_map_for_results;
159
113
  /* IO object internally used for the trace stream */
160
114
  VALUE trace_stream;
161
- /* Cached Encoding object */
162
- VALUE external_encoding;
163
115
  /* Kind of PG::Coder object for casting ruby values to COPY rows */
164
116
  VALUE encoder_for_put_copy_data;
165
117
  /* Kind of PG::Coder object for casting COPY rows to ruby values */
166
118
  VALUE decoder_for_get_copy_data;
119
+ /* Ruby encoding index of the client/internal encoding */
120
+ int enc_idx : PG_ENC_IDX_BITS;
121
+ /* flags controlling Symbol/String field names */
122
+ unsigned int flags : 2;
123
+ /* enable automatic flushing of send data at the end of send_query calls */
124
+ unsigned int flush_data : 1;
167
125
 
126
+ #if defined(_WIN32)
127
+ /* File descriptor to be used for rb_w32_unwrap_io_handle() */
128
+ int ruby_sd;
129
+ #endif
168
130
  } t_pg_connection;
169
131
 
170
132
  typedef struct pg_coder t_pg_coder;
@@ -185,28 +147,40 @@ typedef struct {
185
147
  */
186
148
  t_typemap *p_typemap;
187
149
 
150
+ /* Ruby encoding index of the client/internal encoding */
151
+ int enc_idx : PG_ENC_IDX_BITS;
152
+
188
153
  /* 0 = PGresult is cleared by PG::Result#clear or by the GC
189
154
  * 1 = PGresult is cleared internally by libpq
190
155
  */
191
- int autoclear;
156
+ unsigned int autoclear : 1;
157
+
158
+ /* flags controlling Symbol/String field names */
159
+ unsigned int flags : 2;
192
160
 
193
161
  /* Number of fields in fnames[] .
194
162
  * Set to -1 if fnames[] is not yet initialized.
195
163
  */
196
164
  int nfields;
197
165
 
166
+ /* Size of PGresult as published to ruby memory management. */
167
+ ssize_t result_size;
168
+
198
169
  /* Prefilled tuple Hash with fnames[] as keys. */
199
170
  VALUE tuple_hash;
200
171
 
201
- /* List of field names as frozen String objects.
172
+ /* Hash with fnames[] to field number mapping. */
173
+ VALUE field_map;
174
+
175
+ /* List of field names as frozen String or Symbol objects.
202
176
  * Only valid if nfields != -1
203
177
  */
204
178
  VALUE fnames[0];
205
179
  } t_pg_result;
206
180
 
207
181
 
208
- typedef int (* t_pg_coder_enc_func)(t_pg_coder *, VALUE, char *, VALUE *);
209
- typedef VALUE (* t_pg_coder_dec_func)(t_pg_coder *, char *, int, int, int, int);
182
+ typedef int (* t_pg_coder_enc_func)(t_pg_coder *, VALUE, char *, VALUE *, int);
183
+ typedef VALUE (* t_pg_coder_dec_func)(t_pg_coder *, const char *, int, int, int, int);
210
184
  typedef VALUE (* t_pg_fit_to_result)(VALUE, VALUE);
211
185
  typedef VALUE (* t_pg_fit_to_query)(VALUE, VALUE);
212
186
  typedef int (* t_pg_fit_to_copy_get)(VALUE);
@@ -214,12 +188,27 @@ typedef VALUE (* t_pg_typecast_result)(t_typemap *, VALUE, int, int);
214
188
  typedef t_pg_coder *(* t_pg_typecast_query_param)(t_typemap *, VALUE, int);
215
189
  typedef VALUE (* t_pg_typecast_copy_get)( t_typemap *, VALUE, int, int, int );
216
190
 
191
+ #define PG_RESULT_FIELD_NAMES_MASK 0x03
192
+ #define PG_RESULT_FIELD_NAMES_SYMBOL 0x01
193
+ #define PG_RESULT_FIELD_NAMES_STATIC_SYMBOL 0x02
194
+
195
+ #define PG_CODER_TIMESTAMP_DB_UTC 0x0
196
+ #define PG_CODER_TIMESTAMP_DB_LOCAL 0x1
197
+ #define PG_CODER_TIMESTAMP_APP_UTC 0x0
198
+ #define PG_CODER_TIMESTAMP_APP_LOCAL 0x2
199
+ #define PG_CODER_FORMAT_ERROR_MASK 0xc
200
+ #define PG_CODER_FORMAT_ERROR_TO_RAISE 0x4
201
+ #define PG_CODER_FORMAT_ERROR_TO_STRING 0x8
202
+ #define PG_CODER_FORMAT_ERROR_TO_PARTIAL 0xc
203
+
217
204
  struct pg_coder {
218
205
  t_pg_coder_enc_func enc_func;
219
206
  t_pg_coder_dec_func dec_func;
220
207
  VALUE coder_obj;
221
208
  Oid oid;
222
209
  int format;
210
+ /* OR-ed values out of PG_CODER_* */
211
+ int flags;
223
212
  };
224
213
 
225
214
  typedef struct {
@@ -249,6 +238,8 @@ typedef struct {
249
238
  } convs[0];
250
239
  } t_tmbc;
251
240
 
241
+ extern const rb_data_type_t pg_typemap_type;
242
+ extern const rb_data_type_t pg_coder_type;
252
243
 
253
244
  #include "gvl_wrappers.h"
254
245
 
@@ -256,6 +247,7 @@ typedef struct {
256
247
  * Globals
257
248
  **************************************************************************/
258
249
 
250
+ extern int pg_skip_deprecation_warning;
259
251
  extern VALUE rb_mPG;
260
252
  extern VALUE rb_ePGerror;
261
253
  extern VALUE rb_eServerError;
@@ -314,22 +306,25 @@ void init_pg_type_map_by_oid _(( void ));
314
306
  void init_pg_type_map_in_ruby _(( void ));
315
307
  void init_pg_coder _(( void ));
316
308
  void init_pg_copycoder _(( void ));
309
+ void init_pg_recordcoder _(( void ));
317
310
  void init_pg_text_encoder _(( void ));
318
311
  void init_pg_text_decoder _(( void ));
319
312
  void init_pg_binary_encoder _(( void ));
320
313
  void init_pg_binary_decoder _(( void ));
314
+ void init_pg_tuple _(( void ));
321
315
  VALUE lookup_error_class _(( const char * ));
322
- VALUE pg_bin_dec_bytea _(( t_pg_coder*, char *, int, int, int, int ));
323
- VALUE pg_text_dec_string _(( t_pg_coder*, char *, int, int, int, int ));
324
- int pg_coder_enc_to_s _(( t_pg_coder*, VALUE, char *, VALUE *));
325
- int pg_text_enc_identifier _(( t_pg_coder*, VALUE, char *, VALUE *));
316
+ VALUE pg_bin_dec_bytea _(( t_pg_coder*, const char *, int, int, int, int ));
317
+ VALUE pg_text_dec_string _(( t_pg_coder*, const char *, int, int, int, int ));
318
+ int pg_coder_enc_to_s _(( t_pg_coder*, VALUE, char *, VALUE *, int));
319
+ int pg_text_enc_identifier _(( t_pg_coder*, VALUE, char *, VALUE *, int));
326
320
  t_pg_coder_enc_func pg_coder_enc_func _(( t_pg_coder* ));
327
321
  t_pg_coder_dec_func pg_coder_dec_func _(( t_pg_coder*, int ));
328
- void pg_define_coder _(( const char *, void *, VALUE, VALUE ));
322
+ VALUE pg_define_coder _(( const char *, void *, VALUE, VALUE ));
329
323
  VALUE pg_obj_to_i _(( VALUE ));
330
324
  VALUE pg_tmbc_allocate _(( void ));
331
325
  void pg_coder_init_encoder _(( VALUE ));
332
326
  void pg_coder_init_decoder _(( VALUE ));
327
+ void pg_coder_compact _(( void * ));
333
328
  char *pg_rb_str_ensure_capa _(( VALUE, long, char *, char ** ));
334
329
 
335
330
  #define PG_RB_STR_ENSURE_CAPA( str, expand_len, curr_ptr, end_ptr ) \
@@ -343,26 +338,26 @@ char *pg_rb_str_ensure_capa _(( VALUE, long, char *,
343
338
  (curr_ptr) = (end_ptr) = RSTRING_PTR(str) \
344
339
  )
345
340
 
346
- #define PG_RB_TAINTED_STR_NEW( str, curr_ptr, end_ptr ) ( \
347
- (str) = rb_tainted_str_new( NULL, 0 ), \
348
- (curr_ptr) = (end_ptr) = RSTRING_PTR(str) \
349
- )
350
-
351
341
  VALUE pg_typemap_fit_to_result _(( VALUE, VALUE ));
352
342
  VALUE pg_typemap_fit_to_query _(( VALUE, VALUE ));
353
343
  int pg_typemap_fit_to_copy_get _(( VALUE ));
354
344
  VALUE pg_typemap_result_value _(( t_typemap *, VALUE, int, int ));
355
345
  t_pg_coder *pg_typemap_typecast_query_param _(( t_typemap *, VALUE, int ));
356
346
  VALUE pg_typemap_typecast_copy_get _(( t_typemap *, VALUE, int, int, int ));
347
+ void pg_typemap_mark _(( void * ));
348
+ size_t pg_typemap_memsize _(( const void * ));
349
+ void pg_typemap_compact _(( void * ));
357
350
 
358
351
  PGconn *pg_get_pgconn _(( VALUE ));
359
352
  t_pg_connection *pg_get_connection _(( VALUE ));
353
+ VALUE pgconn_block _(( int, VALUE *, VALUE ));
360
354
 
361
355
  VALUE pg_new_result _(( PGresult *, VALUE ));
362
356
  VALUE pg_new_result_autoclear _(( PGresult *, VALUE ));
363
357
  PGresult* pgresult_get _(( VALUE ));
364
358
  VALUE pg_result_check _(( VALUE ));
365
359
  VALUE pg_result_clear _(( VALUE ));
360
+ VALUE pg_tuple_new _(( VALUE, int ));
366
361
 
367
362
  /*
368
363
  * Fetch the data pointer for the result object
@@ -370,23 +365,28 @@ VALUE pg_result_clear _(( VALUE ));
370
365
  static inline t_pg_result *
371
366
  pgresult_get_this( VALUE self )
372
367
  {
373
- t_pg_result *this = DATA_PTR(self);
374
-
375
- if( this == NULL )
376
- rb_raise(rb_ePGerror, "result has been cleared");
377
-
378
- return this;
368
+ return RTYPEDDATA_DATA(self);
379
369
  }
380
370
 
381
371
 
382
- #ifdef M17N_SUPPORTED
383
372
  rb_encoding * pg_get_pg_encoding_as_rb_encoding _(( int ));
384
373
  rb_encoding * pg_get_pg_encname_as_rb_encoding _(( const char * ));
385
374
  const char * pg_get_rb_encoding_as_pg_encoding _(( rb_encoding * ));
386
375
  rb_encoding *pg_conn_enc_get _(( PGconn * ));
387
- #endif /* M17N_SUPPORTED */
388
376
 
389
377
  void notice_receiver_proxy(void *arg, const PGresult *result);
390
378
  void notice_processor_proxy(void *arg, const char *message);
391
379
 
380
+ /* reports if `-W' specified and PG_SKIP_DEPRECATION_WARNING environment variable isn't set
381
+ *
382
+ * message_id identifies the warning, so that it's reported only once.
383
+ */
384
+ #define pg_deprecated(message_id, format_args) \
385
+ do { \
386
+ if( !(pg_skip_deprecation_warning & (1 << message_id)) ){ \
387
+ pg_skip_deprecation_warning |= 1 << message_id; \
388
+ rb_warning format_args; \
389
+ } \
390
+ } while(0);
391
+
392
392
  #endif /* end __pg_h */
@@ -1,25 +1,30 @@
1
1
  /*
2
2
  * pg_column_map.c - PG::ColumnMap class extension
3
- * $Id: pg_binary_decoder.c,v 185638b52684 2014/11/08 20:43:53 lars $
3
+ * $Id$
4
4
  *
5
5
  */
6
6
 
7
+ #include "ruby/version.h"
7
8
  #include "pg.h"
8
- #include "util.h"
9
+ #include "pg_util.h"
10
+ #ifdef HAVE_INTTYPES_H
9
11
  #include <inttypes.h>
12
+ #endif
10
13
 
11
14
  VALUE rb_mPG_BinaryDecoder;
15
+ static VALUE s_Date;
16
+ static ID s_id_new;
12
17
 
13
18
 
14
19
  /*
15
20
  * Document-class: PG::BinaryDecoder::Boolean < PG::SimpleDecoder
16
21
  *
17
- * This is a decoder class for conversion of PostgreSQL binary bool type
18
- * to Ruby true or false objects.
22
+ * This is a decoder class for conversion of PostgreSQL binary +bool+ type
23
+ * to Ruby +true+ or +false+ objects.
19
24
  *
20
25
  */
21
26
  static VALUE
22
- pg_bin_dec_boolean(t_pg_coder *conv, char *val, int len, int tuple, int field, int enc_idx)
27
+ pg_bin_dec_boolean(t_pg_coder *conv, const char *val, int len, int tuple, int field, int enc_idx)
23
28
  {
24
29
  if (len < 1) {
25
30
  rb_raise( rb_eTypeError, "wrong data for binary boolean converter in tuple %d field %d", tuple, field);
@@ -30,12 +35,12 @@ pg_bin_dec_boolean(t_pg_coder *conv, char *val, int len, int tuple, int field, i
30
35
  /*
31
36
  * Document-class: PG::BinaryDecoder::Integer < PG::SimpleDecoder
32
37
  *
33
- * This is a decoder class for conversion of PostgreSQL binary int2, int4 and int8 types
38
+ * This is a decoder class for conversion of PostgreSQL binary +int2+, +int4+ and +int8+ types
34
39
  * to Ruby Integer objects.
35
40
  *
36
41
  */
37
42
  static VALUE
38
- pg_bin_dec_integer(t_pg_coder *conv, char *val, int len, int tuple, int field, int enc_idx)
43
+ pg_bin_dec_integer(t_pg_coder *conv, const char *val, int len, int tuple, int field, int enc_idx)
39
44
  {
40
45
  switch( len ){
41
46
  case 2:
@@ -52,12 +57,12 @@ pg_bin_dec_integer(t_pg_coder *conv, char *val, int len, int tuple, int field, i
52
57
  /*
53
58
  * Document-class: PG::BinaryDecoder::Float < PG::SimpleDecoder
54
59
  *
55
- * This is a decoder class for conversion of PostgreSQL binary float4 and float8 types
60
+ * This is a decoder class for conversion of PostgreSQL binary +float4+ and +float8+ types
56
61
  * to Ruby Float objects.
57
62
  *
58
63
  */
59
64
  static VALUE
60
- pg_bin_dec_float(t_pg_coder *conv, char *val, int len, int tuple, int field, int enc_idx)
65
+ pg_bin_dec_float(t_pg_coder *conv, const char *val, int len, int tuple, int field, int enc_idx)
61
66
  {
62
67
  union {
63
68
  float f;
@@ -84,15 +89,15 @@ pg_bin_dec_float(t_pg_coder *conv, char *val, int len, int tuple, int field, int
84
89
  * Document-class: PG::BinaryDecoder::Bytea < PG::SimpleDecoder
85
90
  *
86
91
  * This decoder class delivers the data received from the server as binary String object.
87
- * It is therefore suitable for conversion of PostgreSQL bytea data as well as any other
92
+ * It is therefore suitable for conversion of PostgreSQL +bytea+ data as well as any other
88
93
  * data in binary format.
89
94
  *
90
95
  */
91
96
  VALUE
92
- pg_bin_dec_bytea(t_pg_coder *conv, char *val, int len, int tuple, int field, int enc_idx)
97
+ pg_bin_dec_bytea(t_pg_coder *conv, const char *val, int len, int tuple, int field, int enc_idx)
93
98
  {
94
99
  VALUE ret;
95
- ret = rb_tainted_str_new( val, len );
100
+ ret = rb_str_new( val, len );
96
101
  PG_ENCODING_SET_NOCHECK( ret, rb_ascii8bit_encindex() );
97
102
  return ret;
98
103
  }
@@ -100,17 +105,17 @@ pg_bin_dec_bytea(t_pg_coder *conv, char *val, int len, int tuple, int field, int
100
105
  /*
101
106
  * Document-class: PG::BinaryDecoder::ToBase64 < PG::CompositeDecoder
102
107
  *
103
- * This is a decoder class for conversion of binary (bytea) to base64 data.
108
+ * This is a decoder class for conversion of binary +bytea+ to base64 data.
104
109
  *
105
110
  */
106
111
  static VALUE
107
- pg_bin_dec_to_base64(t_pg_coder *conv, char *val, int len, int tuple, int field, int enc_idx)
112
+ pg_bin_dec_to_base64(t_pg_coder *conv, const char *val, int len, int tuple, int field, int enc_idx)
108
113
  {
109
114
  t_pg_composite_coder *this = (t_pg_composite_coder *)conv;
110
115
  t_pg_coder_dec_func dec_func = pg_coder_dec_func(this->elem, this->comp.format);
111
116
  int encoded_len = BASE64_ENCODED_SIZE(len);
112
117
  /* create a buffer of the encoded length */
113
- VALUE out_value = rb_tainted_str_new(NULL, encoded_len);
118
+ VALUE out_value = rb_str_new(NULL, encoded_len);
114
119
 
115
120
  base64_encode( RSTRING_PTR(out_value), val, len );
116
121
 
@@ -128,6 +133,146 @@ pg_bin_dec_to_base64(t_pg_coder *conv, char *val, int len, int tuple, int field,
128
133
  return out_value;
129
134
  }
130
135
 
136
+ #define PG_INT64_MIN (-0x7FFFFFFFFFFFFFFFL - 1)
137
+ #define PG_INT64_MAX 0x7FFFFFFFFFFFFFFFL
138
+
139
+ /*
140
+ * Document-class: PG::BinaryDecoder::Timestamp < PG::SimpleDecoder
141
+ *
142
+ * This is a decoder class for conversion of PostgreSQL binary timestamps
143
+ * to Ruby Time objects.
144
+ *
145
+ * The following flags can be used to specify timezone interpretation:
146
+ * * +PG::Coder::TIMESTAMP_DB_UTC+ : Interpret timestamp as UTC time (default)
147
+ * * +PG::Coder::TIMESTAMP_DB_LOCAL+ : Interpret timestamp as local time
148
+ * * +PG::Coder::TIMESTAMP_APP_UTC+ : Return timestamp as UTC time (default)
149
+ * * +PG::Coder::TIMESTAMP_APP_LOCAL+ : Return timestamp as local time
150
+ *
151
+ * Example:
152
+ * deco = PG::BinaryDecoder::Timestamp.new(flags: PG::Coder::TIMESTAMP_DB_UTC | PG::Coder::TIMESTAMP_APP_LOCAL)
153
+ * deco.decode("\0"*8) # => 2000-01-01 01:00:00 +0100
154
+ */
155
+ static VALUE
156
+ pg_bin_dec_timestamp(t_pg_coder *conv, const char *val, int len, int tuple, int field, int enc_idx)
157
+ {
158
+ int64_t timestamp;
159
+ int64_t sec;
160
+ int64_t nsec;
161
+ VALUE t;
162
+
163
+ if( len != sizeof(timestamp) ){
164
+ rb_raise( rb_eTypeError, "wrong data for timestamp converter in tuple %d field %d length %d", tuple, field, len);
165
+ }
166
+
167
+ timestamp = read_nbo64(val);
168
+
169
+ switch(timestamp){
170
+ case PG_INT64_MAX:
171
+ return rb_str_new2("infinity");
172
+ case PG_INT64_MIN:
173
+ return rb_str_new2("-infinity");
174
+ default:
175
+ /* PostgreSQL's timestamp is based on year 2000 and Ruby's time is based on 1970.
176
+ * Adjust the 30 years difference. */
177
+ sec = (timestamp / 1000000) + 10957L * 24L * 3600L;
178
+ nsec = (timestamp % 1000000) * 1000;
179
+
180
+ #if (RUBY_API_VERSION_MAJOR > 2 || (RUBY_API_VERSION_MAJOR == 2 && RUBY_API_VERSION_MINOR >= 3)) && defined(NEGATIVE_TIME_T) && defined(SIZEOF_TIME_T) && SIZEOF_TIME_T >= 8
181
+ /* Fast path for time conversion */
182
+ {
183
+ struct timespec ts = {sec, nsec};
184
+ t = rb_time_timespec_new(&ts, conv->flags & PG_CODER_TIMESTAMP_APP_LOCAL ? INT_MAX : INT_MAX-1);
185
+ }
186
+ #else
187
+ t = rb_funcall(rb_cTime, rb_intern("at"), 2, LL2NUM(sec), LL2NUM(nsec / 1000));
188
+ if( !(conv->flags & PG_CODER_TIMESTAMP_APP_LOCAL) ) {
189
+ t = rb_funcall(t, rb_intern("utc"), 0);
190
+ }
191
+ #endif
192
+ if( conv->flags & PG_CODER_TIMESTAMP_DB_LOCAL ) {
193
+ /* interpret it as local time */
194
+ t = rb_funcall(t, rb_intern("-"), 1, rb_funcall(t, rb_intern("utc_offset"), 0));
195
+ }
196
+ return t;
197
+ }
198
+ }
199
+
200
+ #define PG_INT32_MIN (-0x7FFFFFFF-1)
201
+ #define PG_INT32_MAX (0x7FFFFFFF)
202
+ #define POSTGRES_EPOCH_JDATE 2451545 /* == date2j(2000, 1, 1) */
203
+ #define MONTHS_PER_YEAR 12
204
+
205
+ /* taken from PostgreSQL sources at src/backend/utils/adt/datetime.c */
206
+ void
207
+ j2date(int jd, int *year, int *month, int *day)
208
+ {
209
+ unsigned int julian;
210
+ unsigned int quad;
211
+ unsigned int extra;
212
+ int y;
213
+
214
+ julian = jd;
215
+ julian += 32044;
216
+ quad = julian / 146097;
217
+ extra = (julian - quad * 146097) * 4 + 3;
218
+ julian += 60 + quad * 3 + extra / 146097;
219
+ quad = julian / 1461;
220
+ julian -= quad * 1461;
221
+ y = julian * 4 / 1461;
222
+ julian = ((y != 0) ? ((julian + 305) % 365) : ((julian + 306) % 366))
223
+ + 123;
224
+ y += quad * 4;
225
+ *year = y - 4800;
226
+ quad = julian * 2141 / 65536;
227
+ *day = julian - 7834 * quad / 256;
228
+ *month = (quad + 10) % MONTHS_PER_YEAR + 1;
229
+ } /* j2date() */
230
+
231
+ /*
232
+ * Document-class: PG::BinaryDecoder::Date < PG::SimpleDecoder
233
+ *
234
+ * This is a decoder class for conversion of PostgreSQL binary date
235
+ * to Ruby Date objects.
236
+ */
237
+ static VALUE
238
+ pg_bin_dec_date(t_pg_coder *conv, const char *val, int len, int tuple, int field, int enc_idx)
239
+ {
240
+ int year, month, day;
241
+ int date;
242
+
243
+ if (len != 4) {
244
+ rb_raise(rb_eTypeError, "unexpected date format != 4 bytes");
245
+ }
246
+
247
+ date = read_nbo32(val);
248
+ switch(date){
249
+ case PG_INT32_MAX:
250
+ return rb_str_new2("infinity");
251
+ case PG_INT32_MIN:
252
+ return rb_str_new2("-infinity");
253
+ default:
254
+ j2date(date + POSTGRES_EPOCH_JDATE, &year, &month, &day);
255
+
256
+ return rb_funcall(s_Date, s_id_new, 3, INT2NUM(year), INT2NUM(month), INT2NUM(day));
257
+ }
258
+ }
259
+
260
+ /* called per autoload when BinaryDecoder::Date is used */
261
+ static VALUE
262
+ init_pg_bin_decoder_date(VALUE rb_mPG_BinaryDecoder)
263
+ {
264
+ rb_require("date");
265
+ s_Date = rb_const_get(rb_cObject, rb_intern("Date"));
266
+ rb_gc_register_mark_object(s_Date);
267
+ s_id_new = rb_intern("new");
268
+
269
+ /* dummy = rb_define_class_under( rb_mPG_BinaryDecoder, "Date", rb_cPG_SimpleDecoder ); */
270
+ pg_define_coder( "Date", pg_bin_dec_date, rb_cPG_SimpleDecoder, rb_mPG_BinaryDecoder );
271
+
272
+ return Qnil;
273
+ }
274
+
275
+
131
276
  /*
132
277
  * Document-class: PG::BinaryDecoder::String < PG::SimpleDecoder
133
278
  *
@@ -138,10 +283,11 @@ pg_bin_dec_to_base64(t_pg_coder *conv, char *val, int len, int tuple, int field,
138
283
  */
139
284
 
140
285
  void
141
- init_pg_binary_decoder()
286
+ init_pg_binary_decoder(void)
142
287
  {
143
288
  /* This module encapsulates all decoder classes with binary input format */
144
289
  rb_mPG_BinaryDecoder = rb_define_module_under( rb_mPG, "BinaryDecoder" );
290
+ rb_define_private_method(rb_singleton_class(rb_mPG_BinaryDecoder), "init_date", init_pg_bin_decoder_date, 0);
145
291
 
146
292
  /* Make RDoc aware of the decoder classes... */
147
293
  /* dummy = rb_define_class_under( rb_mPG_BinaryDecoder, "Boolean", rb_cPG_SimpleDecoder ); */
@@ -154,6 +300,8 @@ init_pg_binary_decoder()
154
300
  pg_define_coder( "String", pg_text_dec_string, rb_cPG_SimpleDecoder, rb_mPG_BinaryDecoder );
155
301
  /* dummy = rb_define_class_under( rb_mPG_BinaryDecoder, "Bytea", rb_cPG_SimpleDecoder ); */
156
302
  pg_define_coder( "Bytea", pg_bin_dec_bytea, rb_cPG_SimpleDecoder, rb_mPG_BinaryDecoder );
303
+ /* dummy = rb_define_class_under( rb_mPG_BinaryDecoder, "Timestamp", rb_cPG_SimpleDecoder ); */
304
+ pg_define_coder( "Timestamp", pg_bin_dec_timestamp, rb_cPG_SimpleDecoder, rb_mPG_BinaryDecoder );
157
305
 
158
306
  /* dummy = rb_define_class_under( rb_mPG_BinaryDecoder, "ToBase64", rb_cPG_CompositeDecoder ); */
159
307
  pg_define_coder( "ToBase64", pg_bin_dec_to_base64, rb_cPG_CompositeDecoder, rb_mPG_BinaryDecoder );