pg 0.21.0 → 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/Gemfile +14 -0
  15. data/History.md +876 -0
  16. data/Manifest.txt +8 -4
  17. data/README-Windows.rdoc +4 -4
  18. data/README.ja.md +276 -0
  19. data/README.md +286 -0
  20. data/Rakefile +38 -139
  21. data/Rakefile.cross +70 -74
  22. data/certs/ged.pem +24 -0
  23. data/certs/larskanis-2022.pem +26 -0
  24. data/certs/larskanis-2023.pem +24 -0
  25. data/ext/errorcodes.def +88 -0
  26. data/ext/errorcodes.rb +1 -1
  27. data/ext/errorcodes.txt +24 -2
  28. data/ext/extconf.rb +111 -54
  29. data/ext/gvl_wrappers.c +8 -0
  30. data/ext/gvl_wrappers.h +40 -33
  31. data/ext/pg.c +224 -199
  32. data/ext/pg.h +93 -96
  33. data/ext/pg_binary_decoder.c +162 -16
  34. data/ext/pg_binary_encoder.c +238 -13
  35. data/ext/pg_coder.c +159 -35
  36. data/ext/pg_connection.c +1585 -1046
  37. data/ext/pg_copy_coder.c +364 -38
  38. data/ext/pg_errors.c +1 -1
  39. data/ext/pg_record_coder.c +522 -0
  40. data/ext/pg_result.c +708 -219
  41. data/ext/pg_text_decoder.c +627 -43
  42. data/ext/pg_text_encoder.c +206 -62
  43. data/ext/pg_tuple.c +572 -0
  44. data/ext/pg_type_map.c +45 -11
  45. data/ext/pg_type_map_all_strings.c +21 -7
  46. data/ext/pg_type_map_by_class.c +59 -27
  47. data/ext/pg_type_map_by_column.c +80 -37
  48. data/ext/pg_type_map_by_mri_type.c +49 -20
  49. data/ext/pg_type_map_by_oid.c +62 -29
  50. data/ext/pg_type_map_in_ruby.c +56 -22
  51. data/ext/{util.c → pg_util.c} +12 -12
  52. data/ext/{util.h → pg_util.h} +2 -2
  53. data/lib/pg/basic_type_map_based_on_result.rb +67 -0
  54. data/lib/pg/basic_type_map_for_queries.rb +198 -0
  55. data/lib/pg/basic_type_map_for_results.rb +104 -0
  56. data/lib/pg/basic_type_registry.rb +299 -0
  57. data/lib/pg/binary_decoder/date.rb +9 -0
  58. data/lib/pg/binary_decoder/timestamp.rb +26 -0
  59. data/lib/pg/binary_encoder/timestamp.rb +20 -0
  60. data/lib/pg/coder.rb +36 -13
  61. data/lib/pg/connection.rb +755 -71
  62. data/lib/pg/exceptions.rb +16 -2
  63. data/lib/pg/result.rb +14 -2
  64. data/lib/pg/text_decoder/date.rb +18 -0
  65. data/lib/pg/text_decoder/inet.rb +9 -0
  66. data/lib/pg/text_decoder/json.rb +14 -0
  67. data/lib/pg/text_decoder/numeric.rb +9 -0
  68. data/lib/pg/text_decoder/timestamp.rb +30 -0
  69. data/lib/pg/text_encoder/date.rb +12 -0
  70. data/lib/pg/text_encoder/inet.rb +28 -0
  71. data/lib/pg/text_encoder/json.rb +14 -0
  72. data/lib/pg/text_encoder/numeric.rb +9 -0
  73. data/lib/pg/text_encoder/timestamp.rb +24 -0
  74. data/lib/pg/tuple.rb +30 -0
  75. data/lib/pg/type_map_by_column.rb +3 -2
  76. data/lib/pg/version.rb +4 -0
  77. data/lib/pg.rb +96 -43
  78. data/misc/openssl-pg-segfault.rb +31 -0
  79. data/misc/postgres/History.txt +9 -0
  80. data/misc/postgres/Manifest.txt +5 -0
  81. data/misc/postgres/README.txt +21 -0
  82. data/misc/postgres/Rakefile +21 -0
  83. data/misc/postgres/lib/postgres.rb +16 -0
  84. data/misc/ruby-pg/History.txt +9 -0
  85. data/misc/ruby-pg/Manifest.txt +5 -0
  86. data/misc/ruby-pg/README.txt +21 -0
  87. data/misc/ruby-pg/Rakefile +21 -0
  88. data/misc/ruby-pg/lib/ruby/pg.rb +16 -0
  89. data/pg.gemspec +34 -0
  90. data/rakelib/task_extension.rb +46 -0
  91. data/sample/array_insert.rb +20 -0
  92. data/sample/async_api.rb +102 -0
  93. data/sample/async_copyto.rb +39 -0
  94. data/sample/async_mixed.rb +56 -0
  95. data/sample/check_conn.rb +21 -0
  96. data/sample/copydata.rb +71 -0
  97. data/sample/copyfrom.rb +81 -0
  98. data/sample/copyto.rb +19 -0
  99. data/sample/cursor.rb +21 -0
  100. data/sample/disk_usage_report.rb +177 -0
  101. data/sample/issue-119.rb +94 -0
  102. data/sample/losample.rb +69 -0
  103. data/sample/minimal-testcase.rb +17 -0
  104. data/sample/notify_wait.rb +72 -0
  105. data/sample/pg_statistics.rb +285 -0
  106. data/sample/replication_monitor.rb +222 -0
  107. data/sample/test_binary_values.rb +33 -0
  108. data/sample/wal_shipper.rb +434 -0
  109. data/sample/warehouse_partitions.rb +311 -0
  110. data/translation/.po4a-version +7 -0
  111. data/translation/po/all.pot +910 -0
  112. data/translation/po/ja.po +1047 -0
  113. data/translation/po4a.cfg +12 -0
  114. data.tar.gz.sig +0 -0
  115. metadata +154 -217
  116. metadata.gz.sig +0 -0
  117. data/ChangeLog +0 -6595
  118. data/History.rdoc +0 -394
  119. data/README.ja.rdoc +0 -14
  120. data/README.rdoc +0 -168
  121. data/lib/pg/basic_type_mapping.rb +0 -426
  122. data/lib/pg/constants.rb +0 -11
  123. data/lib/pg/deprecated_constants.rb +0 -21
  124. data/lib/pg/text_decoder.rb +0 -51
  125. data/lib/pg/text_encoder.rb +0 -35
  126. data/spec/data/expected_trace.out +0 -26
  127. data/spec/data/random_binary_data +0 -0
  128. data/spec/helpers.rb +0 -352
  129. data/spec/pg/basic_type_mapping_spec.rb +0 -305
  130. data/spec/pg/connection_spec.rb +0 -1676
  131. data/spec/pg/result_spec.rb +0 -456
  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 -777
  139. data/spec/pg_spec.rb +0 -50
