andmemasin\myabstract\ActiveRecord

Complexity

Total Complexity

6

Size/Duplication

Total Lines	33
Duplicated Lines	0 %

Importance

Changes

0

3 Methods

Rating	Name	Duplication	Size	Complexity
A	primaryKeySingle()	0	6	2
A	hasOne()	0	6	2
A	hasMany()	0	6	2

<?php

namespace andmemasin\myabstract;

use yii\db\ActiveRecord as BaseActiveRecord;
use yii\base\NotSupportedException;

/**
 * Class ActiveRecord
 * @package andmemasin\myabstract
 * @author Tõnis Ormisson <tonis@andmemasin.eu>
 */
class ActiveRecord extends BaseActiveRecord
{
    /**
     * Get the primary key column as string if the one-column PK
     * NB! Always use single column Primary-keys!
     * NB! this assumes that primary key always has the table_name_id format
     * @return string
     * @throws NotSupportedException if multi-column PrimaryKey is used
     */
    public function primaryKeySingle()
    {
        if (count(static::primaryKey()) === 1) {
            return static::primaryKey()[0];
        }
        throw new NotSupportedException('Not supported for multi-column primary keys');
    }

    /** {@inheritdoc} */
    public function hasMany($class, $link = null)
    {
        if (empty($link)) {
            $link = [$this->primaryKeySingle() => $this->primaryKeySingle()];
        }
        return parent::hasMany($class, $link);
    }

    /** {@inheritdoc} */
    public function hasOne($class, $link = null)
    {
        if (empty($link)) {
            $link = [$this->primaryKeySingle() => $this->primaryKeySingle()];
        }
        return parent::hasOne($class, $link);
    }

}