1
0
mirror of https://github.com/duke-git/lancet.git synced 2026-02-17 03:02:28 +08:00

feat: add IsInt, IsFloat, IsNumber

This commit is contained in:
dudaodong
2023-04-05 14:07:23 +08:00
parent c01c9d14b4
commit 046f3e0bf9
5 changed files with 156 additions and 49 deletions

View File

@@ -5,11 +5,6 @@
package formatter package formatter
import ( import (
"fmt"
"reflect"
"strconv"
"strings"
"golang.org/x/exp/constraints" "golang.org/x/exp/constraints"
) )
@@ -18,54 +13,57 @@ import (
// Comma("12345", "$") => "$12,345", Comma(12345, "$") => "$12,345" // Comma("12345", "$") => "$12,345", Comma(12345, "$") => "$12,345"
// Play: https://go.dev/play/p/eRD5k2vzUVX // Play: https://go.dev/play/p/eRD5k2vzUVX
func Comma[T constraints.Float | constraints.Integer | string](value T, symbol string) string { func Comma[T constraints.Float | constraints.Integer | string](value T, symbol string) string {
s, err := numberToString(value) // s, err := numberToString(value)
if err != nil { // if err != nil {
return "" // return ""
} // }
dotIndex := strings.Index(s, ".") // dotIndex := strings.Index(s, ".")
if dotIndex != -1 { // if dotIndex != -1 {
return symbol + commaString(s[:dotIndex]) + s[dotIndex:] // return symbol + commaString(s[:dotIndex]) + s[dotIndex:]
} // }
// return symbol + commaString(s)
return ""
return symbol + commaString(s)
} }
func commaString(s string) string { // func commaString(s string) string {
if len(s) <= 3 { // if len(s) <= 3 {
return s // return s
} // }
return commaString(s[:len(s)-3]) + "," + commaString(s[len(s)-3:]) // return commaString(s[:len(s)-3]) + "," + commaString(s[len(s)-3:])
} // }
func numberToString(value any) (string, error) { // func numberToString(value any) (string, error) {
switch reflect.TypeOf(value).Kind() { // switch reflect.TypeOf(value).Kind() {
case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64, // case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64,
reflect.Uint, reflect.Uint8, reflect.Uint16, reflect.Uint32, reflect.Uint64: // reflect.Uint, reflect.Uint8, reflect.Uint16, reflect.Uint32, reflect.Uint64:
return fmt.Sprintf("%v", value), nil // return fmt.Sprintf("%v", value), nil
// todo: need to handle 12345678.9 => 1.23456789e+07 // // todo: need to handle 12345678.9 => 1.23456789e+07
case reflect.Float32, reflect.Float64: // case reflect.Float32, reflect.Float64:
return fmt.Sprintf("%v", value), nil // return fmt.Sprintf("%v", value), nil
case reflect.String: // case reflect.String:
{ // {
sv := fmt.Sprintf("%v", value) // sv := fmt.Sprintf("%v", value)
if strings.Contains(sv, ".") { // if strings.Contains(sv, ".") {
_, err := strconv.ParseFloat(sv, 64) // _, err := strconv.ParseFloat(sv, 64)
if err != nil { // if err != nil {
return "", err // return "", err
} // }
return sv, nil // return sv, nil
} else { // } else {
_, err := strconv.ParseInt(sv, 10, 64) // _, err := strconv.ParseInt(sv, 10, 64)
if err != nil { // if err != nil {
return "", nil // return "", nil
} // }
return sv, nil // return sv, nil
} // }
} // }
default: // default:
return "", nil // return "", nil
} // }
} // }

View File

@@ -16,6 +16,7 @@ func TestComma(t *testing.T) {
assert.Equal("12,345", Comma("12345", "")) assert.Equal("12,345", Comma("12345", ""))
assert.Equal("12,345.6789", Comma("12345.6789", "")) assert.Equal("12,345.6789", Comma("12345.6789", ""))
assert.Equal("123,456,789,000", Comma("123456789000", "")) assert.Equal("123,456,789,000", Comma("123456789000", ""))
assert.Equal("12,345,678.9", Comma("12345678.9", ""))
assert.Equal("12,345", Comma(12345, "")) assert.Equal("12,345", Comma(12345, ""))
assert.Equal("$12,345", Comma(12345, "$")) assert.Equal("$12,345", Comma(12345, "$"))

View File

@@ -369,3 +369,29 @@ func IsGBK(data []byte) bool {
return true return true
} }
// IsNumberStr check if the value is Number(integer, float) or not.
// Play: todo
func IsNumber(v any) bool {
return IsInt(v) || IsFloat(v)
}
// IsFloat check if the value is float(float32, float34) or not.
// Play: todo
func IsFloat(v any) bool {
switch v.(type) {
case float32, float64:
return true
}
return false
}
// IsInt check if the value is integer(int, unit) or not.
// Play: todo
func IsInt(v any) bool {
switch v.(type) {
case int, int8, int16, int32, int64, uint, uint8, uint16, uint32, uint64, uintptr:
return true
}
return false
}

View File

@@ -449,3 +449,57 @@ func ExampleIsPrintable() {
// true // true
// false // false
} }
func ExampleIsInt() {
result1 := IsInt("")
result2 := IsInt("3")
result3 := IsInt(0.1)
result4 := IsInt(0)
fmt.Println(result1)
fmt.Println(result2)
fmt.Println(result3)
fmt.Println(result4)
// Output:
// false
// false
// false
// true
}
func ExampleIsFloat() {
result1 := IsFloat("")
result2 := IsFloat("3")
result3 := IsFloat(0)
result4 := IsFloat(0.1)
fmt.Println(result1)
fmt.Println(result2)
fmt.Println(result3)
fmt.Println(result4)
// Output:
// false
// false
// false
// true
}
func ExampleIsNumber() {
result1 := IsNumber("")
result2 := IsNumber("3")
result3 := IsNumber(0)
result4 := IsNumber(0.1)
fmt.Println(result1)
fmt.Println(result2)
fmt.Println(result3)
fmt.Println(result4)
// Output:
// false
// false
// true
// true
}

View File

@@ -100,6 +100,34 @@ func TestIsJSON(t *testing.T) {
assert.Equal(false, IsJSON("&@#$%^&*")) assert.Equal(false, IsJSON("&@#$%^&*"))
} }
func TestIsNumber(t *testing.T) {
assert := internal.NewAssert(t, "TestIsNumber")
assert.Equal(false, IsNumber(""))
assert.Equal(false, IsNumber("3"))
assert.Equal(true, IsNumber(0))
assert.Equal(true, IsNumber(0.1))
}
func TestIsFloat(t *testing.T) {
assert := internal.NewAssert(t, "TestIsFloat")
assert.Equal(false, IsFloat(""))
assert.Equal(false, IsFloat("3"))
assert.Equal(false, IsFloat(0))
assert.Equal(true, IsFloat(0.1))
}
func TestIsInt(t *testing.T) {
assert := internal.NewAssert(t, "TestIsInt")
assert.Equal(false, IsInt(""))
assert.Equal(false, IsInt("3"))
assert.Equal(false, IsInt(0.1))
assert.Equal(true, IsInt(0))
assert.Equal(true, IsInt(-1))
}
func TestIsNumberStr(t *testing.T) { func TestIsNumberStr(t *testing.T) {
assert := internal.NewAssert(t, "TestIsNumberStr") assert := internal.NewAssert(t, "TestIsNumberStr")