123 lines
2.8 KiB
Go
123 lines
2.8 KiB
Go
package cassandra
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
)
|
|
|
|
// 定義統一的錯誤類型
|
|
var (
|
|
// ErrNotFound 表示記錄未找到
|
|
ErrNotFound = &Error{
|
|
Code: "NOT_FOUND",
|
|
Message: "record not found",
|
|
}
|
|
|
|
// ErrAcquireLockFailed 表示獲取鎖失敗
|
|
ErrAcquireLockFailed = &Error{
|
|
Code: "LOCK_ACQUIRE_FAILED",
|
|
Message: "acquire lock failed",
|
|
}
|
|
|
|
// ErrInvalidInput 表示輸入參數無效
|
|
ErrInvalidInput = &Error{
|
|
Code: "INVALID_INPUT",
|
|
Message: "invalid input parameter",
|
|
}
|
|
|
|
// ErrNoPartitionKey 表示缺少 Partition Key
|
|
ErrNoPartitionKey = &Error{
|
|
Code: "NO_PARTITION_KEY",
|
|
Message: "no partition key defined in struct",
|
|
}
|
|
|
|
// ErrMissingTableName 表示缺少 TableName 方法
|
|
ErrMissingTableName = &Error{
|
|
Code: "MISSING_TABLE_NAME",
|
|
Message: "struct must implement TableName() method",
|
|
}
|
|
|
|
// ErrNoFieldsToUpdate 表示沒有欄位需要更新
|
|
ErrNoFieldsToUpdate = &Error{
|
|
Code: "NO_FIELDS_TO_UPDATE",
|
|
Message: "no fields to update",
|
|
}
|
|
|
|
// ErrMissingWhereCondition 表示缺少 WHERE 條件
|
|
ErrMissingWhereCondition = &Error{
|
|
Code: "MISSING_WHERE_CONDITION",
|
|
Message: "operation requires at least one WHERE condition for safety",
|
|
}
|
|
|
|
// ErrMissingPartitionKey 表示 WHERE 條件中缺少 Partition Key
|
|
ErrMissingPartitionKey = &Error{
|
|
Code: "MISSING_PARTITION_KEY",
|
|
Message: "operation requires all partition keys in WHERE clause",
|
|
}
|
|
)
|
|
|
|
// Error 是統一的錯誤類型
|
|
type Error struct {
|
|
Code string // 錯誤代碼
|
|
Message string // 錯誤訊息
|
|
Table string // 相關的表名（可選）
|
|
Err error // 底層錯誤（可選）
|
|
}
|
|
|
|
// Error 實現 error 介面
|
|
func (e *Error) Error() string {
|
|
if e.Table != "" {
|
|
if e.Err != nil {
|
|
return fmt.Sprintf("cassandra [%s] (table: %s): %s: %v", e.Code, e.Table, e.Message, e.Err)
|
|
}
|
|
return fmt.Sprintf("cassandra [%s] (table: %s): %s", e.Code, e.Table, e.Message)
|
|
}
|
|
if e.Err != nil {
|
|
return fmt.Sprintf("cassandra [%s]: %s: %v", e.Code, e.Message, e.Err)
|
|
}
|
|
return fmt.Sprintf("cassandra [%s]: %s", e.Code, e.Message)
|
|
}
|
|
|
|
// Unwrap 返回底層錯誤
|
|
func (e *Error) Unwrap() error {
|
|
return e.Err
|
|
}
|
|
|
|
// WithTable 為錯誤添加表名資訊
|
|
func (e *Error) WithTable(table string) *Error {
|
|
return &Error{
|
|
Code: e.Code,
|
|
Message: e.Message,
|
|
Table: table,
|
|
Err: e.Err,
|
|
}
|
|
}
|
|
|
|
// WithError 為錯誤添加底層錯誤
|
|
func (e *Error) WithError(err error) *Error {
|
|
return &Error{
|
|
Code: e.Code,
|
|
Message: e.Message,
|
|
Table: e.Table,
|
|
Err: err,
|
|
}
|
|
}
|
|
|
|
// NewError 創建新的錯誤
|
|
func NewError(code, message string) *Error {
|
|
return &Error{
|
|
Code: code,
|
|
Message: message,
|
|
}
|
|
}
|
|
|
|
// IsNotFound 檢查錯誤是否為 NotFound
|
|
func IsNotFound(err error) bool {
|
|
return errors.Is(err, ErrNotFound)
|
|
}
|
|
|
|
// IsLockFailed 檢查錯誤是否為獲取鎖失敗
|
|
func IsLockFailed(err error) bool {
|
|
return errors.Is(err, ErrAcquireLockFailed)
|
|
}
|