处理 Azure SQL 错误

问题描述

我有一个用于处理 TSQL 查询的函数。本质上，它尝试创建到服务器的连接、运行查询、提交然后关闭连接。目前，我们依赖 mssqldb driver，因为后端是 Azure SQL。

func (requester *Requester) doTransaction(ctx context.Context,isolation sql.IsolationLevel,txFunc func(*sql.Tx) error) error {

    // First,attempt to get a connection from the connection pool. If this fails return an error
    conn,err := requester.conn.Conn(ctx)
    if err != nil {
        fmt.Printf("Conn failed,Error type: %s\n",reflect.TypeOf(err))
        log.Printf("Conn failed,error: %v",err)
        return err
    }

    // Before we continue on,ensure that the connection is clsoed and returned to the connection pool
    defer func() {
        if err := conn.Close(); err != nil {
            log.Printf("Close failed,err)
        }
    }()

    // Next,start the transaction with the given context and the default isolation
    tx,err := conn.BeginTx(ctx,&sql.TxOptions{Isolation: isolation,ReadOnly: false})
    if err != nil {
        fmt.Printf("BeginTx failed,reflect.TypeOf(err))
        log.Printf("BeginTx failed,err)
        return err
    }

    // Now,ensure that the transaction is either rolled back or committed before
    // the function ends
    defer func() {
        if p := recover(); p != nil {
            tx.Rollback()
            panic(p)
        } else if err != nil {
            log.Printf("An error occurred: %v",err)
            if err := tx.Rollback(); err != nil {
                log.Printf("Rollback failed,err)
            }
        } else {
            if err := tx.Commit(); err != nil {
                log.Printf("Commit failed,err)
            }
        }
    }()

    // Finally,run the function and return the result
    err = txFunc(tx)
    return err
}

这在大多数情况下效果很好。但是，我注意到由于超时、无服务器暂停、超出 IO 限制等原因而发生的许多错误；例如：

Login error: mssql: Database 'my-db' on server 'myserver.database.windows.net' is not currently available. Please retry the connection later. If the problem persists,contact customer support,and provide them the session tracing ID of '{SOME_GUID}'.

我想通过某种退避来处理这些问题，而不仅仅是失败。但是，要做到这一点，我需要能够以某种方式解释错误。但是，返回的错误都具有 *errors.errorString 类型。我试过使用 As(error,interface{}) 来检查错误是否是 mssql.Error 而不是，所以我不确定如何处理它。我如何确定这些错误的原因是什么？

解决方法

您可以尝试使用类型断言将您遇到的错误转换为此处定义的错误 as defined here

err:=//your operation that returns an error
//check and convert the error to a MSSQL error with more context
if msSQLErr,ok:=err.(mssql.Error);ok{
   if msSQLErr.SQLErrorNumber() == someRetryableErrorCode{
      //custom retry logic...
   }
}

或者，您可以使用错误。如下

var msSQLErr mssql.Error
if errors.As(err,&msSQLErr){
   if msSQLErr.SQLErrorNumber() == someRetryableErrorCode{
      //custom retry logic...
   }
}

更新：不幸的是，该库似乎没有包装底层错误，因此您不能使用 errors.Is 或 errors。至于某些错误，例如与网络相关的错误，因此您必须使用 strings.Contains 之类的东西自己匹配错误字符串