data/ext/pg.h CHANGED
@@ -11,6 +11,7 @@
11
11
  #include <sys/types.h>
12
12
  #if !defined(_WIN32)
13
13
  # include <sys/time.h>
14
+ # include <sys/socket.h>
14
15
  #endif
15
16
  #if defined(HAVE_UNISTD_H) && !defined(_WIN32)
16
17
  # include <unistd.h>
@@ -18,89 +19,18 @@
18
19
 
19
20
  /* Ruby headers */
20
21
  #include "ruby.h"
21
- #ifdef HAVE_RUBY_ST_H
22
- # include "ruby/st.h"
23
- #elif HAVE_ST_H
24
- # include "st.h"
25
- #endif
22
+ #include "ruby/st.h"
23
+ #include "ruby/encoding.h"
26
24
 
27
- #if defined(HAVE_RUBY_ENCODING_H) && HAVE_RUBY_ENCODING_H
28
- # include "ruby/encoding.h"
29
- # define M17N_SUPPORTED
30
- # ifdef HAVE_RB_ENCDB_ALIAS
31
- extern int rb_encdb_alias(const char *, const char *);
32
- # define ENC_ALIAS(name, orig) rb_encdb_alias((name), (orig))
33
- # elif HAVE_RB_ENC_ALIAS
34
- extern int rb_enc_alias(const char *, const char *);
35
- # define ENC_ALIAS(name, orig) rb_enc_alias((name), (orig))
36
- # else
37
- extern int rb_enc_alias(const char *alias, const char *orig); /* declaration missing in Ruby 1.9.1 */
38
- # define ENC_ALIAS(name, orig) rb_enc_alias((name), (orig))
39
- # endif
40
-
41
-
42
- # if !defined(ENCODING_SET_INLINED)
43
- /* Rubinius doesn't define ENCODING_SET_INLINED, so we fall back to the more
44
- * portable version.
45
- */
46
- # define PG_ENCODING_SET_NOCHECK(obj,i) \
47
- do { \
48
- rb_enc_set_index((obj), (i)); \
49
- } while(0)
50
- # else
51
- # define PG_ENCODING_SET_NOCHECK(obj,i) \
25
+ #define PG_ENCODING_SET_NOCHECK(obj,i) \
52
26
  do { \
53
27
  if ((i) < ENCODING_INLINE_MAX) \
54
28
  ENCODING_SET_INLINED((obj), (i)); \
55
29
  else \
56
30
  rb_enc_set_index((obj), (i)); \
57
31
  } while(0)
