e6956aa001
- React frontend with route-level code splitting - Backend rebranded from Baihu to TaskPool - DB brand migration script and local compatibility
93 lines
1.7 KiB
Go
93 lines
1.7 KiB
Go
package utils
|
|
|
|
import (
|
|
"bufio"
|
|
"io"
|
|
"unicode/utf8"
|
|
|
|
"golang.org/x/text/encoding/simplifiedchinese"
|
|
"golang.org/x/text/transform"
|
|
)
|
|
|
|
// ToUTF8 converts potentially non-UTF8 data (like GBK on Windows) to UTF-8
|
|
func ToUTF8(data []byte) string {
|
|
if utf8.Valid(data) {
|
|
return string(data)
|
|
}
|
|
// Try GBK (common on Windows)
|
|
reader := transform.NewReader(
|
|
bufio.NewReader(
|
|
&byteReader{data: data},
|
|
),
|
|
simplifiedchinese.GBK.NewDecoder(),
|
|
)
|
|
result, err := io.ReadAll(reader)
|
|
if err != nil {
|
|
return string(data)
|
|
}
|
|
return string(result)
|
|
}
|
|
|
|
type byteReader struct {
|
|
data []byte
|
|
pos int
|
|
}
|
|
|
|
func (r *byteReader) Read(p []byte) (n int, err error) {
|
|
if r.pos >= len(r.data) {
|
|
return 0, io.EOF
|
|
}
|
|
n = copy(p, r.data[r.pos:])
|
|
r.pos += n
|
|
return n, nil
|
|
}
|
|
|
|
// TrimLastRunes 从字符串尾部保留最多 maxRunes 个字符(不仅限于 ASCII,支持中英文混排的真实字符数量)
|
|
func TrimLastRunes(s string, maxRunes int) string {
|
|
// 如果字符串的总字节数小于等于 maxRunes,那么它的字符数一定也小于等于 maxRunes
|
|
if len(s) <= maxRunes {
|
|
return s
|
|
}
|
|
|
|
count := 0
|
|
for i := len(s); i > 0; {
|
|
_, size := utf8.DecodeLastRuneInString(s[:i])
|
|
i -= size
|
|
count++
|
|
if count == maxRunes {
|
|
return s[i:]
|
|
}
|
|
}
|
|
return s
|
|
}
|
|
|
|
// IntToStr 将整数转换为字符串
|
|
func IntToStr(n int) string {
|
|
if n == 0 {
|
|
return "0"
|
|
}
|
|
|
|
var negative bool
|
|
if n < 0 {
|
|
negative = true
|
|
n = -n
|
|
}
|
|
|
|
var digits []byte
|
|
for n > 0 {
|
|
digits = append(digits, byte('0'+n%10))
|
|
n /= 10
|
|
}
|
|
|
|
if negative {
|
|
digits = append(digits, '-')
|
|
}
|
|
|
|
// Reverse
|
|
for i, j := 0, len(digits)-1; i < j; i, j = i+1, j-1 {
|
|
digits[i], digits[j] = digits[j], digits[i]
|
|
}
|
|
|
|
return string(digits)
|
|
}
|