PandaX/pkg/tool/conv.go

package tool

import (
	"encoding/json"
	"fmt"
	"pandax/pkg/global"
	"reflect"
	"strings"
	"time"
)

// SnakeString snake string, XxYy to xx_yy , XxYY to xx_y_y
func SnakeString(s string) string {
	data := make([]byte, 0, len(s)*2)
	j := false
	num := len(s)
	for i := 0; i < num; i++ {
		d := s[i]
		if i > 0 && d >= 'A' && d <= 'Z' && j {
			data = append(data, '_')
		}
		if d != '_' {
			j = true
		}
		data = append(data, d)
	}
	return strings.ToLower(string(data))
}

// CamelString camel string, xx_yy to XxYy
func CamelString(s string) string {
	data := make([]byte, 0, len(s))
	flag, num := true, len(s)-1
	for i := 0; i <= num; i++ {
		d := s[i]
		if d == '_' {
			flag = true
			continue
		} else if flag {
			if d >= 'a' && d <= 'z' {
				d = d - 32
			}
			flag = false
		}
		data = append(data, d)
	}
	return string(data)
}

func convertString(s string, firstLower bool) string {
	data := make([]byte, 0, len(s))
	flag, num := true, len(s)-1
	for i := 0; i <= num; i++ {
		d := s[i]
		if d == '_' {
			flag = true
			continue
		} else if flag {
			if d >= 'a' && d <= 'z' {
				d = d - 32
			}
			flag = false
		}
		data = append(data, d)
	}
	if firstLower && len(data) > 0 && data[0] >= 'A' && data[0] <= 'Z' {
		data[0] = data[0] + 32
	}
	return string(data)
}

// FirstLowCamelString first low camel string, xx_yy to xxYy
func FirstLowCamelString(s string) string {
	return convertString(s, true)
}

// StructToMap convert struct to map
func StructToMap(s interface{}) map[string]interface{} {
	result := make(map[string]interface{})

	value := reflect.ValueOf(s)
	typ := reflect.TypeOf(s)

	for i := 0; i < value.NumField(); i++ {
		field := typ.Field(i)
		fieldValue := value.Field(i).Interface()
		result[field.Name] = fieldValue
	}

	return result
}

// MapToStruct convert map to struct
func MapToStruct(m map[string]interface{}, s interface{}) error {
	data, err := json.Marshal(m)
	if err != nil {
		return err
	}

	err = json.Unmarshal(data, s)
	if err != nil {
		return err
	}

	return nil
}

// InterfaceToStruct convert interface to struct
func InterfaceToStruct(m interface{}, s interface{}) error {
	data, err := json.Marshal(m)
	if err != nil {
		return err
	}

	err = json.Unmarshal(data, s)
	if err != nil {
		return err
	}

	return nil
}

// StringToStruct convert string to struct
func StringToStruct(m string, s interface{}) error {
	err := json.Unmarshal([]byte(m), s)
	if err != nil {
		return err
	}
	return nil
}

// TimeToFormat convert time to formatted string
func TimeToFormat(val interface{}) string {
	switch v := val.(type) {
	case int:
		t := timestampToTime(int64(v))
		// 格式化时间字符串
		formattedTime := t.Format("2006-01-02 15:04:05")
		return formattedTime
	case int64:
		t := timestampToTime(v)
		// 格式化时间字符串
		formattedTime := t.Format("2006-01-02 15:04:05")
		return formattedTime
	case float64:
		t := timestampToTime(int64(v))
		// 格式化时间字符串
		formattedTime := t.Format("2006-01-02 15:04:05")
		return formattedTime
	case string:
		// 如果是字符串类型，将其解析为时间对象
		t, err := time.Parse("2006-01-02 15:04:05", v)
		if err != nil {
			global.Log.Error("时间格式非标准格式")
			return ""
		}
		// 格式化时间字符串
		formattedTime := t.Format("2006-01-02 15:04:05")
		return formattedTime
	default:
		return ""
	}
}

func timestampToTime(val int64) time.Time {
	// 如果是时间戳类型，将其转换为时间对象
	timeLen := len(fmt.Sprintf("%d", val))
	if timeLen == 10 {
		return time.Unix(val, 0)
	}
	if timeLen == 13 {
		return time.UnixMilli(val)
	}
	return time.Now()
}