58
- # endif
59
-
60
- #else
61
- # define PG_ENCODING_SET_NOCHECK(obj,i) /* nothing */
62
- #endif
63
-
64
- #if RUBY_VM != 1
65
- # define RUBY_18_COMPAT
66
- #endif
67
-
68
- #ifndef RARRAY_LEN
69
- # define RARRAY_LEN(x) RARRAY((x))->len
70
- #endif /* RARRAY_LEN */
71
-
72
- #ifndef RSTRING_LEN
73
- # define RSTRING_LEN(x) RSTRING((x))->len
74
- #endif /* RSTRING_LEN */
75
-
76
- #ifndef RSTRING_PTR
77
- # define RSTRING_PTR(x) RSTRING((x))->ptr
78
- #endif /* RSTRING_PTR */
79
32
 
80
- #ifndef StringValuePtr
81
- # define StringValuePtr(x) STR2CSTR(x)
82
- #endif /* StringValuePtr */
83
-
84
- #ifdef RUBY_18_COMPAT
85
- # define rb_io_stdio_file GetWriteFile
86
- # include "rubyio.h"
87
- #else
88
- # include "ruby/io.h"
89
- #endif
90
-
91
- #ifdef RUBINIUS
92
- /* Workaround for wrong FIXNUM_MAX definition */
93
- typedef intptr_t native_int;
94
- #endif
95
-
96
- #ifndef RETURN_SIZED_ENUMERATOR
97
- #define RETURN_SIZED_ENUMERATOR(obj, argc, argv, size_fn) RETURN_ENUMERATOR((obj), (argc), (argv))
98
- #endif
99
-
100
- #ifndef HAVE_RB_HASH_DUP
101
- /* Rubinius doesn't define rb_hash_dup() */
102
- #define rb_hash_dup(tuple) rb_funcall((tuple), rb_intern("dup"), 0)
103
- #endif
33
+ #include "ruby/io.h"
104
34
 
105
35
  #ifndef timeradd
106
36
  #define timeradd(a, b, result) \
@@ -127,6 +57,7 @@
127
57
  #endif
128
58
 
129
59
  /* PostgreSQL headers */
60
+ #include "pg_config.h"
130
61
  #include "libpq-fe.h"
