Contents

require 'cassanity/argument_generators/with_clause'

module Cassanity
  module ArgumentGenerators
    class ColumnFamilyCreate

      # Internal
      def initialize(args = {})
        @with_clause = args.fetch(:with_clause) { WithClause.new }
      end

      # Internal
      def call(args = {})
        name         = args.fetch(:column_family_name)
        schema       = args.fetch(:schema)
        columns      = schema.columns
        primary_keys = schema.primary_keys
        with         = schema.with

        definitions, variables = [], []

        if (keyspace_name = args[:keyspace_name])
          name = "#{keyspace_name}.#{name}"
        end

        columns.each do |name, type|
          definitions << "#{name} #{type}"
        end

        definitions << "PRIMARY KEY (#{compose_primary_key(primary_keys).join(', ')})"

        cql_definition = definitions.join(', ')

        cql = "CREATE COLUMNFAMILY #{name} (#{cql_definition})"

        with_cql, *with_variables = @with_clause.call(with: with)
        cql << with_cql
        variables.concat(with_variables)

        [cql, *variables]
      end

      def compose_primary_key(primary_keys)
        primary_keys.map do |key|
          if key.is_a? Array
            "(#{key.join(', ')})"
          else
            key
          end
        end
      end
    end
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
cassanity-0.6.0	lib/cassanity/argument_generators/column_family_create.rb