일반 프레임워크를 사용하여 Go에서 강력한 SQL 트랜잭션 실행 구축

Go에서 SQL 데이터베이스로 작업할 때 다단계 트랜잭션 중에 원자성을 보장하고 롤백을 관리하는 것이 어려울 수 있습니다. 이 기사에서는 유연성을 위해 제네릭을 사용하여 Go에서 SQL 트랜잭션을 실행하기 위한 강력하고 재사용 가능하며 테스트 가능한 프레임워크를 만드는 과정을 안내하겠습니다.

트랜잭션 내에서 여러 종속 데이터베이스 작업을 실행하기 위한 SqlWriteExec 유틸리티를 구축하겠습니다. 이는 상태 비저장 및 상태 저장 작업을 모두 지원하여 종속성을 원활하게 관리하면서 관련 엔터티 삽입과 같은 정교한 워크플로를 가능하게 합니다.

SQL 트랜잭션을 위한 프레임워크가 필요한 이유는 무엇입니까?

실제 애플리케이션에서는 데이터베이스 작업이 격리되는 경우가 거의 없습니다. 다음 시나리오를 고려하십시오.

사용자를 삽입하고 인벤토리를 원자적으로 업데이트합니다.
일관성을 보장하면서 주문을 생성하고 결제를 처리합니다.
여러 단계가 관련되어 있으므로 장애 발생 시 롤백을 관리하는 것이 데이터 무결성을 보장하는 데 중요합니다.

Txn 관리에서 go를 사용하여 작업합니다.

데이터베이스 txn을 작성하는 경우 핵심 로직을 작성하기 전에 고려해야 할 몇 가지 상용구가 있을 수 있습니다. 이 txn 관리는 Java의 스프링 부트에 의해 관리되며 Java에서 코드를 작성하는 동안에는 그다지 신경 쓰지 않았지만 golang에서는 그렇지 않습니다. 아래에 간단한 예가 나와 있습니다

func basicTxn(db *sql.DB) error {
    // start a transaction
    tx, err := db.Begin()
    if err != nil {
        return err
    }
    defer func() {
        if r := recover(); r != nil {
            tx.Rollback()
        } else if err != nil {
            tx.Rollback()
        } else {
            tx.Commit()
        }
    }()

    // insert data into the orders table
    _, err = tx.Exec("INSERT INTO orders (id, customer_name, order_date) VALUES (1, 'John Doe', '2022-01-01')")
    if err != nil {
        return err
    }
    return nil
}

모든 기능에 대해 롤백/커밋 코드를 반복할 것으로 기대할 수는 없습니다. 여기에는 연기에서 실행될 때 txn을 커밋/롤백하는 반환 유형으로 함수를 제공하는 클래스를 생성하거나 모든 txn 기능을 함께 래핑하고 한 번에 실행하는 래퍼 클래스를 생성하는 두 가지 옵션이 있습니다.

나중에 선택했는데 아래에서 코드 변경 사항을 확인할 수 있습니다.

func TestSqlWriteExec_CreateOrderTxn(t *testing.T) {

    db := setupDatabase()
    // create a new SQL Write Executor
    err := dbutils.NewSqlTxnExec[OrderRequest, OrderProcessingResponse](context.TODO(), db, nil, &OrderRequest{CustomerName: "CustomerA", ProductID: 1, Quantity: 10}).
        StatefulExec(InsertOrder).
        StatefulExec(UpdateInventory).
        StatefulExec(InsertShipment).
        Commit()
    // check if the transaction was committed successfully
    if err != nil {
        t.Fatal(err)
        return
    }
    verifyTransactionSuccessful(t, db)
    t.Cleanup(
        func() { 
            cleanup(db)
            db.Close() 
        },
    )
}

func InsertOrder(ctx context.Context, txn *sql.Tx, order *OrderRequest, orderProcessing *OrderProcessingResponse) error {
    // Insert Order
    result, err := txn.Exec("INSERT INTO orders (customer_name, product_id, quantity) VALUES (, , )", order.CustomerName, order.ProductID, order.Quantity)
    if err != nil {
        return err
    }
    // Get the inserted Order ID
    orderProcessing.OrderID, err = result.LastInsertId()
    return err
}