131
62
  #include "libpq/libpq-fs.h" /* large-object interface */
132
63
  #include "pg_config_manual.h"
@@ -145,12 +76,32 @@ typedef long suseconds_t;
145
76
  #define PG_MAX_COLUMNS 4000
146
77
  #endif
147
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
+
148
96
  /* The data behind each PG::Connection object */
149
97
  typedef struct {
150
98
  PGconn *pgconn;
151
99
 
152
100
  /* Cached IO object for the socket descriptor */
153
101
  VALUE socket_io;
102
+ /* function pointers of the original libpq notice receivers */
103
+ PQnoticeReceiver default_notice_receiver;
104
+ PQnoticeProcessor default_notice_processor;
154
105
  /* Proc object that receives notices as PG::Result objects */
155
106
  VALUE notice_receiver;
156
107
  /* Proc object that receives notices as String objects */
@@ -161,13 +112,21 @@ typedef struct {
161
112
  VALUE type_map_for_results;
162
113
  /* IO object internally used for the trace stream */
163
114
  VALUE trace_stream;
164
- /* Cached Encoding object */
165
- VALUE external_encoding;
166
115
  /* Kind of PG::Coder object for casting ruby values to COPY rows */
167
116
  VALUE encoder_for_put_copy_data;
168
117
  /* Kind of PG::Coder object for casting COPY rows to ruby values */
169
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;
170
125
 
126
+ #if defined(_WIN32)
127
+ /* File descriptor to be used for rb_w32_unwrap_io_handle() */
128
+ int ruby_sd;
129
+ #endif
171
130
  } t_pg_connection;
172
131
 
173
132
  typedef struct pg_coder t_pg_coder;
@@ -188,20 +147,32 @@ typedef struct {
188
147
  */
189
148
  t_typemap *p_typemap;
190
149
 
150
+ /* Ruby encoding index of the client/internal encoding */
151
+ int enc_idx : PG_ENC_IDX_BITS;
152
+
191
153
  /* 0 = PGresult is cleared by PG::Result#clear or by the GC
192
154
  * 1 = PGresult is cleared internally by libpq
193
155
  */
194
- int autoclear;
156
+ unsigned int autoclear : 1;
157
+
158
+ /* flags controlling Symbol/String field names */
159
+ unsigned int flags : 2;
195
160
 
196
161
  /* Number of fields in fnames[] .
197
162
  * Set to -1 if fnames[] is not yet initialized.
198
163
  */
199
164
  int nfields;
200
165
 
166
+ /* Size of PGresult as published to ruby memory management. */
167
+ ssize_t result_size;
168
+
201
169
  /* Prefilled tuple Hash with fnames[] as keys. */
202
170
  VALUE tuple_hash;
203
171
 
204
- /* 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.
205
176
  * Only valid if nfields != -1
206
177
  */
207
178
  VALUE fnames[0];
@@ -209,7 +180,7 @@ typedef struct {
209
180
 
210
181
 
211
182
  typedef int (* t_pg_coder_enc_func)(t_pg_coder *, VALUE, char *, VALUE *, int);
212
- typedef VALUE (* t_pg_coder_dec_func)(t_pg_coder *, char *, int, int, int, int);
183
+ typedef VALUE (* t_pg_coder_dec_func)(t_pg_coder *, const char *, int, int, int, int);
213
184
  typedef VALUE (* t_pg_fit_to_result)(VALUE, VALUE);
214
185
  typedef VALUE (* t_pg_fit_to_query)(VALUE, VALUE);
215
186
  typedef int (* t_pg_fit_to_copy_get)(VALUE);
@@ -217,12 +188,27 @@ typedef VALUE (* t_pg_typecast_result)(t_typemap *, VALUE, int, int);
217
188
  typedef t_pg_coder *(* t_pg_typecast_query_param)(t_typemap *, VALUE, int);
218
189
  typedef VALUE (* t_pg_typecast_copy_get)( t_typemap *, VALUE, int, int, int );
219
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
+
220
204
  struct pg_coder {
221
205
  t_pg_coder_enc_func enc_func;
222
206
  t_pg_coder_dec_func dec_func;
223
207
  VALUE coder_obj;
224
208
  Oid oid;
225
209
  int format;
210
+ /* OR-ed values out of PG_CODER_* */
211
+ int flags;
226
212
  };
227
213
 
228
214
  typedef struct {
@@ -252,6 +238,8 @@ typedef struct {
252
238
  } convs[0];
253
239
  } t_tmbc;
254
240
 
241
+ extern const rb_data_type_t pg_typemap_type;
242
+ extern const rb_data_type_t pg_coder_type;
255
243
 
256
244
  #include "gvl_wrappers.h"
257
245
 
@@ -259,6 +247,7 @@ typedef struct {
259
247
  * Globals
260
248
  **************************************************************************/
261
249
 
250
+ extern int pg_skip_deprecation_warning;
262
251
  extern VALUE rb_mPG;
263
252
  extern VALUE rb_ePGerror;
264
253
  extern VALUE rb_eServerError;
@@ -317,22 +306,25 @@ void init_pg_type_map_by_oid _(( void ));
317
306
  void init_pg_type_map_in_ruby _(( void ));
318
307
  void init_pg_coder _(( void ));
319
308
  void init_pg_copycoder _(( void ));
309
+ void init_pg_recordcoder _(( void ));
320
310
  void init_pg_text_encoder _(( void ));
321
311
  void init_pg_text_decoder _(( void ));
322
312
  void init_pg_binary_encoder _(( void ));
323
313
  void init_pg_binary_decoder _(( void ));
314
+ void init_pg_tuple _(( void ));
324
315
  VALUE lookup_error_class _(( const char * ));
325
- VALUE pg_bin_dec_bytea _(( t_pg_coder*, char *, int, int, int, int ));
326
- VALUE pg_text_dec_string _(( t_pg_coder*, char *, int, int, int, int ));
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 ));
327
318
  int pg_coder_enc_to_s _(( t_pg_coder*, VALUE, char *, VALUE *, int));
328
319
  int pg_text_enc_identifier _(( t_pg_coder*, VALUE, char *, VALUE *, int));
329
320
  t_pg_coder_enc_func pg_coder_enc_func _(( t_pg_coder* ));
330
321
  t_pg_coder_dec_func pg_coder_dec_func _(( t_pg_coder*, int ));
331
- void pg_define_coder _(( const char *, void *, VALUE, VALUE ));
322
+ VALUE pg_define_coder _(( const char *, void *, VALUE, VALUE ));
332
323
  VALUE pg_obj_to_i _(( VALUE ));
333
324
  VALUE pg_tmbc_allocate _(( void ));
334
325
  void pg_coder_init_encoder _(( VALUE ));
335
326
  void pg_coder_init_decoder _(( VALUE ));
327
+ void pg_coder_compact _(( void * ));
336
328
  char *pg_rb_str_ensure_capa _(( VALUE, long, char *, char ** ));
337
329
 
338
330
  #define PG_RB_STR_ENSURE_CAPA( str, expand_len, curr_ptr, end_ptr ) \
@@ -346,26 +338,26 @@ char *pg_rb_str_ensure_capa _(( VALUE, long, char *,
346
338
  (curr_ptr) = (end_ptr) = RSTRING_PTR(str) \
347
339
  )
348
340
 
349
- #define PG_RB_TAINTED_STR_NEW( str, curr_ptr, end_ptr ) ( \
350
- (str) = rb_tainted_str_new( NULL, 0 ), \
351
- (curr_ptr) = (end_ptr) = RSTRING_PTR(str) \
352
- )
353
-
354
341
  VALUE pg_typemap_fit_to_result _(( VALUE, VALUE ));
355
342
  VALUE pg_typemap_fit_to_query _(( VALUE, VALUE ));
356
343
  int pg_typemap_fit_to_copy_get _(( VALUE ));
357
344
  VALUE pg_typemap_result_value _(( t_typemap *, VALUE, int, int ));
358
345
  t_pg_coder *pg_typemap_typecast_query_param _(( t_typemap *, VALUE, int ));
359
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 * ));
360
350
 
