Sha256: 440506225e1f177a3433f0dc0addaf57259959fbbac57342cc6baefc165dee97
Contents?: true
Size: 1.89 KB
Versions: 29
Compression:
Stored size: 1.89 KB
Contents
# frozen_string_literal: true
module ActiveRecord # :nodoc:
module ConnectionAdapters # :nodoc:
module CockroachDB # :nodoc:
class TableDefinition < PostgreSQL::TableDefinition # :nodoc:
include ColumnMethods
# Support for spatial columns in tables
# super: https://github.com/rails/rails/blob/master/activerecord/lib/active_record/connection_adapters/abstract/schema_definitions.rb
def new_column_definition(name, type, **options)
if (info = CockroachDBAdapter.spatial_column_options(type.to_sym))
if (limit = options.delete(:limit)) && limit.is_a?(::Hash)
options.merge!(limit)
end
geo_type = ColumnDefinitionUtils.geo_type(options[:type] || type || info[:type])
base_type = info[:type] || (options[:geographic] ? :geography : :geometry)
options[:limit] = ColumnDefinitionUtils.limit_from_options(geo_type, options)
options[:spatial_type] = geo_type
column = super(name, base_type, **options)
else
column = super(name, type, **options)
end
column
end
end
module ColumnDefinitionUtils
class << self
def geo_type(type = 'GEOMETRY')
g_type = type.to_s.delete('_').upcase
return 'POINT' if g_type == 'STPOINT'
return 'POLYGON' if g_type == 'STPOLYGON'
g_type
end
def limit_from_options(type, options = {})
spatial_type = geo_type(type)
spatial_type << 'Z' if options[:has_z]
spatial_type << 'M' if options[:has_m]
spatial_type << ",#{options[:srid] || default_srid(options)}"
spatial_type
end
def default_srid(options)
options[:geographic] ? 4326 : CockroachDBAdapter::DEFAULT_SRID
end
end
end
end
end
end
Version data entries
29 entries across 29 versions & 1 rubygems