activerecord-postgis-adapter 0.6.6 → 0.7.0
Sign up to get free protection for your applications and to get access to all the features.
- checksums.yaml +4 -4
- data/History.rdoc +7 -0
- data/{lib/rgeo/active_record/postgis_adapter/railtie.rb → LICENSE.txt} +2 -12
- data/lib/active_record/connection_adapters/postgis_adapter.rb +1 -60
- data/lib/active_record/connection_adapters/postgis_adapter/rails4/create_connection.rb +15 -74
- data/lib/active_record/connection_adapters/postgis_adapter/rails4/databases.rake +0 -38
- data/lib/active_record/connection_adapters/postgis_adapter/rails4/main_adapter.rb +0 -68
- data/lib/active_record/connection_adapters/postgis_adapter/rails4/postgis_database_tasks.rb +0 -36
- data/lib/active_record/connection_adapters/postgis_adapter/rails4/spatial_column.rb +0 -36
- data/lib/active_record/connection_adapters/postgis_adapter/rails4/spatial_table_definition.rb +0 -36
- data/lib/active_record/connection_adapters/postgis_adapter/railtie.rb +0 -36
- data/lib/active_record/connection_adapters/postgis_adapter/shared/arel_tosql.rb +0 -36
- data/lib/active_record/connection_adapters/postgis_adapter/shared/common_adapter_methods.rb +3 -70
- data/lib/active_record/connection_adapters/postgis_adapter/shared/railtie.rb +1 -54
- data/lib/active_record/connection_adapters/postgis_adapter/shared/setup.rb +0 -36
- data/lib/active_record/connection_adapters/postgis_adapter/version.rb +7 -0
- data/lib/activerecord-postgis-adapter.rb +0 -36
- data/lib/activerecord/postgis/adapter.rb +0 -36
- data/test/database.yml +5 -0
- data/test/tc_basic.rb +0 -35
- data/test/tc_ddl.rb +0 -35
- data/test/tc_nested_class.rb +0 -35
- data/test/tc_spatial_queries.rb +0 -35
- data/test/tc_tasks.rb +0 -78
- metadata +33 -38
- data/Version +0 -1
- data/lib/active_record/connection_adapters/postgis_adapter/rails3/create_connection.rb +0 -96
- data/lib/active_record/connection_adapters/postgis_adapter/rails3/databases.rake +0 -232
- data/lib/active_record/connection_adapters/postgis_adapter/rails3/main_adapter.rb +0 -298
- data/lib/active_record/connection_adapters/postgis_adapter/rails3/spatial_column.rb +0 -195
- data/lib/active_record/connection_adapters/postgis_adapter/rails3/spatial_table_definition.rb +0 -145
- data/lib/active_record/connection_adapters/postgis_adapter/shared/jdbc_compat.rb +0 -134
- data/lib/active_record/connection_adapters/postgis_adapter/shared/version.rb +0 -62
data/Version
DELETED
@@ -1 +0,0 @@
|
|
1
|
-
0.6.6
|
@@ -1,96 +0,0 @@
|
|
1
|
-
# -----------------------------------------------------------------------------
|
2
|
-
#
|
3
|
-
# PostGIS adapter for ActiveRecord
|
4
|
-
#
|
5
|
-
# -----------------------------------------------------------------------------
|
6
|
-
# Copyright 2010-2012 Daniel Azuma
|
7
|
-
#
|
8
|
-
# All rights reserved.
|
9
|
-
#
|
10
|
-
# Redistribution and use in source and binary forms, with or without
|
11
|
-
# modification, are permitted provided that the following conditions are met:
|
12
|
-
#
|
13
|
-
# * Redistributions of source code must retain the above copyright notice,
|
14
|
-
# this list of conditions and the following disclaimer.
|
15
|
-
# * Redistributions in binary form must reproduce the above copyright notice,
|
16
|
-
# this list of conditions and the following disclaimer in the documentation
|
17
|
-
# and/or other materials provided with the distribution.
|
18
|
-
# * Neither the name of the copyright holder, nor the names of any other
|
19
|
-
# contributors to this software, may be used to endorse or promote products
|
20
|
-
# derived from this software without specific prior written permission.
|
21
|
-
#
|
22
|
-
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
|
23
|
-
# AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
24
|
-
# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
25
|
-
# ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
|
26
|
-
# LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
|
27
|
-
# CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
|
28
|
-
# SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
|
29
|
-
# INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
|
30
|
-
# CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
|
31
|
-
# ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
|
32
|
-
# POSSIBILITY OF SUCH DAMAGE.
|
33
|
-
# -----------------------------------------------------------------------------
|
34
|
-
;
|
35
|
-
|
36
|
-
|
37
|
-
module ActiveRecord # :nodoc:
|
38
|
-
|
39
|
-
class Base # :nodoc:
|
40
|
-
|
41
|
-
class << self
|
42
|
-
|
43
|
-
|
44
|
-
if defined?(::RUBY_ENGINE) && ::RUBY_ENGINE == 'jruby'
|
45
|
-
|
46
|
-
|
47
|
-
require 'active_record/connection_adapters/jdbcpostgresql_adapter'
|
48
|
-
require 'active_record/connection_adapters/postgis_adapter/shared/jdbc_compat'
|
49
|
-
|
50
|
-
|
51
|
-
def postgis_connection(config_)
|
52
|
-
::ActiveRecord::ConnectionAdapters::PostGISAdapter.create_jdbc_connection(self, config_)
|
53
|
-
end
|
54
|
-
|
55
|
-
alias_method :jdbcpostgis_connection, :postgis_connection
|
56
|
-
|
57
|
-
|
58
|
-
else
|
59
|
-
|
60
|
-
|
61
|
-
require 'pg'
|
62
|
-
|
63
|
-
|
64
|
-
# ActiveRecord looks for the postgis_connection factory method in
|
65
|
-
# this class.
|
66
|
-
#
|
67
|
-
# Based on the default <tt>postgresql_connection</tt> definition from
|
68
|
-
# ActiveRecord.
|
69
|
-
|
70
|
-
def postgis_connection(config_)
|
71
|
-
config_ = config_.symbolize_keys
|
72
|
-
host_ = config_[:host]
|
73
|
-
port_ = config_[:port] || 5432
|
74
|
-
username_ = config_[:username].to_s if config_[:username]
|
75
|
-
password_ = config_[:password].to_s if config_[:password]
|
76
|
-
|
77
|
-
if config_.key?(:database)
|
78
|
-
database_ = config_[:database]
|
79
|
-
else
|
80
|
-
raise ::ArgumentError, "No database specified. Missing argument: database."
|
81
|
-
end
|
82
|
-
|
83
|
-
# The postgres drivers don't allow the creation of an unconnected PGconn object,
|
84
|
-
# so just pass a nil connection object for the time being.
|
85
|
-
::ActiveRecord::ConnectionAdapters::PostGISAdapter::MainAdapter.new(nil, logger, [host_, port_, nil, nil, database_, username_, password_], config_)
|
86
|
-
end
|
87
|
-
|
88
|
-
|
89
|
-
end
|
90
|
-
|
91
|
-
|
92
|
-
end
|
93
|
-
|
94
|
-
end
|
95
|
-
|
96
|
-
end
|
@@ -1,232 +0,0 @@
|
|
1
|
-
# -----------------------------------------------------------------------------
|
2
|
-
#
|
3
|
-
# PostGIS adapter for ActiveRecord
|
4
|
-
#
|
5
|
-
# -----------------------------------------------------------------------------
|
6
|
-
# Copyright 2010-2012 Daniel Azuma
|
7
|
-
#
|
8
|
-
# All rights reserved.
|
9
|
-
#
|
10
|
-
# Redistribution and use in source and binary forms, with or without
|
11
|
-
# modification, are permitted provided that the following conditions are met:
|
12
|
-
#
|
13
|
-
# * Redistributions of source code must retain the above copyright notice,
|
14
|
-
# this list of conditions and the following disclaimer.
|
15
|
-
# * Redistributions in binary form must reproduce the above copyright notice,
|
16
|
-
# this list of conditions and the following disclaimer in the documentation
|
17
|
-
# and/or other materials provided with the distribution.
|
18
|
-
# * Neither the name of the copyright holder, nor the names of any other
|
19
|
-
# contributors to this software, may be used to endorse or promote products
|
20
|
-
# derived from this software without specific prior written permission.
|
21
|
-
#
|
22
|
-
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
|
23
|
-
# AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
24
|
-
# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
25
|
-
# ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
|
26
|
-
# LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
|
27
|
-
# CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
|
28
|
-
# SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
|
29
|
-
# INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
|
30
|
-
# CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
|
31
|
-
# ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
|
32
|
-
# POSSIBILITY OF SUCH DAMAGE.
|
33
|
-
# -----------------------------------------------------------------------------
|
34
|
-
;
|
35
|
-
|
36
|
-
|
37
|
-
# :stopdoc:
|
38
|
-
|
39
|
-
|
40
|
-
require 'rgeo/active_record/task_hacker'
|
41
|
-
|
42
|
-
|
43
|
-
class Object
|
44
|
-
alias_method :create_database_without_postgis, :create_database
|
45
|
-
alias_method :drop_database_without_postgis, :drop_database
|
46
|
-
end
|
47
|
-
|
48
|
-
|
49
|
-
namespace :db do
|
50
|
-
namespace :gis do
|
51
|
-
desc "Setup PostGIS data in the database"
|
52
|
-
task :setup => [:load_config, :rails_env] do
|
53
|
-
if self.respond_to?(:configs_for_environment)
|
54
|
-
configs_ = configs_for_environment
|
55
|
-
else
|
56
|
-
environments_ = [::Rails.env]
|
57
|
-
environments_ << 'test' if ::Rails.env.development?
|
58
|
-
configs_ = ::ActiveRecord::Base.configurations.values_at(*environments_).compact.reject{ |config_| config_['database'].blank? }
|
59
|
-
end
|
60
|
-
configs_.each { |config_| setup_gis(config_) }
|
61
|
-
end
|
62
|
-
end
|
63
|
-
end
|
64
|
-
|
65
|
-
|
66
|
-
# superuser name/password
|
67
|
-
def get_su_auth(config_)
|
68
|
-
username_ = config_['username']
|
69
|
-
su_username_ = config_['su_username'] || username_
|
70
|
-
su_password_ = config_['su_password'] || config_['password']
|
71
|
-
has_su_ = config_.include?('su_username')
|
72
|
-
|
73
|
-
[username_, su_username_, su_password_, has_su_]
|
74
|
-
end
|
75
|
-
|
76
|
-
|
77
|
-
def get_search_path(config_)
|
78
|
-
config_["schema_search_path"].to_s.strip.split(',').map(&:strip)
|
79
|
-
end
|
80
|
-
|
81
|
-
|
82
|
-
# Install postgis definitions into the database.
|
83
|
-
# Note: a superuser is required to run the postgis definitions.
|
84
|
-
# If a separate superuser is provided, we need to grant privileges on
|
85
|
-
# the postgis definitions over to the regular user afterwards.
|
86
|
-
# We also need to set the ownership of the postgis tables (spatial_ref_sys
|
87
|
-
# and geometry_columns) to the regular user. This is required to e.g.
|
88
|
-
# be able to disable referential integrity on the database when using
|
89
|
-
# a database cleaner truncation strategy during testing.
|
90
|
-
# The schema for the postgis definitions is chosen as follows:
|
91
|
-
# If "postgis" is present in the search path, use it.
|
92
|
-
# Otherwise, use the last schema in the search path.
|
93
|
-
# If no search path is given, use "public".
|
94
|
-
def setup_gis(config_)
|
95
|
-
# Initial setup of the database: Add schemas from the search path.
|
96
|
-
# If a superuser is given, we log in as the superuser, but we make sure
|
97
|
-
# the schemas are owned by the regular user.
|
98
|
-
username_, su_username_, su_password_, has_su_ = get_su_auth(config_)
|
99
|
-
::ActiveRecord::Base.establish_connection(config_.merge('schema_search_path' => 'public', 'username' => su_username_, 'password' => su_password_))
|
100
|
-
conn_ = ::ActiveRecord::Base.connection
|
101
|
-
search_path_ = get_search_path(config_)
|
102
|
-
quoted_username_ = ::ActiveRecord::Base.connection.quote_column_name(username_)
|
103
|
-
auth_ = has_su_ ? " AUTHORIZATION #{quoted_username_}" : ''
|
104
|
-
search_path_.each do |schema_|
|
105
|
-
exists = schema_.downcase == 'public' || conn_.execute("SELECT 1 FROM pg_catalog.pg_namespace WHERE nspname='#{schema_}'").try(:first)
|
106
|
-
conn_.execute("CREATE SCHEMA #{schema_}#{auth_}") unless exists
|
107
|
-
end
|
108
|
-
|
109
|
-
script_dir_ = config_['script_dir']
|
110
|
-
postgis_extension_ = config_['postgis_extension']
|
111
|
-
if script_dir_ || postgis_extension_
|
112
|
-
postgis_schema_ = search_path_.include?('postgis') ? 'postgis' : (search_path_.last || 'public')
|
113
|
-
if script_dir_
|
114
|
-
# Use script_dir (for postgresql < 9.1 or postgis < 2.0)
|
115
|
-
conn_.execute("SET search_path TO #{postgis_schema_}")
|
116
|
-
conn_.execute(::File.read(::File.expand_path('postgis.sql', script_dir_)))
|
117
|
-
conn_.execute(::File.read(::File.expand_path('spatial_ref_sys.sql', script_dir_)))
|
118
|
-
elsif postgis_extension_
|
119
|
-
# Use postgis_extension (for postgresql >= 9.1 and postgis >= 2.0)
|
120
|
-
postgis_extension_ = 'postgis' if postgis_extension_ == true
|
121
|
-
postgis_extension_ = postgis_extension_.to_s.split(',') unless postgis_extension_.is_a?(::Array)
|
122
|
-
postgis_extension_.each do |extname_|
|
123
|
-
if extname_ == 'postgis_topology'
|
124
|
-
raise ArgumentError, "'topology' must be in schema_search_path for postgis_topology" unless search_path_.include?('topology')
|
125
|
-
conn_.execute("CREATE EXTENSION IF NOT EXISTS #{extname_} SCHEMA topology")
|
126
|
-
else
|
127
|
-
conn_.execute("CREATE EXTENSION IF NOT EXISTS #{extname_} SCHEMA #{postgis_schema_}")
|
128
|
-
end
|
129
|
-
end
|
130
|
-
end
|
131
|
-
if has_su_
|
132
|
-
conn_.execute("GRANT ALL PRIVILEGES ON ALL TABLES IN SCHEMA #{postgis_schema_} TO #{quoted_username_}")
|
133
|
-
conn_.execute("GRANT ALL PRIVILEGES ON ALL FUNCTIONS IN SCHEMA #{postgis_schema_} TO #{quoted_username_}")
|
134
|
-
|
135
|
-
postgis_version = conn_.execute( "SELECT #{postgis_schema_}.postgis_version();" ).first[ 'postgis_version' ]
|
136
|
-
if postgis_version =~ /^2/
|
137
|
-
conn_.execute("ALTER VIEW #{postgis_schema_}.geometry_columns OWNER TO #{quoted_username_}")
|
138
|
-
conn_.execute("ALTER VIEW #{postgis_schema_}.geography_columns OWNER TO #{quoted_username_}")
|
139
|
-
else
|
140
|
-
conn_.execute("ALTER TABLE #{postgis_schema_}.geometry_columns OWNER TO #{quoted_username_}")
|
141
|
-
end
|
142
|
-
conn_.execute("ALTER TABLE #{postgis_schema_}.spatial_ref_sys OWNER TO #{quoted_username_}")
|
143
|
-
end
|
144
|
-
end
|
145
|
-
end
|
146
|
-
|
147
|
-
|
148
|
-
def create_database(config_)
|
149
|
-
if config_['adapter'] == 'postgis'
|
150
|
-
@encoding = config_['encoding'] || ::ENV['CHARSET'] || 'utf8'
|
151
|
-
begin
|
152
|
-
username_, su_username_, su_password_, has_su_ = get_su_auth(config_)
|
153
|
-
|
154
|
-
# Create the database. Optionally do so as the given superuser.
|
155
|
-
# But make sure the database is owned by the regular user.
|
156
|
-
::ActiveRecord::Base.establish_connection(config_.merge('database' => 'postgres', 'schema_search_path' => 'public', 'username' => su_username_, 'password' => su_password_))
|
157
|
-
extra_configs_ = {'encoding' => @encoding}
|
158
|
-
extra_configs_['owner'] = username_ if has_su_
|
159
|
-
::ActiveRecord::Base.connection.create_database(config_['database'], config_.merge(extra_configs_))
|
160
|
-
|
161
|
-
setup_gis(config_)
|
162
|
-
|
163
|
-
# Done
|
164
|
-
::ActiveRecord::Base.establish_connection(config_)
|
165
|
-
rescue ::Exception => e_
|
166
|
-
$stderr.puts(e_, *(e_.backtrace))
|
167
|
-
$stderr.puts("Couldn't create database for #{config_.inspect}")
|
168
|
-
end
|
169
|
-
else
|
170
|
-
create_database_without_postgis(config_)
|
171
|
-
end
|
172
|
-
end
|
173
|
-
|
174
|
-
|
175
|
-
def drop_database(config_)
|
176
|
-
if config_['adapter'] == 'postgis'
|
177
|
-
::ActiveRecord::Base.establish_connection(config_.merge('database' => 'postgres', 'schema_search_path' => 'public', 'username' => config_['su_username'] || config_['username'], 'password' => config_['su_password'] || config_['password']))
|
178
|
-
::ActiveRecord::Base.connection.drop_database(config_['database'])
|
179
|
-
else
|
180
|
-
drop_database_without_postgis(config_)
|
181
|
-
end
|
182
|
-
end
|
183
|
-
|
184
|
-
|
185
|
-
def set_psql_env(config_)
|
186
|
-
::ENV['PGHOST'] = config_["host"] if config_["host"]
|
187
|
-
::ENV['PGPORT'] = config_["port"].to_s if config_["port"]
|
188
|
-
::ENV['PGPASSWORD'] = config_["password"].to_s if config_["password"]
|
189
|
-
::ENV['PGUSER'] = config_["username"].to_s if config_["username"]
|
190
|
-
end
|
191
|
-
|
192
|
-
|
193
|
-
::RGeo::ActiveRecord::TaskHacker.modify('db:charset', nil, 'postgis') do |config_|
|
194
|
-
::ActiveRecord::Base.establish_connection(config_)
|
195
|
-
puts(::ActiveRecord::Base.connection.encoding)
|
196
|
-
end
|
197
|
-
|
198
|
-
|
199
|
-
::RGeo::ActiveRecord::TaskHacker.modify('db:structure:dump', nil, 'postgis') do |config_|
|
200
|
-
set_psql_env(config_)
|
201
|
-
filename_ = ::File.join(::Rails.root, "db/#{::Rails.env}_structure.sql")
|
202
|
-
search_path_ = config_["schema_search_path"].to_s.strip
|
203
|
-
search_path_ = search_path_.split(",").map{ |sp_| sp_.strip }
|
204
|
-
search_path_.delete('postgis')
|
205
|
-
search_path_ = ['public'] if search_path_.length == 0
|
206
|
-
search_path_ = search_path_.map{ |sp_| "--schema=#{sp_}" }.join(" ")
|
207
|
-
`pg_dump -i -U "#{config_["username"]}" -s -x -O -f #{filename_} #{search_path_} #{config_["database"]}`
|
208
|
-
raise "Error dumping database" if $?.exitstatus == 1
|
209
|
-
end
|
210
|
-
|
211
|
-
|
212
|
-
::RGeo::ActiveRecord::TaskHacker.modify('db:structure:load', nil, 'postgis') do |config_|
|
213
|
-
set_psql_env(config_)
|
214
|
-
filename_ = ::File.join(::Rails.root, "db/#{::Rails.env}_structure.sql")
|
215
|
-
`psql -f #{filename_} #{config_["database"]}`
|
216
|
-
end
|
217
|
-
|
218
|
-
|
219
|
-
::RGeo::ActiveRecord::TaskHacker.modify('db:test:clone_structure', 'test', 'postgis') do |config_|
|
220
|
-
set_psql_env(config_)
|
221
|
-
`psql -U "#{config_["username"]}" -f #{::Rails.root}/db/#{::Rails.env}_structure.sql #{config_["database"]}`
|
222
|
-
end
|
223
|
-
|
224
|
-
|
225
|
-
::RGeo::ActiveRecord::TaskHacker.modify('db:test:purge', 'test', 'postgis') do |config_|
|
226
|
-
::ActiveRecord::Base.clear_active_connections!
|
227
|
-
drop_database(config_)
|
228
|
-
create_database(config_)
|
229
|
-
end
|
230
|
-
|
231
|
-
|
232
|
-
# :startdoc:
|
@@ -1,298 +0,0 @@
|
|
1
|
-
# -----------------------------------------------------------------------------
|
2
|
-
#
|
3
|
-
# PostGIS adapter for ActiveRecord
|
4
|
-
#
|
5
|
-
# -----------------------------------------------------------------------------
|
6
|
-
# Copyright 2010-2012 Daniel Azuma
|
7
|
-
#
|
8
|
-
# All rights reserved.
|
9
|
-
#
|
10
|
-
# Redistribution and use in source and binary forms, with or without
|
11
|
-
# modification, are permitted provided that the following conditions are met:
|
12
|
-
#
|
13
|
-
# * Redistributions of source code must retain the above copyright notice,
|
14
|
-
# this list of conditions and the following disclaimer.
|
15
|
-
# * Redistributions in binary form must reproduce the above copyright notice,
|
16
|
-
# this list of conditions and the following disclaimer in the documentation
|
17
|
-
# and/or other materials provided with the distribution.
|
18
|
-
# * Neither the name of the copyright holder, nor the names of any other
|
19
|
-
# contributors to this software, may be used to endorse or promote products
|
20
|
-
# derived from this software without specific prior written permission.
|
21
|
-
#
|
22
|
-
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
|
23
|
-
# AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
24
|
-
# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
25
|
-
# ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
|
26
|
-
# LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
|
27
|
-
# CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
|
28
|
-
# SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
|
29
|
-
# INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
|
30
|
-
# CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
|
31
|
-
# ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
|
32
|
-
# POSSIBILITY OF SUCH DAMAGE.
|
33
|
-
# -----------------------------------------------------------------------------
|
34
|
-
;
|
35
|
-
|
36
|
-
|
37
|
-
module ActiveRecord # :nodoc:
|
38
|
-
|
39
|
-
module ConnectionAdapters # :nodoc:
|
40
|
-
|
41
|
-
module PostGISAdapter # :nodoc:
|
42
|
-
|
43
|
-
|
44
|
-
class MainAdapter < PostgreSQLAdapter # :nodoc:
|
45
|
-
|
46
|
-
|
47
|
-
def initialize(*args_)
|
48
|
-
super
|
49
|
-
# Rails 3.2 way of defining the visitor: do so in the constructor
|
50
|
-
if defined?(@visitor) && @visitor
|
51
|
-
@visitor = ::Arel::Visitors::PostGIS.new(self)
|
52
|
-
end
|
53
|
-
end
|
54
|
-
|
55
|
-
|
56
|
-
include PostGISAdapter::CommonAdapterMethods
|
57
|
-
|
58
|
-
|
59
|
-
@@native_database_types = nil
|
60
|
-
|
61
|
-
def native_database_types
|
62
|
-
@@native_database_types ||= super.merge(:spatial => {:name => 'geometry'})
|
63
|
-
end
|
64
|
-
|
65
|
-
|
66
|
-
def type_cast(value_, column_)
|
67
|
-
if ::RGeo::Feature::Geometry.check_type(value_)
|
68
|
-
::RGeo::WKRep::WKBGenerator.new(:hex_format => true, :type_format => :ewkb, :emit_ewkb_srid => true).generate(value_)
|
69
|
-
else
|
70
|
-
super
|
71
|
-
end
|
72
|
-
end
|
73
|
-
|
74
|
-
|
75
|
-
def columns(table_name_, name_=nil)
|
76
|
-
# FULL REPLACEMENT. RE-CHECK ON NEW VERSIONS.
|
77
|
-
# We needed to return a spatial column subclass.
|
78
|
-
table_name_ = table_name_.to_s
|
79
|
-
spatial_info_ = spatial_column_info(table_name_)
|
80
|
-
column_definitions(table_name_).collect do |col_name_, type_, default_, notnull_|
|
81
|
-
# JDBC support: JDBC adapter returns a hash for column definitions,
|
82
|
-
# instead of an array of values.
|
83
|
-
if col_name_.kind_of?(::Hash)
|
84
|
-
notnull_ = col_name_["column_not_null"]
|
85
|
-
default_ = col_name_["column_default"]
|
86
|
-
type_ = col_name_["column_type"]
|
87
|
-
col_name_ = col_name_["column_name"]
|
88
|
-
end
|
89
|
-
|
90
|
-
SpatialColumn.new(col_name_, default_, type_, notnull_ == 'f',
|
91
|
-
((type_ =~ /geometry/i ? spatial_info_[col_name_] : nil) || {}).merge({
|
92
|
-
:table_name => table_name_,
|
93
|
-
:factory_settings => @rgeo_factory_settings
|
94
|
-
}))
|
95
|
-
end
|
96
|
-
end
|
97
|
-
|
98
|
-
|
99
|
-
def indexes(table_name_, name_=nil)
|
100
|
-
# FULL REPLACEMENT. RE-CHECK ON NEW VERSIONS.
|
101
|
-
# We needed to modify the catalog queries to pull the index type info.
|
102
|
-
|
103
|
-
# Remove postgis from schemas
|
104
|
-
schemas_ = schema_search_path.split(/,/)
|
105
|
-
schemas_.delete('postgis')
|
106
|
-
schemas_ = schemas_.map{ |p_| quote(p_) }.join(',')
|
107
|
-
|
108
|
-
# Get index type by joining with pg_am.
|
109
|
-
result_ = query(<<-SQL, name_)
|
110
|
-
SELECT DISTINCT i.relname, d.indisunique, d.indkey, t.oid, am.amname
|
111
|
-
FROM pg_class t, pg_class i, pg_index d, pg_am am
|
112
|
-
WHERE i.relkind = 'i'
|
113
|
-
AND d.indexrelid = i.oid
|
114
|
-
AND d.indisprimary = 'f'
|
115
|
-
AND t.oid = d.indrelid
|
116
|
-
AND t.relname = '#{table_name_}'
|
117
|
-
AND i.relnamespace IN (SELECT oid FROM pg_namespace WHERE nspname IN (#{schemas_}) )
|
118
|
-
AND i.relam = am.oid
|
119
|
-
ORDER BY i.relname
|
120
|
-
SQL
|
121
|
-
|
122
|
-
result_.map do |row_|
|
123
|
-
index_name_ = row_[0]
|
124
|
-
unique_ = row_[1] == 't'
|
125
|
-
indkey_ = row_[2].split(" ")
|
126
|
-
oid_ = row_[3]
|
127
|
-
indtype_ = row_[4]
|
128
|
-
|
129
|
-
columns_ = query(<<-SQL, "Columns for index #{row_[0]} on #{table_name_}").inject({}){ |h_, r_| h_[r_[0].to_s] = [r_[1], r_[2]]; h_ }
|
130
|
-
SELECT a.attnum, a.attname, t.typname
|
131
|
-
FROM pg_attribute a, pg_type t
|
132
|
-
WHERE a.attrelid = #{oid_}
|
133
|
-
AND a.attnum IN (#{indkey_.join(",")})
|
134
|
-
AND a.atttypid = t.oid
|
135
|
-
SQL
|
136
|
-
|
137
|
-
spatial_ = indtype_ == 'gist' && columns_.size == 1 && (columns_.values.first[1] == 'geometry' || columns_.values.first[1] == 'geography')
|
138
|
-
column_names_ = columns_.values_at(*indkey_).compact.map{ |a_| a_[0] }
|
139
|
-
if column_names_.empty?
|
140
|
-
nil
|
141
|
-
else
|
142
|
-
::RGeo::ActiveRecord::SpatialIndexDefinition.new(table_name_, index_name_, unique_, column_names_, nil, nil, nil, spatial_ ? true : false)
|
143
|
-
end
|
144
|
-
end.compact
|
145
|
-
end
|
146
|
-
|
147
|
-
|
148
|
-
def create_table(table_name_, options_={})
|
149
|
-
# FULL REPLACEMENT. RE-CHECK ON NEW VERSIONS.
|
150
|
-
# Note: we have to do a full replacement for Rails 3.0 because
|
151
|
-
# there is no way to override the creation of the table
|
152
|
-
# definition object. In Rails 3.1, this has been factored out
|
153
|
-
# into the table_definition method, so we could rewrite this
|
154
|
-
# to call super if we're willing to go 3.1 only.
|
155
|
-
table_name_ = table_name_.to_s
|
156
|
-
table_definition_ = SpatialTableDefinition.new(self)
|
157
|
-
table_definition_.primary_key(options_[:primary_key] || ::ActiveRecord::Base.get_primary_key(table_name_.singularize)) unless options_[:id] == false
|
158
|
-
yield table_definition_ if block_given?
|
159
|
-
if options_[:force] && table_exists?(table_name_)
|
160
|
-
drop_table(table_name_, options_)
|
161
|
-
end
|
162
|
-
|
163
|
-
create_sql_ = "CREATE#{' TEMPORARY' if options_[:temporary]} TABLE "
|
164
|
-
create_sql_ << "#{quote_table_name(table_name_)} ("
|
165
|
-
create_sql_ << table_definition_.to_sql
|
166
|
-
create_sql_ << ") #{options_[:options]}"
|
167
|
-
execute create_sql_
|
168
|
-
|
169
|
-
table_definition_.non_geographic_spatial_columns.each do |col_|
|
170
|
-
type_ = col_.spatial_type.gsub('_', '').upcase
|
171
|
-
has_z_ = col_.has_z?
|
172
|
-
has_m_ = col_.has_m?
|
173
|
-
type_ = "#{type_}M" if has_m_ && !has_z_
|
174
|
-
dimensions_ = 2
|
175
|
-
dimensions_ += 1 if has_z_
|
176
|
-
dimensions_ += 1 if has_m_
|
177
|
-
execute("SELECT AddGeometryColumn('#{quote_string(table_name_)}', '#{quote_string(col_.name.to_s)}', #{col_.srid}, '#{quote_string(type_)}', #{dimensions_})")
|
178
|
-
end
|
179
|
-
end
|
180
|
-
|
181
|
-
|
182
|
-
def drop_table(table_name_, *options_)
|
183
|
-
if postgis_lib_version.to_s.split('.').first.to_i == 1
|
184
|
-
execute("DELETE from geometry_columns where f_table_name='#{quote_string(table_name_.to_s)}'")
|
185
|
-
end
|
186
|
-
super
|
187
|
-
end
|
188
|
-
|
189
|
-
|
190
|
-
def add_column(table_name_, column_name_, type_, options_={})
|
191
|
-
table_name_ = table_name_.to_s
|
192
|
-
if (info_ = spatial_column_constructor(type_.to_sym))
|
193
|
-
limit_ = options_[:limit]
|
194
|
-
if type_.to_s == 'geometry' &&
|
195
|
-
(options_[:no_constraints] || limit_.is_a?(::Hash) && limit_[:no_constraints])
|
196
|
-
then
|
197
|
-
options_.delete(:limit)
|
198
|
-
super
|
199
|
-
else
|
200
|
-
options_.merge!(limit_) if limit_.is_a?(::Hash)
|
201
|
-
type_ = (options_[:type] || info_[:type] || type_).to_s.gsub('_', '').upcase
|
202
|
-
has_z_ = options_[:has_z]
|
203
|
-
has_m_ = options_[:has_m]
|
204
|
-
srid_ = (options_[:srid] || -1).to_i
|
205
|
-
if options_[:geographic]
|
206
|
-
type_ << 'Z' if has_z_
|
207
|
-
type_ << 'M' if has_m_
|
208
|
-
execute("ALTER TABLE #{quote_table_name(table_name_)} ADD COLUMN #{quote_column_name(column_name_)} GEOGRAPHY(#{type_},#{srid_})")
|
209
|
-
change_column_default(table_name_, column_name_, options_[:default]) if options_include_default?(options_)
|
210
|
-
change_column_null(table_name_, column_name_, false, options_[:default]) if options_[:null] == false
|
211
|
-
else
|
212
|
-
type_ = "#{type_}M" if has_m_ && !has_z_
|
213
|
-
dimensions_ = 2
|
214
|
-
dimensions_ += 1 if has_z_
|
215
|
-
dimensions_ += 1 if has_m_
|
216
|
-
execute("SELECT AddGeometryColumn('#{quote_string(table_name_)}', '#{quote_string(column_name_.to_s)}', #{srid_}, '#{quote_string(type_)}', #{dimensions_})")
|
217
|
-
end
|
218
|
-
end
|
219
|
-
else
|
220
|
-
super
|
221
|
-
end
|
222
|
-
end
|
223
|
-
|
224
|
-
|
225
|
-
def remove_column(table_name_, *column_names_)
|
226
|
-
column_names_ = column_names_.flatten.map{ |n_| n_.to_s }
|
227
|
-
spatial_info_ = spatial_column_info(table_name_)
|
228
|
-
remaining_column_names_ = []
|
229
|
-
column_names_.each do |name_|
|
230
|
-
if spatial_info_.include?(name_)
|
231
|
-
execute("SELECT DropGeometryColumn('#{quote_string(table_name_.to_s)}','#{quote_string(name_)}')")
|
232
|
-
else
|
233
|
-
remaining_column_names_ << name_.to_sym
|
234
|
-
end
|
235
|
-
end
|
236
|
-
if remaining_column_names_.size > 0
|
237
|
-
super(table_name_, *remaining_column_names_)
|
238
|
-
end
|
239
|
-
end
|
240
|
-
|
241
|
-
|
242
|
-
def add_index(table_name_, column_name_, options_={})
|
243
|
-
# FULL REPLACEMENT. RE-CHECK ON NEW VERSIONS.
|
244
|
-
# We have to fully-replace because of the gist_clause.
|
245
|
-
table_name_ = table_name_.to_s
|
246
|
-
column_names_ = ::Array.wrap(column_name_)
|
247
|
-
index_name_ = index_name(table_name_, :column => column_names_)
|
248
|
-
gist_clause_ = ''
|
249
|
-
index_type_ = ''
|
250
|
-
if ::Hash === options_ # legacy support, since this param was a string
|
251
|
-
index_type_ = 'UNIQUE' if options_[:unique]
|
252
|
-
index_name_ = options_[:name].to_s if options_.key?(:name)
|
253
|
-
gist_clause_ = 'USING GIST' if options_[:spatial]
|
254
|
-
else
|
255
|
-
index_type_ = options_
|
256
|
-
end
|
257
|
-
if index_name_.length > index_name_length
|
258
|
-
raise ::ArgumentError, "Index name '#{index_name_}' on table '#{table_name_}' is too long; the limit is #{index_name_length} characters"
|
259
|
-
end
|
260
|
-
if index_name_exists?(table_name_, index_name_, false)
|
261
|
-
raise ::ArgumentError, "Index name '#{index_name_}' on table '#{table_name_}' already exists"
|
262
|
-
end
|
263
|
-
quoted_column_names_ = quoted_columns_for_index(column_names_, options_).join(", ")
|
264
|
-
execute "CREATE #{index_type_} INDEX #{quote_column_name(index_name_)} ON #{quote_table_name(table_name_)} #{gist_clause_} (#{quoted_column_names_})"
|
265
|
-
end
|
266
|
-
|
267
|
-
|
268
|
-
def spatial_column_info(table_name_)
|
269
|
-
info_ = query("SELECT * FROM geometry_columns WHERE f_table_name='#{quote_string(table_name_.to_s)}'")
|
270
|
-
result_ = {}
|
271
|
-
info_.each do |row_|
|
272
|
-
name_ = row_[3]
|
273
|
-
type_ = row_[6]
|
274
|
-
dimension_ = row_[4].to_i
|
275
|
-
has_m_ = type_ =~ /m$/i ? true : false
|
276
|
-
type_.sub!(/m$/, '')
|
277
|
-
has_z_ = dimension_ > 3 || dimension_ == 3 && !has_m_
|
278
|
-
result_[name_] = {
|
279
|
-
:name => name_,
|
280
|
-
:type => type_,
|
281
|
-
:dimension => dimension_,
|
282
|
-
:srid => row_[5].to_i,
|
283
|
-
:has_z => has_z_,
|
284
|
-
:has_m => has_m_,
|
285
|
-
}
|
286
|
-
end
|
287
|
-
result_
|
288
|
-
end
|
289
|
-
|
290
|
-
|
291
|
-
end
|
292
|
-
|
293
|
-
|
294
|
-
end
|
295
|
-
|
296
|
-
end
|
297
|
-
|
298
|
-
end
|