361
351
  PGconn *pg_get_pgconn _(( VALUE ));
362
352
  t_pg_connection *pg_get_connection _(( VALUE ));
353
+ VALUE pgconn_block _(( int, VALUE *, VALUE ));
363
354
 
364
355
  VALUE pg_new_result _(( PGresult *, VALUE ));
365
356
  VALUE pg_new_result_autoclear _(( PGresult *, VALUE ));
366
357
  PGresult* pgresult_get _(( VALUE ));
367
358
  VALUE pg_result_check _(( VALUE ));
368
359
  VALUE pg_result_clear _(( VALUE ));
360
+ VALUE pg_tuple_new _(( VALUE, int ));
369
361
 
370
362
  /*
371
363
  * Fetch the data pointer for the result object
@@ -373,23 +365,28 @@ VALUE pg_result_clear _(( VALUE ));
373
365
  static inline t_pg_result *
374
366
  pgresult_get_this( VALUE self )
375
367
  {
376
- t_pg_result *this = DATA_PTR(self);
377
-
378
- if( this == NULL )
379
- rb_raise(rb_ePGerror, "result has been cleared");
380
-
381
- return this;
368
+ return RTYPEDDATA_DATA(self);
382
369
  }
383
370
 
384
371
 
385
- #ifdef M17N_SUPPORTED
386
372
  rb_encoding * pg_get_pg_encoding_as_rb_encoding _(( int ));
387
373
  rb_encoding * pg_get_pg_encname_as_rb_encoding _(( const char * ));
388
374
  const char * pg_get_rb_encoding_as_pg_encoding _(( rb_encoding * ));
389
375
  rb_encoding *pg_conn_enc_get _(( PGconn * ));
390
- #endif /* M17N_SUPPORTED */
391
376
 
