lib/mysql_point.rb in mysql_point-1.0.0 vs lib/mysql_point.rb in mysql_point-1.0.1

@@ -6,15 +6,15 @@
 require_relative 'mysql_point/version'
 require_relative 'mysql_point/coordinate'
 require_relative 'coordinate_validator'
 
 # enables column type in ActiveRecord
-require_relative 'mysql_point/type'
-ActiveRecord::Base.connection.type_map.register_type(/^point/, MySQLPoint::Type.new)
+require_relative 'mysql_point/type_map'
+ActiveRecord::ConnectionAdapters::Mysql2Adapter.prepend MySQLPoint::TypeMap
 
 # enables a Coordinate to be sent as WKT and converted by a MySQL function on insert/update
 require_relative 'mysql_point/quoting'
-ActiveRecord::Base.connection.class.prepend MySQLPoint::Quoting
+ActiveRecord::ConnectionAdapters::Mysql2Adapter.prepend MySQLPoint::Quoting
 
 # enables shorthand syntax in migrations and db/schema.rb
 require_relative 'mysql_point/column_method'
 ActiveRecord::ConnectionAdapters::MySQL::TableDefinition.include MySQLPoint::ColumnMethod
 ActiveRecord::ConnectionAdapters::MySQL::Table.include MySQLPoint::ColumnMethod