CartoDB/cartodb20

View on GitHub
services/importer/lib/importer/overviews.rb

Summary

Maintainability
A
0 mins
Test Coverage
module CartoDB
  module Importer2
    # Overviews creation service
    #
    # Pending issues: metrics, quotas/limits, timing, logging, ...
    #
    class Overviews

      include ::LoggerHelper

      DEFAULT_MIN_ROWS = 1000000
      DEFAULT_STATEMENT_TIMEOUT = 10 * 60 * 1000 # ms
      DEFAULT_TOLERANCE_PX = 1.0 # px

      def initialize(runner, user, options = {})
        @runner            = runner
        @user              = user
        @schema            = user.database_schema
        @min_rows          = options[:min_rows] ||
                             Cartodb.get_config(:overviews, 'min_rows') ||
                             DEFAULT_MIN_ROWS
        @statement_timeout = options[:statement_timeout] ||
                             Cartodb.get_config(:overviews, 'statement_timeout') ||
                             DEFAULT_STATEMENT_TIMEOUT
        @tolerance_px      = options[:tolerance_px] ||
                             Cartodb.get_config(:overviews, 'tolerance_px') ||
                             DEFAULT_TOLERANCE_PX
        @importer_stats = CartoDB::Stats::Importer.instance # TODO: delegate to @runner?
        @logger = options[:log]
      end

      attr_reader :user, :min_rows, :schema

      # Obtain an object fo thandle the creation of overviews for a dataset
      def dataset(table_name)
        Dataset.new(self, table_name)
      end

      def log(message)
        log = @logger || @runner.log
        log.append(message)
      end

      def get_table(table_name)
        UserTable.find_by_identifier(@user.id, table_name).service
      end

      def can_create_overviews?
        false
      end

      def create_overviews!(table_name)
        log_info(message: 'create_overviews! called: Doing nothing',
                 target_user: @user, table: { name: table_name })
      end

      def delete_overviews!(table_name)
        @user.transaction_with_timeout statement_timeout: @statement_timeout do |db|
          log("Will delete overviews for #{table_name}")
          Carto::OverviewsService.new(db).delete_overviews(table_name)
          log("Overviews deleted for #{table_name}")
        end
      end

      # Dataset overview creation
      class Dataset
        def initialize(overviews_service, table_name)
          @service = overviews_service
          @name = table_name
          @table = @service.get_table(@name)
          @table_name = table_name_with_schema(@service.schema, @name)
        end

        def should_create_overviews?
          false
        end

        def create_overviews!
          @service.create_overviews! @table_name if should_create_overviews?
        end

        def delete_overviews!
          @service.delete_overviews! @table_name
        end

        def update_overviews!
          delete_overviews!
          create_overviews!
        end

        private

        def table_row_count
          @table.rows_estimated(@service.user)
        end

        def supported_geometry?
          @table.geometry_types == ['ST_Point']
        end

        def table_name_with_schema(schema, table)
          # TODO: quote if necessary
          "#{schema}.#{table}"
        end
      end
    end
  end
end