392
377
  void notice_receiver_proxy(void *arg, const PGresult *result);
393
378
  void notice_processor_proxy(void *arg, const char *message);
394
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
+
395
392
  #endif /* end __pg_h */
@@ -1,27 +1,30 @@
1
1
  /*
2
2
  * pg_column_map.c - PG::ColumnMap class extension
3
- * $Id: pg_binary_decoder.c,v fcf731d3dff7 2015/09/08 12:25:06 jfali $
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"
9
10
  #ifdef HAVE_INTTYPES_H
10
11
  #include <inttypes.h>
11
12
  #endif
12
13
 
13
14
  VALUE rb_mPG_BinaryDecoder;
15
+ static VALUE s_Date;
16
+ static ID s_id_new;
14
17
 
15
18
 
16
19
  /*
17
20
  * Document-class: PG::BinaryDecoder::Boolean < PG::SimpleDecoder
18
21
  *
19
- * This is a decoder class for conversion of PostgreSQL binary bool type
20
- * 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.
21
24
  *
22
25
  */
23
26
  static VALUE
24
- 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)
25
28
  {
26
29
  if (len < 1) {
27
30
  rb_raise( rb_eTypeError, "wrong data for binary boolean converter in tuple %d field %d", tuple, field);
@@ -32,12 +35,12 @@ pg_bin_dec_boolean(t_pg_coder *conv, char *val, int len, int tuple, int field, i
32
35
  /*
33
36
  * Document-class: PG::BinaryDecoder::Integer < PG::SimpleDecoder
34
37
  *
35
- * 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
36
39
  * to Ruby Integer objects.
37
40
  *
38
41
  */
39
42
  static VALUE
40
- 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)
41
44
  {
42
45
  switch( len ){
43
46
  case 2:
@@ -54,12 +57,12 @@ pg_bin_dec_integer(t_pg_coder *conv, char *val, int len, int tuple, int field, i
54
57
  /*
55
58
  * Document-class: PG::BinaryDecoder::Float < PG::SimpleDecoder
56
59
  *
57
- * 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
58
61
  * to Ruby Float objects.
59
62
  *
60
63
  */
61
64
  static VALUE
62
- 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)
63
66
  {
64
67
  union {
65
68
  float f;
@@ -86,15 +89,15 @@ pg_bin_dec_float(t_pg_coder *conv, char *val, int len, int tuple, int field, int
86
89
  * Document-class: PG::BinaryDecoder::Bytea < PG::SimpleDecoder
87
90
  *
88
91
  * This decoder class delivers the data received from the server as binary String object.
89
- * 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
90
93
  * data in binary format.
91
94
  *
92
95
  */
93
96
  VALUE
94
- 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)
95
98
  {
96
99
  VALUE ret;
97
- ret = rb_tainted_str_new( val, len );
100
+ ret = rb_str_new( val, len );
98
101
  PG_ENCODING_SET_NOCHECK( ret, rb_ascii8bit_encindex() );
99
102
  return ret;
100
103
  }
@@ -102,17 +105,17 @@ pg_bin_dec_bytea(t_pg_coder *conv, char *val, int len, int tuple, int field, int
102
105
  /*
103
106
  * Document-class: PG::BinaryDecoder::ToBase64 < PG::CompositeDecoder
104
107
  *
105
- * 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.
106
109
  *
107
110
  */
108
111
  static VALUE
109
- 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)
110
113
  {
111
114
  t_pg_composite_coder *this = (t_pg_composite_coder *)conv;
112
115
  t_pg_coder_dec_func dec_func = pg_coder_dec_func(this->elem, this->comp.format);
113
116
  int encoded_len = BASE64_ENCODED_SIZE(len);
114
117
  /* create a buffer of the encoded length */
115
- VALUE out_value = rb_tainted_str_new(NULL, encoded_len);
118
+ VALUE out_value = rb_str_new(NULL, encoded_len);
116
119
 
117
120
  base64_encode( RSTRING_PTR(out_value), val, len );
118
121
 
@@ -130,6 +133,146 @@ pg_bin_dec_to_base64(t_pg_coder *conv, char *val, int len, int tuple, int field,
130
133
  return out_value;
131
134
  }
132
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
+
133
276
  /*
134
277
  * Document-class: PG::BinaryDecoder::String < PG::SimpleDecoder
135
278
  *
@@ -140,10 +283,11 @@ pg_bin_dec_to_base64(t_pg_coder *conv, char *val, int len, int tuple, int field,
140
283
  */
141
284
 
142
285
  void
143
- init_pg_binary_decoder()
286
+ init_pg_binary_decoder(void)
144
287
  {
145
288
  /* This module encapsulates all decoder classes with binary input format */
146
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);
147
291
 
148
292
  /* Make RDoc aware of the decoder classes... */
149
293
  /* dummy = rb_define_class_under( rb_mPG_BinaryDecoder, "Boolean", rb_cPG_SimpleDecoder ); */
@@ -156,6 +300,8 @@ init_pg_binary_decoder()
156
300
  pg_define_coder( "String", pg_text_dec_string, rb_cPG_SimpleDecoder, rb_mPG_BinaryDecoder );
157
301
  /* dummy = rb_define_class_under( rb_mPG_BinaryDecoder, "Bytea", rb_cPG_SimpleDecoder ); */
158
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 );
159
305
 
160
306
  /* dummy = rb_define_class_under( rb_mPG_BinaryDecoder, "ToBase64", rb_cPG_CompositeDecoder ); */
161
307
  pg_define_coder( "ToBase64", pg_bin_dec_to_base64, rb_cPG_CompositeDecoder, rb_mPG_BinaryDecoder );