go-pkg/strings/strings.go

package strings

import (
	"reflect"
	"strconv"
	"strings"
	"unsafe"

	"github.com/pyihe/go-pkg/bytes"
)

// Bytes string转换为[]byte
func Bytes(s string) (b []byte) {
	sh := (*reflect.StringHeader)(unsafe.Pointer(&s))
	bh := (*reflect.SliceHeader)(unsafe.Pointer(&b))
	bh.Data, bh.Len, bh.Cap = sh.Data, sh.Len, sh.Len
	return b
}

// Uint64 string转换为uint64
func Uint64(s string) (uint64, error) {
	val, err := strconv.Atoi(s)
	if err != nil {
		return 0, err
	}

	return uint64(val), nil
}

// Int64 string转换为int64
func Int64(s string) (int64, error) {
	val, err := strconv.Atoi(s)
	if err != nil {
		return 0, err
	}
	return int64(val), nil
}

// Int string转换为int
func Int(s string) (int, error) {
	val, err := strconv.Atoi(s)
	if err != nil {
		return 0, err
	}
	return val, err
}

// Bool string转换为bool
func Bool(s string) (bool, error) {
	return strconv.ParseBool(s)
}

// Float32 string转换为float32
func Float32(s string) (float32, error) {
	fv, err := strconv.ParseFloat(s, 32)
	if err != nil {
		return 0, err
	}
	return float32(fv), nil
}

// Float64 string转换为float64
func Float64(s string) (float64, error) {
	return strconv.ParseFloat(s, 64)
}

// Last 返回s中最后一个b的位置
func Last(s string, b byte) int {
	i := len(s)
	for i--; i >= 0; i-- {
		if s[i] == b {
			break
		}
	}
	return i
}

// IsEmpty 判断字符串本身或者去除空格后是否为空字符串
func IsEmpty(s string) bool {
	if s == "" {
		return true
	}
	s = strings.TrimSpace(s)
	return s == ""
}

// Equal 判断两个字符串切片值是否相等
func Equal(a, b []string) bool {
	aLen, bLen := len(a), len(b)
	if aLen != bLen {
		return false
	}

	if (a == nil) != (b == nil) {
		return false
	}

	b = b[:aLen]
	for i, v := range a {
		if b[i] != v {
			return false
		}
	}
	return true
}

// Contain 判断字符串切片是否包含ele
func Contain(ele string, s []string) bool {
	for _, v := range s {
		if v == ele {
			return true
		}
	}
	return false
}

// Reverse 字符切片反转
func Reverse(s []string) {
	l := len(s)
	for i := l/2 - 1; i >= 0; i-- {
		opp := l - i - 1
		s[i], s[opp] = s[opp], s[i]
	}
}

// Remove 从s中删除第一个ele
func Remove(s *[]string, eles ...string) {
	if s == nil || len(*s) == 0 {
		return
	}
	for _, e := range eles {
		for i := 0; i < len(*s); {
			if e == (*s)[i] {
				copy((*s)[i:], (*s)[i+1:])
				(*s)[len(*s)-1] = ""
				*s = (*s)[:len(*s)-1]
				break
			} else {
				i++
			}
		}
	}
}

// RemoveAll 删除所有的eles
func RemoveAll(s *[]string, eles ...string) {
	for _, e := range eles {
		for i := 0; i < len(*s); {
			if (*s)[i] == e {
				copy((*s)[i:], (*s)[i+1:])
				(*s)[len(*s)-1] = ""
				*s = (*s)[:len(*s)-1]
			} else {
				i++
			}
		}
	}
}

// MaxSubStrLen 寻找最长不重复子串长度
func MaxSubStrLen(str string) int {
	var maxLen int
	var start int
	m := make(map[rune]int)
	for i, ch := range []rune(str) {
		if lastIndex, ok := m[ch]; ok && lastIndex >= start {
			start = lastIndex + 1
		}
		if i-start+1 > maxLen {
			maxLen = i - start + 1
		}
		m[ch] = i
	}
	return maxLen
}

func ToLower(s string) string {
	if isLower(s) {
		return s
	}
	b := make([]byte, len(s))
	for i := range s {
		c := s[i]
		if c >= 'A' && c <= 'Z' {
			c += 'a' - 'A'
		}
		b[i] = c
	}
	return bytes.String(b)
}

func ToUpper(s string) string {
	if isUpper(s) {
		return s
	}
	b := make([]byte, len(s))
	for i := range s {
		c := s[i]
		if c >= 'a' && c <= 'z' {
			c -= 'a' - 'A'
		}
		b[i] = c
	}
	return bytes.String(b)
}

func Add(s []string, a string) []string {
	return append(s, a)
}

func AddWithDistinct(s []string, a string) []string {
	for _, e := range s {
		if e == a {
			return s
		}
	}
	return Add(s, a)
}

func Union(s []string, a []string) []string {
	for _, ae := range a {
		found := false
		for _, se := range s {
			if ae == se {
				found = true
				break
			}
		}
		if !found {
			s = append(s, ae)
		}
	}
	return s
}

// Unique 字符串切片去重
func Unique(s []string) (r []string) {
loop:
	for _, se := range s {
		for _, rv := range r {
			if se == rv {
				continue loop
			}
		}
		r = append(r, se)
	}
	return
}