Contents

module DbClustering
  module DatasourceAdapters
    class ActiveRecord

      def initialize(relation:)
        @relation = relation
      end

      def iterate_all_points
        @relation.find_each do |datasource_point|
          point = DbClustering::Models::Point.new(datasource_point)
          yield(point)
        end
      end

      def neighbors(point:, distance_metric:, max_distance:)
        neighbors = []

        @relation.find_each do |neighbor_candidate|
          candidate_point = DbClustering::Models::Point.new(neighbor_candidate)

          if distance_metric.distance(point.vector, candidate_point.vector) <= max_distance
            neighbors << candidate_point
          end
        end

        neighbors
      end

    end
  end
end

Version data entries

4 entries across 4 versions & 1 rubygems

Version	Path
db_clustering-0.1.4	lib/datasource_adapters/active_record.rb
db_clustering-0.1.3	lib/datasource_adapters/active_record.rb
db_clustering-0.1.2	lib/datasource_adapters/active_record.rb
db_clustering-0.1.1	lib/datasource_adapters/active_record.rb