pg 0.18.1 → 1.5.6

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 (137) 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 +141 -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 +17 -0
  16. data/History.md +901 -0
  17. data/Manifest.txt +8 -21
  18. data/README-Windows.rdoc +17 -28
  19. data/README.ja.md +300 -0
  20. data/README.md +286 -0
  21. data/Rakefile +43 -131
  22. data/Rakefile.cross +89 -70
  23. data/certs/ged.pem +24 -0
  24. data/certs/kanis@comcard.de.pem +20 -0
  25. data/certs/larskanis-2022.pem +26 -0
  26. data/certs/larskanis-2023.pem +24 -0
  27. data/certs/larskanis-2024.pem +24 -0
  28. data/ext/errorcodes.def +113 -0
  29. data/ext/errorcodes.rb +1 -1
  30. data/ext/errorcodes.txt +36 -2
  31. data/ext/extconf.rb +128 -55
  32. data/ext/gvl_wrappers.c +8 -0
  33. data/ext/gvl_wrappers.h +44 -33
  34. data/ext/pg.c +228 -202
  35. data/ext/pg.h +108 -99
  36. data/ext/pg_binary_decoder.c +164 -16
  37. data/ext/pg_binary_encoder.c +249 -22
  38. data/ext/pg_coder.c +189 -44
  39. data/ext/pg_connection.c +1889 -1195
  40. data/ext/pg_copy_coder.c +398 -42
  41. data/ext/pg_errors.c +1 -1
  42. data/ext/pg_record_coder.c +522 -0
  43. data/ext/pg_result.c +729 -234
  44. data/ext/pg_text_decoder.c +635 -52
  45. data/ext/pg_text_encoder.c +294 -130
  46. data/ext/pg_tuple.c +572 -0
  47. data/ext/pg_type_map.c +64 -23
  48. data/ext/pg_type_map_all_strings.c +21 -7
  49. data/ext/pg_type_map_by_class.c +59 -27
  50. data/ext/pg_type_map_by_column.c +86 -43
  51. data/ext/pg_type_map_by_mri_type.c +50 -21
  52. data/ext/pg_type_map_by_oid.c +62 -29
  53. data/ext/pg_type_map_in_ruby.c +59 -28
  54. data/ext/{util.c → pg_util.c} +13 -13
  55. data/ext/{util.h → pg_util.h} +3 -3
  56. data/lib/pg/basic_type_map_based_on_result.rb +67 -0
  57. data/lib/pg/basic_type_map_for_queries.rb +202 -0
  58. data/lib/pg/basic_type_map_for_results.rb +104 -0
  59. data/lib/pg/basic_type_registry.rb +303 -0
  60. data/lib/pg/binary_decoder/date.rb +9 -0
  61. data/lib/pg/binary_decoder/timestamp.rb +26 -0
  62. data/lib/pg/binary_encoder/timestamp.rb +20 -0
  63. data/lib/pg/coder.rb +36 -13
  64. data/lib/pg/connection.rb +813 -74
  65. data/lib/pg/exceptions.rb +16 -2
  66. data/lib/pg/result.rb +24 -7
  67. data/lib/pg/text_decoder/date.rb +18 -0
  68. data/lib/pg/text_decoder/inet.rb +9 -0
  69. data/lib/pg/text_decoder/json.rb +14 -0
  70. data/lib/pg/text_decoder/numeric.rb +9 -0
  71. data/lib/pg/text_decoder/timestamp.rb +30 -0
  72. data/lib/pg/text_encoder/date.rb +12 -0
  73. data/lib/pg/text_encoder/inet.rb +28 -0
  74. data/lib/pg/text_encoder/json.rb +14 -0
  75. data/lib/pg/text_encoder/numeric.rb +9 -0
  76. data/lib/pg/text_encoder/timestamp.rb +24 -0
  77. data/lib/pg/tuple.rb +30 -0
  78. data/lib/pg/type_map_by_column.rb +3 -2
  79. data/lib/pg/version.rb +4 -0
  80. data/lib/pg.rb +106 -41
  81. data/misc/openssl-pg-segfault.rb +31 -0
  82. data/misc/postgres/History.txt +9 -0
  83. data/misc/postgres/Manifest.txt +5 -0
  84. data/misc/postgres/README.txt +21 -0
  85. data/misc/postgres/Rakefile +21 -0
  86. data/misc/postgres/lib/postgres.rb +16 -0
  87. data/misc/ruby-pg/History.txt +9 -0
  88. data/misc/ruby-pg/Manifest.txt +5 -0
  89. data/misc/ruby-pg/README.txt +21 -0
  90. data/misc/ruby-pg/Rakefile +21 -0
  91. data/misc/ruby-pg/lib/ruby/pg.rb +16 -0
  92. data/pg.gemspec +34 -0
  93. data/rakelib/task_extension.rb +46 -0
  94. data/sample/array_insert.rb +1 -1
  95. data/sample/async_api.rb +4 -8
  96. data/sample/async_copyto.rb +1 -1
  97. data/sample/async_mixed.rb +1 -1
  98. data/sample/check_conn.rb +1 -1
  99. data/sample/copydata.rb +71 -0
  100. data/sample/copyfrom.rb +1 -1
  101. data/sample/copyto.rb +1 -1
  102. data/sample/cursor.rb +1 -1
  103. data/sample/disk_usage_report.rb +6 -15
  104. data/sample/issue-119.rb +2 -2
  105. data/sample/losample.rb +1 -1
  106. data/sample/minimal-testcase.rb +2 -2
  107. data/sample/notify_wait.rb +1 -1
  108. data/sample/pg_statistics.rb +6 -15
  109. data/sample/replication_monitor.rb +9 -18
  110. data/sample/test_binary_values.rb +1 -1
  111. data/sample/wal_shipper.rb +2 -2
  112. data/sample/warehouse_partitions.rb +8 -17
  113. data.tar.gz.sig +0 -0
  114. metadata +135 -207
  115. metadata.gz.sig +0 -0
  116. data/ChangeLog +0 -5378
  117. data/History.rdoc +0 -297
  118. data/README.ja.rdoc +0 -14
  119. data/README.rdoc +0 -161
  120. data/lib/pg/basic_type_mapping.rb +0 -399
  121. data/lib/pg/constants.rb +0 -11
  122. data/lib/pg/text_decoder.rb +0 -42
  123. data/lib/pg/text_encoder.rb +0 -27
  124. data/spec/data/expected_trace.out +0 -26
  125. data/spec/data/random_binary_data +0 -0
  126. data/spec/helpers.rb +0 -355
  127. data/spec/pg/basic_type_mapping_spec.rb +0 -251
  128. data/spec/pg/connection_spec.rb +0 -1459
  129. data/spec/pg/result_spec.rb +0 -449
  130. data/spec/pg/type_map_by_class_spec.rb +0 -138
  131. data/spec/pg/type_map_by_column_spec.rb +0 -222
  132. data/spec/pg/type_map_by_mri_type_spec.rb +0 -136
  133. data/spec/pg/type_map_by_oid_spec.rb +0 -149
  134. data/spec/pg/type_map_in_ruby_spec.rb +0 -164
  135. data/spec/pg/type_map_spec.rb +0 -22
  136. data/spec/pg/type_spec.rb +0 -665
  137. 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
