qqq
This commit is contained in:
@@ -15,9 +15,11 @@ use Database\Affair\Commit;
|
||||
use Database\Affair\Rollback;
|
||||
use Database\Traits\QueryTrait;
|
||||
use Exception;
|
||||
use Kiri;
|
||||
use Kiri\Exception\ConfigException;
|
||||
use Psr\Container\ContainerExceptionInterface;
|
||||
use Psr\Container\NotFoundExceptionInterface;
|
||||
use Throwable;
|
||||
|
||||
/**
|
||||
* Class Db
|
||||
@@ -25,345 +27,223 @@ use Psr\Container\NotFoundExceptionInterface;
|
||||
*/
|
||||
class Db implements ISqlBuilder
|
||||
{
|
||||
use QueryTrait;
|
||||
use QueryTrait;
|
||||
|
||||
|
||||
private static bool $_inTransaction = false;
|
||||
|
||||
/**
|
||||
* @return void
|
||||
*/
|
||||
public static function beginTransaction(): void
|
||||
{
|
||||
fire(new BeginTransaction());
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* @param Closure $closure
|
||||
* @param mixed ...$params
|
||||
* @return mixed
|
||||
* @throws ContainerExceptionInterface
|
||||
* @throws NotFoundExceptionInterface
|
||||
* @throws Exception
|
||||
*/
|
||||
public static function Transaction(Closure $closure, ...$params): mixed
|
||||
{
|
||||
static::beginTransaction();
|
||||
try {
|
||||
$result = call_user_func($closure, ...$params);
|
||||
} catch (\Throwable $throwable) {
|
||||
error($throwable);
|
||||
$result = addError($throwable->getMessage(), 'mysql');
|
||||
} finally {
|
||||
if ($result === false) {
|
||||
static::rollback();
|
||||
} else {
|
||||
static::commit();
|
||||
}
|
||||
return $result;
|
||||
}
|
||||
}
|
||||
private static bool $_inTransaction = false;
|
||||
|
||||
|
||||
/**
|
||||
* @return void
|
||||
* @var Connection|null
|
||||
*/
|
||||
public static function commit(): void
|
||||
{
|
||||
fire(new Commit());
|
||||
}
|
||||
private ?Connection $connection = null;
|
||||
|
||||
|
||||
/**
|
||||
* @return void
|
||||
* @param string|Connection $dbname
|
||||
* @return Db
|
||||
* @throws \Exception
|
||||
*/
|
||||
public static function rollback(): void
|
||||
{
|
||||
fire(new Rollback());
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* @param $table
|
||||
*
|
||||
* @return static
|
||||
*/
|
||||
public static function table($table): Db|static
|
||||
{
|
||||
$connection = new Db();
|
||||
$connection->from($table);
|
||||
return $connection;
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* @param string $column
|
||||
* @param string $alias
|
||||
* @return string
|
||||
*/
|
||||
public static function any_value(string $column, string $alias = ''): string
|
||||
{
|
||||
if (empty($alias)) {
|
||||
$alias = $column . '_any_value';
|
||||
}
|
||||
return 'ANY_VALUE(' . $column . ') as ' . $alias;
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* @param string $column
|
||||
* @return string
|
||||
*/
|
||||
public static function increment(string $column): string
|
||||
{
|
||||
return '+ ' . $column;
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* @param string $column
|
||||
* @return string
|
||||
*/
|
||||
public static function decrement(string $column): string
|
||||
{
|
||||
return '- ' . $column;
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* @param Connection|null $connection
|
||||
* @return mixed
|
||||
* @throws Exception
|
||||
*/
|
||||
public function get(Connection $connection = NULL): mixed
|
||||
{
|
||||
$connection = static::getDefaultConnection($connection);
|
||||
|
||||
return $connection->createCommand(SqlBuilder::builder($this)->one())
|
||||
->all();
|
||||
}
|
||||
|
||||
/**
|
||||
* @param $column
|
||||
* @return string
|
||||
*/
|
||||
public static function raw($column): string
|
||||
{
|
||||
return '`' . $column . '`';
|
||||
}
|
||||
|
||||
/**
|
||||
* @param Connection|null $connection
|
||||
* @return array|bool|null
|
||||
* @throws Exception
|
||||
*/
|
||||
public function find(Connection $connection = NULL): array|bool|null
|
||||
public static function connect(string|Connection $dbname): Db
|
||||
{
|
||||
$connection = static::getDefaultConnection($connection);
|
||||
$db = new Db();
|
||||
if (is_string($dbname)) {
|
||||
$dbname = Kiri::service()->get($dbname);
|
||||
}
|
||||
$db->connection = $dbname;
|
||||
return $db;
|
||||
}
|
||||
|
||||
return $connection->createCommand(SqlBuilder::builder($this)->all())
|
||||
->one();
|
||||
}
|
||||
|
||||
/**
|
||||
* @param Connection|NULL $connection
|
||||
* @return bool|int
|
||||
* @throws Exception
|
||||
*/
|
||||
public function count(Connection $connection = NULL): bool|int
|
||||
{
|
||||
$connection = static::getDefaultConnection($connection);
|
||||
|
||||
return $connection->createCommand(SqlBuilder::builder($this)->count())
|
||||
->exec();
|
||||
}
|
||||
|
||||
/**
|
||||
* @param Connection|NULL $connection
|
||||
* @return bool|int
|
||||
* @throws Exception
|
||||
*/
|
||||
public function exists(Connection $connection = NULL): bool|int
|
||||
{
|
||||
$connection = static::getDefaultConnection($connection);
|
||||
|
||||
return $connection->createCommand(SqlBuilder::builder($this)->one())
|
||||
->fetchColumn();
|
||||
}
|
||||
|
||||
/**
|
||||
* @param string $sql
|
||||
* @param array $attributes
|
||||
* @param Connection|null $connection
|
||||
* @return array|bool|int|string|null
|
||||
* @throws Exception
|
||||
*/
|
||||
public static function findAllBySql(string $sql, array $attributes = [], Connection $connection = NULL): int|bool|array|string|null
|
||||
{
|
||||
$connection = static::getDefaultConnection($connection);
|
||||
|
||||
return $connection->createCommand($sql, $attributes)->all();
|
||||
}
|
||||
|
||||
/**
|
||||
* @param string $sql
|
||||
* @param array $attributes
|
||||
* @param Connection|NULL $connection
|
||||
* @return string|array|bool|int|null
|
||||
* @throws Exception
|
||||
*/
|
||||
public static function findBySql(string $sql, array $attributes = [], Connection $connection = NULL): string|array|bool|int|null
|
||||
{
|
||||
$connection = static::getDefaultConnection($connection);
|
||||
|
||||
return $connection->createCommand($sql, $attributes)->one();
|
||||
}
|
||||
|
||||
/**
|
||||
* @param string $field
|
||||
* @return array|null
|
||||
* @throws Exception
|
||||
*/
|
||||
public function values(string $field): ?array
|
||||
{
|
||||
$data = $this->get();
|
||||
if (empty($data) || empty($field)) {
|
||||
return NULL;
|
||||
}
|
||||
$first = current($data);
|
||||
if (!isset($first[$field])) {
|
||||
return NULL;
|
||||
}
|
||||
return array_column($data, $field);
|
||||
}
|
||||
|
||||
/**
|
||||
* @param $field
|
||||
* @return mixed
|
||||
* @throws Exception
|
||||
*/
|
||||
public function value($field): mixed
|
||||
{
|
||||
$data = $this->find();
|
||||
if (!empty($field) && isset($data[$field])) {
|
||||
return $data[$field];
|
||||
}
|
||||
return $data;
|
||||
}
|
||||
|
||||
/**
|
||||
* @param Connection|null $connection
|
||||
* @return bool|int
|
||||
* @throws ConfigException
|
||||
* @throws Exception
|
||||
*/
|
||||
public function delete(?Connection $connection = null): bool|int
|
||||
{
|
||||
$connection = static::getDefaultConnection($connection);
|
||||
|
||||
return $connection->createCommand($connection->getBuild()->builder($this))->delete();
|
||||
}
|
||||
|
||||
/**
|
||||
* @param string $table
|
||||
* @param null $connection
|
||||
* @return bool|int
|
||||
* @throws ConfigException
|
||||
* @throws Exception
|
||||
*/
|
||||
public static function drop(string $table, $connection = null): bool|int
|
||||
{
|
||||
$connection = static::getDefaultConnection($connection);
|
||||
|
||||
$sprint = sprintf('DROP TABLE' . ' `%s`.`%s`', $connection->database, $table);
|
||||
return $connection->createCommand($sprint)->delete();
|
||||
}
|
||||
|
||||
/**
|
||||
* @param string $table
|
||||
* @param null $connection
|
||||
* @return bool|int
|
||||
* @throws Exception
|
||||
*/
|
||||
public static function truncate(string $table, $connection = null): bool|int
|
||||
{
|
||||
$connection = static::getDefaultConnection($connection);
|
||||
|
||||
$sprint = sprintf('TRUNCATE `%s`.`%s`', $connection->database, $table);
|
||||
return $connection->createCommand($sprint)->exec();
|
||||
}
|
||||
|
||||
/**
|
||||
* @param string $table
|
||||
* @param Connection|NULL $connection
|
||||
* @return array|bool|null
|
||||
* @throws ConfigException
|
||||
* @throws Exception
|
||||
*/
|
||||
public static function showCreateSql(string $table, Connection $connection = NULL): array|bool|null
|
||||
{
|
||||
$connection = static::getDefaultConnection($connection);
|
||||
|
||||
$sprint = sprintf('SHOW CREATE TABLE `%s`.`%s`', $connection->database, $table);
|
||||
return $connection->createCommand($sprint)->one();
|
||||
}
|
||||
|
||||
/**
|
||||
* @param string $table
|
||||
* @param Connection|NULL $connection
|
||||
* @return array|bool|null
|
||||
* @throws ConfigException
|
||||
* @throws Exception
|
||||
*/
|
||||
public static function desc(string $table, Connection $connection = NULL): array|bool
|
||||
{
|
||||
$connection = static::getDefaultConnection($connection);
|
||||
|
||||
$sprint = sprintf('SHOW FULL FIELDS FROM `%s`.`%s`', $connection->database, $table);
|
||||
return $connection->createCommand($sprint)->all();
|
||||
}
|
||||
/**
|
||||
* @return void
|
||||
* @throws Exception
|
||||
*/
|
||||
public static function beginTransaction(): void
|
||||
{
|
||||
fire(new BeginTransaction());
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* @param string $table
|
||||
* @param Connection|NULL $connection
|
||||
* @return array|bool|null
|
||||
* @throws Exception
|
||||
*/
|
||||
public static function show(string $table, Connection $connection = NULL): array|bool|null
|
||||
{
|
||||
if ($table == '') {
|
||||
return null;
|
||||
}
|
||||
$connection = static::getDefaultConnection($connection);
|
||||
|
||||
$table = [' const TABLE = \'select * from %s where REFERENCED_TABLE_NAME=%s\';'];
|
||||
return $connection->createCommand((new Query())
|
||||
->select('*')
|
||||
->from('INFORMATION_SCHEMA.KEY_COLUMN_USAGE')
|
||||
->where(['REFERENCED_TABLE_NAME' => $table])
|
||||
->getSql())->one();
|
||||
}
|
||||
/**
|
||||
* @param Closure $closure
|
||||
* @param mixed ...$params
|
||||
* @return mixed
|
||||
* @throws ContainerExceptionInterface
|
||||
* @throws NotFoundExceptionInterface
|
||||
* @throws Exception
|
||||
*/
|
||||
public static function Transaction(Closure $closure, ...$params): mixed
|
||||
{
|
||||
static::beginTransaction();
|
||||
try {
|
||||
$result = call_user_func($closure, ...$params);
|
||||
} catch (Throwable $throwable) {
|
||||
error($throwable);
|
||||
$result = addError($throwable->getMessage(), 'mysql');
|
||||
} finally {
|
||||
if ($result === false) {
|
||||
static::rollback();
|
||||
} else {
|
||||
static::commit();
|
||||
}
|
||||
return $result;
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* @param null|Connection $connection
|
||||
* @param string $name
|
||||
* @return mixed
|
||||
* @throws Exception
|
||||
*/
|
||||
public static function getDefaultConnection(?Connection $connection, string $name = 'db'): Connection
|
||||
{
|
||||
if ($connection instanceof Connection) {
|
||||
return $connection;
|
||||
}
|
||||
$databases = \config('databases.connections', []);
|
||||
if (empty($databases) || !is_array($databases)) {
|
||||
throw new Exception('Please configure the database link.');
|
||||
}
|
||||
return \Kiri::service()->get($databases[$name]);
|
||||
}
|
||||
/**
|
||||
* @return void
|
||||
*/
|
||||
public static function commit(): void
|
||||
{
|
||||
fire(new Commit());
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* @return void
|
||||
*/
|
||||
public static function rollback(): void
|
||||
{
|
||||
fire(new Rollback());
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* @param $table
|
||||
*
|
||||
* @return static
|
||||
*/
|
||||
public static function table($table): Db|static
|
||||
{
|
||||
$connection = new Db();
|
||||
$connection->connection = current(\config('databases.connections'));
|
||||
$connection->from($table);
|
||||
return $connection;
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* @param string $column
|
||||
* @param string $alias
|
||||
* @return string
|
||||
*/
|
||||
public static function any_value(string $column, string $alias = ''): string
|
||||
{
|
||||
if (empty($alias)) {
|
||||
$alias = $column . '_any_value';
|
||||
}
|
||||
return 'ANY_VALUE(' . $column . ') as ' . $alias;
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* @return array|bool
|
||||
* @throws Exception
|
||||
*/
|
||||
public function get(): array|bool
|
||||
{
|
||||
return $this->connection->createCommand(SqlBuilder::builder($this)->all())->all();
|
||||
}
|
||||
|
||||
/**
|
||||
* @return array|bool|null
|
||||
* @throws Exception
|
||||
*/
|
||||
public function first(): array|bool|null
|
||||
{
|
||||
return $this->connection->createCommand(SqlBuilder::builder($this)->all())->one();
|
||||
}
|
||||
|
||||
/**
|
||||
* @return bool|int
|
||||
* @throws Exception
|
||||
*/
|
||||
public function count(): bool|int
|
||||
{
|
||||
return $this->connection->createCommand(SqlBuilder::builder($this)->count())->one()['row_count'];
|
||||
}
|
||||
|
||||
/**
|
||||
* @return bool|int
|
||||
* @throws Exception
|
||||
*/
|
||||
public function exists(): bool|int
|
||||
{
|
||||
return $this->connection->createCommand(SqlBuilder::builder($this)->one())->fetchColumn();
|
||||
}
|
||||
|
||||
/**
|
||||
* @param string $sql
|
||||
* @param array $attributes
|
||||
* @return array|bool|int|string|null
|
||||
* @throws Exception
|
||||
*/
|
||||
public function query(string $sql, array $attributes = []): int|bool|array|string|null
|
||||
{
|
||||
return $this->connection->createCommand($sql, $attributes)->all();
|
||||
}
|
||||
|
||||
/**
|
||||
* @param string $sql
|
||||
* @param array $attributes
|
||||
* @return array|bool|int|string|null
|
||||
* @throws Exception
|
||||
*/
|
||||
public function one(string $sql, array $attributes = []): int|bool|array|string|null
|
||||
{
|
||||
return $this->connection->createCommand($sql, $attributes)->one();
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* @return bool|int
|
||||
* @throws ConfigException
|
||||
* @throws Exception
|
||||
*/
|
||||
public function delete(): bool|int
|
||||
{
|
||||
return $this->connection->createCommand(SqlBuilder::builder($this)->delete())->delete();
|
||||
}
|
||||
|
||||
/**
|
||||
* @param string $table
|
||||
* @return array|bool|null
|
||||
* @throws Exception
|
||||
*/
|
||||
public static function show(string $table): array|bool|null
|
||||
{
|
||||
if ($table == '') {
|
||||
return null;
|
||||
}
|
||||
$connection = static::getDefaultConnection();
|
||||
|
||||
$table = [' const TABLE = \'select * from %s where REFERENCED_TABLE_NAME=%s\';'];
|
||||
return $connection->createCommand((new Query())
|
||||
->select('*')
|
||||
->from('INFORMATION_SCHEMA.KEY_COLUMN_USAGE')
|
||||
->where(['REFERENCED_TABLE_NAME' => $table])
|
||||
->getSql())->one();
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* @param null|Connection $connection
|
||||
* @param string $name
|
||||
* @return mixed
|
||||
* @throws Exception
|
||||
*/
|
||||
public static function getDefaultConnection(?Connection $connection = null, string $name = 'db'): Connection
|
||||
{
|
||||
if ($connection instanceof Connection) {
|
||||
return $connection;
|
||||
}
|
||||
$databases = \config('databases.connections', []);
|
||||
if (empty($databases) || !is_array($databases)) {
|
||||
throw new Exception('Please configure the database link.');
|
||||
}
|
||||
return Kiri::service()->get($databases[$name]);
|
||||
}
|
||||
|
||||
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user