pg 0.18.1 → 1.5.6

Sign up to get free protection for your applications and to get access to all the features.
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 );