-
73
- #ifndef RSTRING_PTR
74
- # define RSTRING_PTR(x) RSTRING((x))->ptr
75
- #endif /* RSTRING_PTR */
76
32
 
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"
@@ -133,12 +67,41 @@
133
67
  typedef long suseconds_t;
134
68
  #endif
135
69
 
70
+ #if defined(HAVE_VARIABLE_LENGTH_ARRAYS)
71
+ #define PG_VARIABLE_LENGTH_ARRAY(type, name, len, maxlen) type name[(len)];
72
+ #else
73
+ #define PG_VARIABLE_LENGTH_ARRAY(type, name, len, maxlen) \
74
+ type name[(maxlen)] = {(len)>(maxlen) ? (rb_raise(rb_eArgError, "Number of " #name " (%d) exceeds allowed maximum of " #maxlen, (len) ), (type)1) : (type)0};
75
+
76
+ #define PG_MAX_COLUMNS 4000
77
+ #endif
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
+
136
96
  /* The data behind each PG::Connection object */
137
97
  typedef struct {
138
98
  PGconn *pgconn;
139
99
 
140
100
  /* Cached IO object for the socket descriptor */
141
101
  VALUE socket_io;
102
+ /* function pointers of the original libpq notice receivers */
103
+ PQnoticeReceiver default_notice_receiver;
104
+ PQnoticeProcessor default_notice_processor;
142
105
  /* Proc object that receives notices as PG::Result objects */
143
106
  VALUE notice_receiver;
144
107
  /* Proc object that receives notices as String objects */
@@ -149,13 +112,21 @@ typedef struct {
149
112
  VALUE type_map_for_results;
150
113
  /* IO object internally used for the trace stream */
151
114
  VALUE trace_stream;
152
- /* Cached Encoding object */
153
- VALUE external_encoding;
154
115
  /* Kind of PG::Coder object for casting ruby values to COPY rows */
155
116
  VALUE encoder_for_put_copy_data;
156
117
  /* Kind of PG::Coder object for casting COPY rows to ruby values */
157
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;
158
125
 
126
+ #if defined(_WIN32)
127
+ /* File descriptor to be used for rb_w32_unwrap_io_handle() */
128
+ int ruby_sd;
129
+ #endif
159
130
  } t_pg_connection;
160
131
 
161
132
  typedef struct pg_coder t_pg_coder;
@@ -176,28 +147,40 @@ typedef struct {
176
147
  */
177
148
  t_typemap *p_typemap;
178
149
 
150
+ /* Ruby encoding index of the client/internal encoding */
151
+ int enc_idx : PG_ENC_IDX_BITS;
152
+
179
153
  /* 0 = PGresult is cleared by PG::Result#clear or by the GC
180
154
  * 1 = PGresult is cleared internally by libpq
181
155
  */
182
- int autoclear;
156
+ unsigned int autoclear : 1;
157
+
158
+ /* flags controlling Symbol/String field names */
159
+ unsigned int flags : 2;
183
160
 
184
161
  /* Number of fields in fnames[] .
185
162
  * Set to -1 if fnames[] is not yet initialized.
186
163
  */
187
164
  int nfields;
188
165
 
166
+ /* Size of PGresult as published to ruby memory management. */
167
+ ssize_t result_size;
168
+
189
169
  /* Prefilled tuple Hash with fnames[] as keys. */
190
170
  VALUE tuple_hash;
191
171
 
192
- /* 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.
193
176
  * Only valid if nfields != -1
194
177
  */
195
178
  VALUE fnames[0];
196
179
  } t_pg_result;
197
180
 
198
181
 
199
- typedef int (* t_pg_coder_enc_func)(t_pg_coder *, VALUE, char *, VALUE *);
200
- 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);
201
184
  typedef VALUE (* t_pg_fit_to_result)(VALUE, VALUE);
202
185
  typedef VALUE (* t_pg_fit_to_query)(VALUE, VALUE);
203
186
  typedef int (* t_pg_fit_to_copy_get)(VALUE);
@@ -205,12 +188,27 @@ typedef VALUE (* t_pg_typecast_result)(t_typemap *, VALUE, int, int);
205
188
  typedef t_pg_coder *(* t_pg_typecast_query_param)(t_typemap *, VALUE, int);
206
189
  typedef VALUE (* t_pg_typecast_copy_get)( t_typemap *, VALUE, int, int, int );
207
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
+
208
204
  struct pg_coder {
209
205
  t_pg_coder_enc_func enc_func;
210
206
  t_pg_coder_dec_func dec_func;
211
207
  VALUE coder_obj;
212
208
  Oid oid;
213
209
  int format;
210
+ /* OR-ed values out of PG_CODER_* */
211
+ int flags;
214
212
  };
215
213
 
216
214
  typedef struct {
@@ -240,6 +238,8 @@ typedef struct {
240
238
  } convs[0];
241
239
  } t_tmbc;
242
240
 
241
+ extern const rb_data_type_t pg_typemap_type;
242
+ extern const rb_data_type_t pg_coder_type;
243
243
 
244
244
  #include "gvl_wrappers.h"
245
245
 
@@ -247,6 +247,7 @@ typedef struct {
247
247
  * Globals
248
248
  **************************************************************************/
249
249
 
250
+ extern int pg_skip_deprecation_warning;
250
251
  extern VALUE rb_mPG;
251
252
  extern VALUE rb_ePGerror;
252
253
  extern VALUE rb_eServerError;
@@ -305,21 +306,25 @@ void init_pg_type_map_by_oid _(( void ));
305
306
  void init_pg_type_map_in_ruby _(( void ));
306
307
  void init_pg_coder _(( void ));
307
308
  void init_pg_copycoder _(( void ));
309
+ void init_pg_recordcoder _(( void ));
308
310
  void init_pg_text_encoder _(( void ));
309
311
  void init_pg_text_decoder _(( void ));
310
312
  void init_pg_binary_encoder _(( void ));
311
313
  void init_pg_binary_decoder _(( void ));
314
+ void init_pg_tuple _(( void ));
312
315
  VALUE lookup_error_class _(( const char * ));
313
- VALUE pg_bin_dec_bytea _(( t_pg_coder*, char *, int, int, int, int ));
314
- VALUE pg_text_dec_string _(( t_pg_coder*, char *, int, int, int, int ));
315
- int pg_coder_enc_to_s _(( 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));
316
320
  t_pg_coder_enc_func pg_coder_enc_func _(( t_pg_coder* ));
317
321
  t_pg_coder_dec_func pg_coder_dec_func _(( t_pg_coder*, int ));
318
- void pg_define_coder _(( const char *, void *, VALUE, VALUE ));
322
+ VALUE pg_define_coder _(( const char *, void *, VALUE, VALUE ));
319
323
  VALUE pg_obj_to_i _(( VALUE ));
320
324
  VALUE pg_tmbc_allocate _(( void ));
321
325
  void pg_coder_init_encoder _(( VALUE ));
322
326
  void pg_coder_init_decoder _(( VALUE ));
327
+ void pg_coder_compact _(( void * ));
323
328
  char *pg_rb_str_ensure_capa _(( VALUE, long, char *, char ** ));
324
329
 
325
330
  #define PG_RB_STR_ENSURE_CAPA( str, expand_len, curr_ptr, end_ptr ) \
@@ -333,26 +338,26 @@ char *pg_rb_str_ensure_capa _(( VALUE, long, char *,
333
338
  (curr_ptr) = (end_ptr) = RSTRING_PTR(str) \
334
339
  )
335
340
 
336
- #define PG_RB_TAINTED_STR_NEW( str, curr_ptr, end_ptr ) ( \
337
- (str) = rb_tainted_str_new( NULL, 0 ), \
338
- (curr_ptr) = (end_ptr) = RSTRING_PTR(str) \
339
- )
340
-
341
341
  VALUE pg_typemap_fit_to_result _(( VALUE, VALUE ));
342
342
  VALUE pg_typemap_fit_to_query _(( VALUE, VALUE ));
343
343
  int pg_typemap_fit_to_copy_get _(( VALUE ));
344
344
  VALUE pg_typemap_result_value _(( t_typemap *, VALUE, int, int ));
345
345
  t_pg_coder *pg_typemap_typecast_query_param _(( t_typemap *, VALUE, int ));
346
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 * ));
347
350
 
348
351
  PGconn *pg_get_pgconn _(( VALUE ));
349
352
  t_pg_connection *pg_get_connection _(( VALUE ));
353
+ VALUE pgconn_block _(( int, VALUE *, VALUE ));
350
354
 
351
355
  VALUE pg_new_result _(( PGresult *, VALUE ));
352
356
  VALUE pg_new_result_autoclear _(( PGresult *, VALUE ));
353
357
  PGresult* pgresult_get _(( VALUE ));
354
358
  VALUE pg_result_check _(( VALUE ));
355
359
  VALUE pg_result_clear _(( VALUE ));
360
+ VALUE pg_tuple_new _(( VALUE, int ));
356
361
 
357
362
  /*
358
363
  * Fetch the data pointer for the result object
@@ -360,23 +365,27 @@ VALUE pg_result_clear _(( VALUE ));
360
365
  static inline t_pg_result *
361
366
  pgresult_get_this( VALUE self )
362
367
  {
363
- t_pg_result *this = DATA_PTR(self);
364
-
365
- if( this == NULL )
366
- rb_raise(rb_ePGerror, "result has been cleared");
367
-
368
- return this;
368
+ return RTYPEDDATA_DATA(self);
369
369
  }
370
370
 
371
371
 
372
- #ifdef M17N_SUPPORTED
373
- rb_encoding * pg_get_pg_encoding_as_rb_encoding _(( int ));
374
372
  rb_encoding * pg_get_pg_encname_as_rb_encoding _(( const char * ));
375
373
  const char * pg_get_rb_encoding_as_pg_encoding _(( rb_encoding * ));
376
374
  rb_encoding *pg_conn_enc_get _(( PGconn * ));
377
- #endif /* M17N_SUPPORTED */
378
375
 
379
376
  void notice_receiver_proxy(void *arg, const PGresult *result);
380
377
  void notice_processor_proxy(void *arg, const char *message);
381
378
 
379
+ /* reports if `-W' specified and PG_SKIP_DEPRECATION_WARNING environment variable isn't set
380
+ *
381
+ * message_id identifies the warning, so that it's reported only once.
382
+ */
383
+ #define pg_deprecated(message_id, format_args) \
384
+ do { \
385
+ if( !(pg_skip_deprecation_warning & (1 << message_id)) ){ \
386
+ pg_skip_deprecation_warning |= 1 << message_id; \
387
+ rb_warning format_args; \
388
+ } \
389
+ } while(0);
390
+
382
391
  #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 );