1
+ # frozen_string_literal: true
2
+
1
3
require "sqlite3/errors"
2
4
3
5
module SQLite3
@@ -58,11 +60,20 @@ def get_enum_pragma(name)
58
60
# have duplicate values. See #synchronous, #default_synchronous,
59
61
# #temp_store, and #default_temp_store for usage examples.
60
62
def set_enum_pragma ( name , mode , enums )
61
- match = enums . find { |p | p . find { |i | i . to_s . downcase == mode . to_s . downcase } }
63
+ match = if enums . is_a? ( Array )
64
+ # maybe deprecate this?
65
+ enums . find { |p | p . find { |i | i . to_s . downcase == mode . to_s . downcase } }
66
+ elsif mode . is_a? ( String )
67
+ enums . fetch ( mode . downcase )
68
+ else
69
+ mode
70
+ end
71
+
62
72
unless match
63
73
raise SQLite3 ::Exception , "unrecognized #{ name } #{ mode . inspect } "
64
74
end
65
- execute ( "PRAGMA #{ name } ='#{ match . first . upcase } '" )
75
+
76
+ execute ( "PRAGMA #{ name } ='#{ match } '" )
66
77
end
67
78
68
79
# Returns the value of the given pragma as an integer.
@@ -77,26 +88,57 @@ def set_int_pragma(name, value)
77
88
end
78
89
79
90
# The enumeration of valid synchronous modes.
80
- SYNCHRONOUS_MODES = [ [ "full" , 2 ] , [ "normal" , 1 ] , [ "off" , 0 ] ]
91
+ SYNCHRONOUS_MODES = {
92
+ "full" => 2 ,
93
+ "normal" => 1 ,
94
+ "off" => 0
95
+ } . freeze
81
96
82
97
# The enumeration of valid temp store modes.
83
- TEMP_STORE_MODES = [ [ "default" , 0 ] , [ "file" , 1 ] , [ "memory" , 2 ] ]
98
+ TEMP_STORE_MODES = {
99
+ "default" => 0 ,
100
+ "file" => 1 ,
101
+ "memory" => 2
102
+ } . freeze
84
103
85
104
# The enumeration of valid auto vacuum modes.
86
- AUTO_VACUUM_MODES = [ [ "none" , 0 ] , [ "full" , 1 ] , [ "incremental" , 2 ] ]
105
+ AUTO_VACUUM_MODES = {
106
+ "none" => 0 ,
107
+ "full" => 1 ,
108
+ "incremental" => 2
109
+ } . freeze
87
110
88
111
# The list of valid journaling modes.
89
- JOURNAL_MODES = [ [ "delete" ] , [ "truncate" ] , [ "persist" ] , [ "memory" ] ,
90
- [ "wal" ] , [ "off" ] ]
112
+ JOURNAL_MODES = {
113
+ "delete" => "delete" ,
114
+ "truncate" => "truncate" ,
115
+ "persist" => "persist" ,
116
+ "memory" => "memory" ,
117
+ "wal" => "wal" ,
118
+ "off" => "off"
119
+ } . freeze
91
120
92
121
# The list of valid locking modes.
93
- LOCKING_MODES = [ [ "normal" ] , [ "exclusive" ] ]
122
+ LOCKING_MODES = {
123
+ "normal" => "normal" ,
124
+ "exclusive" => "exclusive"
125
+ } . freeze
94
126
95
127
# The list of valid encodings.
96
- ENCODINGS = [ [ "utf-8" ] , [ "utf-16" ] , [ "utf-16le" ] , [ "utf-16be" ] ]
128
+ ENCODINGS = {
129
+ "utf-8" => "utf-8" ,
130
+ "utf-16" => "utf-16" ,
131
+ "utf-16le" => "utf-16le" ,
132
+ "utf-16be" => "utf-16be"
133
+ } . freeze
97
134
98
135
# The list of valid WAL checkpoints.
99
- WAL_CHECKPOINTS = [ [ "passive" ] , [ "full" ] , [ "restart" ] , [ "truncate" ] ]
136
+ WAL_CHECKPOINTS = {
137
+ "passive" => "passive" ,
138
+ "full" => "full" ,
139
+ "restart" => "restart" ,
140
+ "truncate" => "truncate"
141
+ } . freeze
100
142
101
143
def application_id
102
144
get_int_pragma "application_id"
@@ -227,7 +269,7 @@ def encoding
227
269
end
228
270
229
271
def encoding = ( mode )
230
- set_enum_pragma "encoding" , mode , ENCODINGS
272
+ set_string_pragma "encoding" , mode , ENCODINGS
231
273
end
232
274
233
275
def foreign_key_check ( *table , &block ) # :yields: row
@@ -295,7 +337,7 @@ def journal_mode
295
337
end
296
338
297
339
def journal_mode = ( mode )
298
- set_enum_pragma "journal_mode" , mode , JOURNAL_MODES
340
+ set_string_pragma "journal_mode" , mode , JOURNAL_MODES
299
341
end
300
342
301
343
def journal_size_limit
@@ -319,7 +361,7 @@ def locking_mode
319
361
end
320
362
321
363
def locking_mode = ( mode )
322
- set_enum_pragma "locking_mode" , mode , LOCKING_MODES
364
+ set_string_pragma "locking_mode" , mode , LOCKING_MODES
323
365
end
324
366
325
367
def max_page_count
@@ -525,7 +567,7 @@ def wal_checkpoint
525
567
end
526
568
527
569
def wal_checkpoint = ( mode )
528
- set_enum_pragma "wal_checkpoint" , mode , WAL_CHECKPOINTS
570
+ set_string_pragma "wal_checkpoint" , mode , WAL_CHECKPOINTS
529
571
end
530
572
531
573
def writable_schema = ( mode )
@@ -568,6 +610,13 @@ def table_info table
568
610
569
611
private
570
612
613
+ def set_string_pragma ( pragma_name , value , valid_values )
614
+ valid_values . fetch ( value . to_s . downcase ) {
615
+ raise SQLite3 ::Exception , "unrecognized #{ pragma_name } #{ value . inspect } "
616
+ }
617
+ set_enum_pragma ( pragma_name , value , valid_values )
618
+ end
619
+
571
620
# Compares two version strings
572
621
def version_compare ( v1 , v2 )
573
622
v1 = v1 . split ( "." ) . map { |i | i . to_i }
0 commit comments