Forbid cloning and serialization

amphp · Dec 17, 2023 · 8917498 · 8917498
1 parent de27f1c
commit 8917498
Show file tree

Hide file tree

Showing 9 changed files with 45 additions and 0 deletions.
diff --git a/src/CommandResult.php b/src/CommandResult.php
@@ -2,6 +2,8 @@
 
 namespace Amp\Sql\Common;
 
+use Amp\ForbidCloning;
+use Amp\ForbidSerialization;
 use Amp\Future;
 use Amp\Sql\Result;
 
@@ -13,6 +15,9 @@
  */
 abstract class CommandResult implements Result, \IteratorAggregate
 {
+    use ForbidCloning;
+    use ForbidSerialization;
+
     /**
      * @param Future<TResult|null> $nextResult
      */

diff --git a/src/ConnectionPool.php b/src/ConnectionPool.php
@@ -3,6 +3,8 @@
 namespace Amp\Sql\Common;
 
 use Amp\DeferredFuture;
+use Amp\ForbidCloning;
+use Amp\ForbidSerialization;
 use Amp\Future;
 use Amp\Sql\Connection;
 use Amp\Sql\Link;
@@ -29,6 +31,9 @@
  */
 abstract class ConnectionPool implements Pool
 {
+    use ForbidCloning;
+    use ForbidSerialization;
+
     public const DEFAULT_MAX_CONNECTIONS = 100;
     public const DEFAULT_IDLE_TIMEOUT = 60;
 

diff --git a/src/ConnectionTransaction.php b/src/ConnectionTransaction.php
@@ -3,6 +3,8 @@
 namespace Amp\Sql\Common;
 
 use Amp\DeferredFuture;
+use Amp\ForbidCloning;
+use Amp\ForbidSerialization;
 use Amp\Sql\Result;
 use Amp\Sql\SqlException;
 use Amp\Sql\Statement;
@@ -21,6 +23,9 @@
  */
 abstract class ConnectionTransaction implements Transaction
 {
+    use ForbidCloning;
+    use ForbidSerialization;
+
     /** @var \Closure():void */
     private readonly \Closure $release;
 

diff --git a/src/NestedTransaction.php b/src/NestedTransaction.php
@@ -3,6 +3,8 @@
 namespace Amp\Sql\Common;
 
 use Amp\DeferredFuture;
+use Amp\ForbidCloning;
+use Amp\ForbidSerialization;
 use Amp\Sql\Result;
 use Amp\Sql\SqlException;
 use Amp\Sql\Statement;
@@ -21,6 +23,9 @@
  */
 abstract class NestedTransaction implements Transaction
 {
+    use ForbidCloning;
+    use ForbidSerialization;
+
     /** @var \Closure():void */
     private readonly \Closure $release;
 

diff --git a/src/PooledResult.php b/src/PooledResult.php
@@ -2,6 +2,8 @@
 
 namespace Amp\Sql\Common;
 
+use Amp\ForbidCloning;
+use Amp\ForbidSerialization;
 use Amp\Future;
 use Amp\Sql\Result;
 use Revolt\EventLoop;
@@ -15,6 +17,9 @@
  */
 abstract class PooledResult implements Result, \IteratorAggregate
 {
+    use ForbidCloning;
+    use ForbidSerialization;
+
     /** @var Future<TResult|null>|null */
     private ?Future $next = null;
 

diff --git a/src/PooledStatement.php b/src/PooledStatement.php
@@ -2,6 +2,8 @@
 
 namespace Amp\Sql\Common;
 
+use Amp\ForbidCloning;
+use Amp\ForbidSerialization;
 use Amp\Sql\Result;
 use Amp\Sql\SqlException;
 use Amp\Sql\Statement;
@@ -15,6 +17,9 @@
  */
 abstract class PooledStatement implements Statement
 {
+    use ForbidCloning;
+    use ForbidSerialization;
+
     /** @var null|\Closure():void */
     private ?\Closure $release;
 

diff --git a/src/PooledTransaction.php b/src/PooledTransaction.php
@@ -2,6 +2,8 @@
 
 namespace Amp\Sql\Common;
 
+use Amp\ForbidCloning;
+use Amp\ForbidSerialization;
 use Amp\Sql\Result;
 use Amp\Sql\Statement;
 use Amp\Sql\Transaction;
@@ -17,6 +19,9 @@
  */
 abstract class PooledTransaction implements Transaction
 {
+    use ForbidCloning;
+    use ForbidSerialization;
+
     /** @var \Closure():void */
     private readonly \Closure $release;
 

diff --git a/src/RetrySqlConnector.php b/src/RetrySqlConnector.php
@@ -4,6 +4,8 @@
 
 use Amp\Cancellation;
 use Amp\CompositeException;
+use Amp\ForbidCloning;
+use Amp\ForbidSerialization;
 use Amp\Sql\Connection;
 use Amp\Sql\ConnectionException;
 use Amp\Sql\SqlConfig;
@@ -16,6 +18,9 @@
  */
 final class RetrySqlConnector implements SqlConnector
 {
+    use ForbidCloning;
+    use ForbidSerialization;
+
     /**
      * @param SqlConnector<TConfig, TConnection> $connector
      */

diff --git a/src/StatementPool.php b/src/StatementPool.php
@@ -3,6 +3,8 @@
 namespace Amp\Sql\Common;
 
 use Amp\DeferredFuture;
+use Amp\ForbidCloning;
+use Amp\ForbidSerialization;
 use Amp\Sql\Pool;
 use Amp\Sql\Result;
 use Amp\Sql\SqlConfig;
@@ -20,6 +22,9 @@
  */
 abstract class StatementPool implements Statement
 {
+    use ForbidCloning;
+    use ForbidSerialization;
+
     private readonly Pool $pool;
 
     /** @var \SplQueue<TStatement> */