diff --git a/convert.go b/convert.go deleted file mode 100644 index 30b19de..0000000 --- a/convert.go +++ /dev/null @@ -1,394 +0,0 @@ -// Package wrapper ... -// -// Description : 任意类型之间的相互转换 -// -// Author : go_developer@163.com<白茶清欢> -// -// Date : 2021-02-23 10:23 下午 -package wrapper - -/* - Desc : 在处理一些参数的时候，可能需要将参数转换为各种类型，这里实现一个通用的转换函数，实现各种类型之间的相互转换。 - - 当然，如果源数据格式和目标数据类型不一致，是会返回错误的。例如将字符串“一二三”转换为数值类型则会报错，而将字符串“123”转换为数值类型则OK。 - - 这段代码实际抄自go自带的“database/sql”库，只是源代码作为内部函数无法在外面调用，可以单独把需要的功能拎出来使用： - - 代码中有一个Scaner接口，可以自行实现，然后通过"convertAssign()"函数，作为dst参数传入。 - - Author : zhangdeman001@ke.com<白茶清欢> - -*/ -import ( - "errors" - "fmt" - "reflect" - "strconv" - "time" -) - -// RawBytes is a byte slice that holds a reference to memory owned by -// the database itself. After a Scan into a RawBytes, the slice is only -// valid until the next call to Next, Scan, or Close. -type RawBytes []byte - -var errNilPtr = errors.New("destination pointer is nil") // embedded in descriptive error - -// ConvertAssign ... -// convertAssign copies to dest the value in src, converting it if possible. -// An error is returned if the copy would result in loss of information. -// dest should be a pointer type. -func ConvertAssign(dest, src any) error { - // Common cases, without reflect. - switch s := src.(type) { - case string: - switch d := dest.(type) { - case *string: - if d == nil { - return errNilPtr - } - *d = s - return nil - case *[]byte: - if d == nil { - return errNilPtr - } - *d = []byte(s) - return nil - case *RawBytes: - if d == nil { - return errNilPtr - } - *d = append((*d)[:0], s...) - return nil - } - case []byte: - switch d := dest.(type) { - case *string: - if d == nil { - return errNilPtr - } - *d = string(s) - return nil - case *any: - if d == nil { - return errNilPtr - } - *d = cloneBytes(s) - return nil - case *[]byte: - if d == nil { - return errNilPtr - } - *d = cloneBytes(s) - return nil - case *RawBytes: - if d == nil { - return errNilPtr - } - *d = s - return nil - } - case time.Time: - switch d := dest.(type) { - case *time.Time: - *d = s - return nil - case *string: - *d = s.Format(time.RFC3339Nano) - return nil - case *[]byte: - if d == nil { - return errNilPtr - } - *d = []byte(s.Format(time.RFC3339Nano)) - return nil - case *RawBytes: - if d == nil { - return errNilPtr - } - *d = s.AppendFormat((*d)[:0], time.RFC3339Nano) - return nil - } - case nil: - switch d := dest.(type) { - case *any: - if d == nil { - return errNilPtr - } - *d = nil - return nil - case *[]byte: - if d == nil { - return errNilPtr - } - *d = nil - return nil - case *RawBytes: - if d == nil { - return errNilPtr - } - *d = nil - return nil - } - } - - var sv reflect.Value - - switch d := dest.(type) { - case *string: - sv = reflect.ValueOf(src) - switch sv.Kind() { - case reflect.Bool, - reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64, - reflect.Uint, reflect.Uint8, reflect.Uint16, reflect.Uint32, reflect.Uint64, - reflect.Float32, reflect.Float64: - *d = asString(src) - return nil - } - case *[]byte: - sv = reflect.ValueOf(src) - if b, ok := asBytes(nil, sv); ok { - *d = b - return nil - } - case *RawBytes: - sv = reflect.ValueOf(src) - if b, ok := asBytes([]byte(*d)[:0], sv); ok { - *d = RawBytes(b) - return nil - } - case *bool: - bv, err := Bool.ConvertValue(src) - if err == nil { - *d = bv.(bool) - } - return err - case *any: - *d = src - return nil - } - - if scanner, ok := dest.(Scanner); ok { - return scanner.Scan(src) - } - - dpv := reflect.ValueOf(dest) - if dpv.Kind() != reflect.Ptr { - return errors.New("destination not a pointer") - } - if dpv.IsNil() { - return errNilPtr - } - - if !sv.IsValid() { - sv = reflect.ValueOf(src) - } - - dv := reflect.Indirect(dpv) - if sv.IsValid() && sv.Type().AssignableTo(dv.Type()) { - switch b := src.(type) { - case []byte: - dv.Set(reflect.ValueOf(cloneBytes(b))) - default: - dv.Set(sv) - } - return nil - } - - if dv.Kind() == sv.Kind() && sv.Type().ConvertibleTo(dv.Type()) { - dv.Set(sv.Convert(dv.Type())) - return nil - } - - // The following conversions use a string value as an intermediate representation - // to convert between various numeric types. - // - // This also allows scanning into user defined types such as "type Int int64". - // For symmetry, also check for string destination types. - switch dv.Kind() { - case reflect.Ptr: - if src == nil { - dv.Set(reflect.Zero(dv.Type())) - return nil - } - dv.Set(reflect.New(dv.Type().Elem())) - return ConvertAssign(dv.Interface(), src) - case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64: - s := asString(src) - i64, err := strconv.ParseInt(s, 10, dv.Type().Bits()) - if err != nil { - err = strconvErr(err) - return fmt.Errorf("converting driver.Value type %T (%q) to a %s: %v", src, s, dv.Kind(), err) - } - dv.SetInt(i64) - return nil - case reflect.Uint, reflect.Uint8, reflect.Uint16, reflect.Uint32, reflect.Uint64: - s := asString(src) - u64, err := strconv.ParseUint(s, 10, dv.Type().Bits()) - if err != nil { - err = strconvErr(err) - return fmt.Errorf("converting driver.Value type %T (%q) to a %s: %v", src, s, dv.Kind(), err) - } - dv.SetUint(u64) - return nil - case reflect.Float32, reflect.Float64: - s := asString(src) - f64, err := strconv.ParseFloat(s, dv.Type().Bits()) - if err != nil { - err = strconvErr(err) - return fmt.Errorf("converting driver.Value type %T (%q) to a %s: %v", src, s, dv.Kind(), err) - } - dv.SetFloat(f64) - return nil - case reflect.String: - switch v := src.(type) { - case string: - dv.SetString(v) - return nil - case []byte: - dv.SetString(string(v)) - return nil - } - } - - return fmt.Errorf("unsupported Scan, storing driver.Value type %T into type %T", src, dest) -} - -func strconvErr(err error) error { - if ne, ok := err.(*strconv.NumError); ok { - return ne.Err - } - return err -} - -func cloneBytes(b []byte) []byte { - if b == nil { - return nil - } - c := make([]byte, len(b)) - copy(c, b) - return c -} - -func ToString(src any) string { - return asString(src) -} - -func asString(src any) string { - switch v := src.(type) { - case string: - return v - case []byte: - return string(v) - } - rv := reflect.ValueOf(src) - switch rv.Kind() { - case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64: - return strconv.FormatInt(rv.Int(), 10) - case reflect.Uint, reflect.Uint8, reflect.Uint16, reflect.Uint32, reflect.Uint64: - return strconv.FormatUint(rv.Uint(), 10) - case reflect.Float64: - return strconv.FormatFloat(rv.Float(), 'g', -1, 64) - case reflect.Float32: - return strconv.FormatFloat(rv.Float(), 'g', -1, 32) - case reflect.Bool: - return strconv.FormatBool(rv.Bool()) - } - return fmt.Sprintf("%v", src) -} - -func asBytes(buf []byte, rv reflect.Value) (b []byte, ok bool) { - switch rv.Kind() { - case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64: - return strconv.AppendInt(buf, rv.Int(), 10), true - case reflect.Uint, reflect.Uint8, reflect.Uint16, reflect.Uint32, reflect.Uint64: - return strconv.AppendUint(buf, rv.Uint(), 10), true - case reflect.Float32: - return strconv.AppendFloat(buf, rv.Float(), 'g', -1, 32), true - case reflect.Float64: - return strconv.AppendFloat(buf, rv.Float(), 'g', -1, 64), true - case reflect.Bool: - return strconv.AppendBool(buf, rv.Bool()), true - case reflect.String: - s := rv.String() - return append(buf, s...), true - } - return -} - -// Value is a value that drivers must be able to handle. -// It is either nil, a type handled by a database driver's NamedValueChecker -// interface, or an instance of one of these types: -// -// int64 -// float64 -// bool -// []byte -// string -// time.Time -type Value any - -type boolType struct{} - -var Bool boolType - -func (boolType) String() string { return "Bool" } -func (boolType) ConvertValue(src any) (Value, error) { - switch s := src.(type) { - case bool: - return s, nil - case string: - b, err := strconv.ParseBool(s) - if err != nil { - return nil, fmt.Errorf("sql/driver: couldn't convert %q into type bool", s) - } - return b, nil - case []byte: - b, err := strconv.ParseBool(string(s)) - if err != nil { - return nil, fmt.Errorf("sql/driver: couldn't convert %q into type bool", s) - } - return b, nil - } - - sv := reflect.ValueOf(src) - switch sv.Kind() { - case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64: - iv := sv.Int() - if iv == 1 || iv == 0 { - return iv == 1, nil - } - return nil, fmt.Errorf("sql/driver: couldn't convert %d into type bool", iv) - case reflect.Uint, reflect.Uint8, reflect.Uint16, reflect.Uint32, reflect.Uint64: - uv := sv.Uint() - if uv == 1 || uv == 0 { - return uv == 1, nil - } - return nil, fmt.Errorf("sql/driver: couldn't convert %d into type bool", uv) - } - - return nil, fmt.Errorf("sql/driver: couldn't convert %v (%T) into type bool", src, src) -} - -type Scanner interface { - // Scan assigns a value from a database driver. - // - // The src value will be of one of the following types: - // - // int64 - // float64 - // bool - // []byte - // string - // time.Time - // nil - for NULL values - // - // An error should be returned if the value cannot be stored - // without loss of information. - // - // Reference types such as []byte are only valid until the next call to Scan - // and should not be retained. Their underlying memory is owned by the driver. - // If retention is necessary, copy their values before the next call to Scan. - Scan(src any) error -} diff --git a/time.go b/op_time/time.go similarity index 61% rename from time.go rename to op_time/time.go index df2938c..28bbed4 100644 --- a/time.go +++ b/op_time/time.go @@ -1,11 +1,11 @@ -// Package wrapper ... +// Package op_time ... // // Description : wrapper ... // // Author : go_developer@163.com<白茶清欢> // // Date : 2023-08-09 18:22 -package wrapper +package op_time import ( "fmt" @@ -13,17 +13,11 @@ import ( ) const ( - // StandTimeFormat 标准时间格式 - StandTimeFormat = "2006-01-02 15:04:05" // StandISO8601Time 标准ISO时间格式 StandISO8601Time = "2006-01-02T15:04:05+08:00" ) // OwnTime ... -// -// Author : go_developer@163.com<白茶清欢> -// -// Date : 18:28 2023/8/9 func OwnTime(inputTime time.Time) *Time { instance := &Time{} instance.Time = inputTime @@ -31,59 +25,35 @@ func OwnTime(inputTime time.Time) *Time { } // OwnTimeFromNow 从当前时间获取实例 -// -// Author : go_developer@163.com<白茶清欢> -// -// Date : 12:44 2023/8/10 func OwnTimeFromNow() *Time { return OwnTime(time.Now()) } // OwnTimeFromSecond 从秒获取实例 -// -// Author : go_developer@163.com<白茶清欢> -// -// Date : 18:47 2023/8/9 func OwnTimeFromSecond(second int64) *Time { t := time.UnixMilli(second * 1000) return OwnTime(t) } // OwnTimeFromMilli 从ms获取实例 -// -// Author : go_developer@163.com<白茶清欢> -// -// Date : 18:56 2023/8/9 func OwnTimeFromMilli(milli int64) *Time { t := time.UnixMilli(milli) return OwnTime(t) } // OwnTimeFromMicro 从微秒获取实例 -// -// Author : go_developer@163.com<白茶清欢> -// -// Date : 19:00 2023/8/9 func OwnTimeFromMicro(micro int64) *Time { t := time.UnixMicro(micro) return OwnTime(t) } // OwnTimeFromNano 从纳秒获取实例 -// -// Author : go_developer@163.com<白茶清欢> -// -// Date : 19:00 2023/8/9 func OwnTimeFromNano(nano int64) *Time { t := time.Unix(nano/1e9, nano%1e9) return OwnTime(t) } // OwnTimeFromISO8601Time ... -// -// Author : go_developer@163.com<白茶清欢> -// -// Date : 11:02 2023/8/10 func OwnTimeFromISO8601Time(parseTime string, layout ...string) *Time { var ( err error @@ -98,66 +68,38 @@ func OwnTimeFromISO8601Time(parseTime string, layout ...string) *Time { } // Time 时间类型 -// -// Author : go_developer@163.com<白茶清欢> -// -// Date : 18:23 2023/8/9 type Time struct { time.Time } // GetCurrentFormatTime 获取当前时间的格式化时间(秒) -// -// Author : go_developer@163.com<白茶清欢> -// -// Date : 1:34 上午 2021/10/7 func (t *Time) GetCurrentFormatTime(layout ...string) string { - return time.Now().In(time.Local).Format(t.getTimeFormat(StandTimeFormat, layout...)) + return time.Now().In(time.Local).Format(t.getTimeFormat(time.DateTime, layout...)) } // FormatUnixNano 格式化纳秒时间戳 -// -// Author : go_developer@163.com<白茶清欢> -// -// Date : 11:54 2022/7/14 func (t *Time) FormatUnixNano(layout ...string) string { nano := t.UnixNano() % 1e6 return t.FormatUnixMilli(layout...) + fmt.Sprintf(" %v", nano) } // FormatUnixMilli 格式化毫秒时间戳 -// -// Author : go_developer@163.com<白茶清欢> -// -// Date : 11:55 2022/7/14 func (t *Time) FormatUnixMilli(layout ...string) string { return time.UnixMilli(t.UnixMilli()).In(time.Local). - Format(t.getTimeFormat(StandTimeFormat, layout...)) + fmt.Sprintf(".%v", t.UnixMilli()%1000) + Format(t.getTimeFormat(time.DateTime, layout...)) + fmt.Sprintf(".%v", t.UnixMilli()%1000) } // FormatUnixSec 格式化秒级时间戳 -// -// Author : go_developer@163.com<白茶清欢> -// -// Date : 12:06 2022/7/14 func (t *Time) FormatUnixSec(layout ...string) string { - return time.Unix(t.Unix(), 0).In(time.Local).Format(t.getTimeFormat(StandTimeFormat, layout...)) + return time.Unix(t.Unix(), 0).In(time.Local).Format(t.getTimeFormat(time.DateTime, layout...)) } // ToString 转字符串 -// -// Author : go_developer@163.com<白茶清欢> -// -// Date : 12:16 2023/8/10 func (t *Time) ToString(layout ...string) string { return t.FormatUnixSec(layout...) } // getTimeFormat 获取时间格式 -// -// Author : go_developer@163.com<白茶清欢> -// -// Date : 18:37 2023/8/9 func (t *Time) getTimeFormat(defaultFormat string, layout ...string) string { if len(layout) > 0 && len(layout[0]) > 0 { return layout[0]