func UpdateInventory(ctx context.Context, txn *sql.Tx, order *OrderRequest, orderProcessing *OrderProcessingResponse) error {
    // Update Inventory if it exists and the quantity is greater than the quantity check if it exists
    result, err := txn.Exec("UPDATE inventory SET product_quantity = product_quantity -  WHERE id =  AND product_quantity >= ", order.Quantity, order.ProductID)
    if err != nil {
        return err
    }
    // Get the number of rows affected
    rowsAffected, err := result.RowsAffected()
    if rowsAffected == 0 {
        return errors.New("Insufficient inventory")
    }
    return err
}

func InsertShipment(ctx context.Context, txn *sql.Tx, order *OrderRequest, orderProcessing *OrderProcessingResponse) error {
    // Insert Shipment
    result, err := txn.Exec("INSERT INTO shipping_info (customer_name, shipping_address) VALUES (, 'Shipping Address')", order.CustomerName)
    if err != nil {
        return err
    }
    // Get the inserted Shipping ID
    orderProcessing.ShippingID, err = result.LastInsertId()
    return err
}

이 코드는 훨씬 더 정확하고 간결해졌습니다.

핵심 로직이 구현되는 방식

여러 txns를 허용할 수 있도록 txn을 단일 go 구조체로 분리하는 것이 아이디어입니다. txn이란 클래스를 위해 생성한 txn으로 작업을 수행하는 함수를 의미합니다.

type TxnFn[T any] func(ctx context.Context, txn *sql.Tx, processingReq *T) error
type StatefulTxnFn[T any, R any] func(ctx context.Context, txn *sql.Tx, processingReq *T, processedRes *R) error

이 두 가지는 무언가를 처리하기 위해 txn을 사용하는 함수 유형입니다. 이제 데이터 계층에서 이와 같은 함수를 생성하고 이를 인수 주입 및 함수 실행을 담당하는 실행자 클래스에 전달합니다.

// SQL Write Executor is responsible when executing write operations
// For dependent writes you may need to add the dependent data to processReq and proceed to the next function call
type SqlTxnExec[T any, R any] struct {
    db               *sql.DB
    txn              *sql.Tx
    txnFns         []TxnFn[T]
    statefulTxnFns []StatefulTxnFn[T, R]
    processingReq    *T
    processedRes     *R
    ctx              context.Context
    err              error
}

여기에는 모든 txn_fn 세부 정보가 저장되며 txn 커밋을 시도하는 Commit() 메서드가 있습니다.

func (s *SqlTxnExec[T, R]) Commit() (err error) {
    defer func() {
        if p := recover(); p != nil {
            s.txn.Rollback()
            panic(p)
        } else if err != nil {
            err = errors.Join(err, s.txn.Rollback())
        } else {
            err = errors.Join(err, s.txn.Commit())
        }
        return
    }()

    for _, writeFn := range s.txnFns {
        if err = writeFn(s.ctx, s.txn, s.processingReq); err != nil {
            return
        }
    }

    for _, statefulWriteFn := range s.statefulTxnFns {
        if err = statefulWriteFn(s.ctx, s.txn, s.processingReq, s.processedRes); err != nil {
            return
        }
    }
    return
}

저장소에서 더 많은 예제와 테스트를 찾을 수 있습니다.
https://github.com/mahadev-k/go-utils/tree/main/examples

요즘 우리는 분산 시스템과 합의 프로토콜을 선호하지만 여전히 SQL을 사용하고 있으며 여전히 존재합니다.

이것을 기반으로 기여하고 싶은 사람이 있으면 알려주세요!!
여기까지 읽어주셔서 감사합니다!!
https://in.linkedin.com/in/mahadev-k-934520223
https://x.com/mahadev_k_

위 내용은 일반 프레임워크를 사용하여 Go에서 강력한 SQL 트랜잭션 실행 구축의 상세 내용입니다. 자세한 내용은 PHP 중국어 웹사이트의 기타 관련 기사를 참조하세요!