Module: E2Model::DatasetMethods

Defined in:
lib/engine2/core.rb

Instance Method Summary collapse

Instance Method Details

#ensure_primary_key ⇒ Object

Raises:



331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
 
# File 'lib/engine2/core.rb', line 331

def ensure_primary_key
    pk = model.primary_keys
    raise Engine2::E2Error.new("No primary key defined for model #{model}") unless pk && pk.all?

    if opts_select = @opts[:select]
        sel_pk = []
        opts_select.each do |sel|
            name = case sel
                when Symbol
                    sel
                when Sequel::SQL::QualifiedIdentifier
                    sel.column
                when Sequel::SQL::AliasedExpression
                    sel
                    # nil #sel.aliaz # ?
                    # sel.expression
                end
            sel_pk << name if name && pk.include?(name)
        end

        if pk.length == sel_pk.length
            self
        else
            sels = (pk - sel_pk).map{|k| model.table_name.q(k)}
            select_more(*sels)
        end
    else
        select(*pk.map{|k| model.table_name.q(k)})
    end

end

#extract_select(sel, al = nil, &blk) ⇒ Object 



363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
 
# File 'lib/engine2/core.rb', line 363

def extract_select sel, al = nil, &blk
    case sel
    when Symbol
        yield nil, sel, nil
    when Sequel::SQL::QualifiedIdentifier
        yield sel.table, sel.column, al
    when Sequel::SQL::AliasedExpression, Sequel::SQL::Function
        sel
        # extract_select sel.expression, sel.aliaz, &blk
        # expr = sel.expression
        # yield  expr.table, expr.column
    else
        raise Engine2::E2Error.new("Unknown selection #{sel}")
    end
end

#get_opts ⇒ Object 



436
437
438
 
# File 'lib/engine2/core.rb', line 436

def get_opts
    @opts
end

#setup!(fields) ⇒ Object 



379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
 
# File 'lib/engine2/core.rb', line 379

def setup! fields
    joins = {}
    type_info = model.type_info
    model_table_name = model.table_name

    @opts[:select] = @opts[:select].map do |sel|
        extract_select sel do |table, name, aliaz|
            info = if table
                if table == model_table_name
                    model
                else
                    assoc = model.many_to_one_associations[table] || model.many_to_many_associations[table]
                    raise Engine2::E2Error.new("Association #{table} not found for model #{model}") unless assoc
                    assoc.associated_class
                end.type_info
            else
                type_info
            end

            table ||= model_table_name
            if table == model_table_name
                fields << name
            else
                fields << table.q(name)
                joins[table] ||= model.many_to_one_associations[table] || model.many_to_many_associations[table]
            end

            f_info = info[name]
            raise Engine2::E2Error.new("Column #{name} not found for table #{table || model_table_name}") unless f_info
            if f_info[:dummy]
                nil
            else
                qname = table.q(name)
                if table != model_table_name
                    Sequel.alias_columns_in_joins ? qname.as(:"#{table}__#{name}") : qname
                else
                    qname
                end
            end
        end
    end

    @opts[:select].compact!.freeze

    joins.reduce(self) do |joined, (table, assoc)|
        m = assoc.associated_class
        case assoc[:type]
        when :many_to_one
            keys = assoc[:qualified_key]
            joined.left_join(table, m.primary_keys.zip(keys.is_a?(Array) ? keys : [keys]))
        when :many_to_many
            joined.left_join(assoc[:join_table], assoc[:left_keys].zip(model.primary_keys)).left_join(m.table_name, m.primary_keys.zip(assoc[:right_keys]))
        else unsupported_association
        end
    end
end

#with_proc(&blk) ⇒ Object 



440
441
442
443
444
 
# File 'lib/engine2/core.rb', line 440

def with_proc &blk
    ds = clone
    ds.row_proc = blk
    ds
end