Add CopyFrom to transaction

Author: aopoltorzhicky

pkg/storage/postgres/transaction.go

@@ -2,6 +2,7 @@ package postgres
 
 import (
 	"context"
+	"io"
 
 	"github.com/go-pg/pg/v10"
 	"github.com/pkg/errors"
@@ -105,3 +106,13 @@ func (t *Transaction) Exec(ctx context.Context, query any, params ...any) (int,
 	}
 	return result.RowsAffected(), nil
 }
+
+// CopyFrom -
+func (t *Transaction) CopyFrom(r io.Reader, query string, args ...any) error {
+	if t.tx == nil {
+		return errNilTx
+	}
+
+	_, err := t.tx.CopyFrom(r, query, args...)
+	return err
+}

pkg/storage/storage.go

@@ -2,6 +2,7 @@ package storage
 
 import (
 	"context"
+	"io"
 )
 
 // SortOrder - asc or desc
@@ -72,6 +73,7 @@ type Transaction interface {
 	Close(ctx context.Context) error
 	HandleError(ctx context.Context, err error) error
 	Exec(ctx context.Context, query any, params ...any) (int, error)
+	CopyFrom(r io.Reader, query string, args ...any) error
 }
 
 // Model - general data type interface