mirror of
https://github.com/lwch/natpass
synced 2025-10-28 06:21:26 +08:00
24 lines
377 B
Go
24 lines
377 B
Go
package utils
|
|
|
|
import "github.com/dustin/go-humanize"
|
|
|
|
type Bytes uint64
|
|
|
|
func (bt *Bytes) UnmarshalYAML(unmarshal func(interface{}) error) error {
|
|
var str string
|
|
err := unmarshal(&str)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
n, err := humanize.ParseBytes(str)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
*bt = Bytes(n)
|
|
return nil
|
|
}
|
|
|
|
func (bt *Bytes) Bytes() uint64 {
|
|
return uint64(*bt)
|
|
}
|