wrapper/bigint.go

// Package wrapper ...
//
// Description : wrapper ...
//
// Author : go_developer@163.com<白茶清欢>
//
// Date : 2024-11-19 16:33
package wrapper

import (
	"database/sql/driver"
	"fmt"
	"reflect"
	"strconv"
)

// BigInt 是一个自定义类型，用于在JSON编码时将int64转换为字符串。
// 建议只在Output中使用它，Input时直接使用int64。
type BigInt string

var EmptyBigInt BigInt = "0"

func (f *BigInt) IsNil() bool {
	if nil == f {
		return true
	}
	return reflect.ValueOf(*f).IsNil()
}

// MarshalJSON 自定义的JSON编码逻辑。
// @implements json.Marshaler
func (f *BigInt) MarshalJSON() ([]byte, error) {
	return []byte(fmt.Sprintf(`"%d"`, f.Int64())), nil
}

// ToString 返回BigId的字符串表示。
func (f *BigInt) String() string {
	if f.IsNil() {
		return string(EmptyBigInt)
	}
	return string(*f)
}

// Value 在orm写入数据库时会调用它进行格式转换.
// @implements database/sql/driver.Valuer
func (f *BigInt) Value() (driver.Value, error) {
	if f.IsNil() {
		return 0, nil
	}
	return f.Uint64(), nil
}

// Int64 返回BigId的int64表示。
// @implements github.com/gogf/gf/v2/util/gconv.iInt64
func (f *BigInt) Int64() int64 {
	if f.IsNil() {
		return 0
	}
	i, err := strconv.ParseInt(string(*f), 10, 64)
	if err != nil || i <= 0 {
		return 0
	}
	return i
}

// Uint64 返回BigId的uint64表示。
// @implements github.com/gogf/gf/v2/util/gconv.iUint64
func (f *BigInt) Uint64() uint64 {
	if f.IsNil() {
		return 0
	}
	return uint64(f.Int64())
}

// IsEmpty 判断BigId是否为空
func (f *BigInt) IsEmpty() bool {
	if f.IsNil() {
		return true
	}
	return *f == "" || *f == EmptyBigInt || f.Int64() == 0
}

// UnmarshalJSON converts a json byte array of a big ID into an BigInt type.
// @implements json.Unmarshaler
func (f *BigInt) UnmarshalJSON(b []byte) error {
	var (
		s = string(b)
	)
	// 如果是null，设置为0
	if s == "null" || s == "" || s == "0" {
		*f = EmptyBigInt
		return nil
	}
	// 如果是字符串，去掉引号
	if len(b) >= 3 && b[0] == '"' && b[len(b)-1] == '"' {
		s = string(b[1 : len(b)-1])
	}

	*f = StrToBigInt(s)
	return nil
}

// ToBigInt 将int转换为BigInt
func ToBigInt(id int64) BigInt {
	if id <= 0 {
		return EmptyBigInt
	}
	return BigInt(fmt.Sprintf("%d", id))
}

// StrToBigInt 将str转换为BigInt类型
func StrToBigInt(id string) BigInt {
	i, err := strconv.ParseInt(id, 10, 64)
	if err != nil || i <= 0 {
		return EmptyBigInt
	}
	return BigInt(id)
}