79 lines
2.2 KiB
Go
79 lines
2.2 KiB
Go
package utility
|
||
|
||
import (
|
||
"strings"
|
||
|
||
"github.com/shopspring/decimal"
|
||
)
|
||
|
||
// DesensitizeGeneric 通用脱敏函数：
|
||
// startReserveCount: 从字符串开头保留的字符数
|
||
// endReserveCount: 从字符串末尾保留的字符数
|
||
// asteriskChar: 用于替换的字符 (例如 '*')
|
||
// 例如：DesensitizeGeneric("这是一个秘密信息", 2, 2, '*') -> 这**信息
|
||
func DesensitizeGeneric(text string, startReserveCount int, endReserveCount int, asteriskChar rune) string {
|
||
if text == "" {
|
||
return ""
|
||
}
|
||
runes := []rune(text)
|
||
length := len(runes)
|
||
|
||
// 计算实际需要保留的总字符数
|
||
totalReservedLength := startReserveCount + endReserveCount
|
||
|
||
// 如果需要保留的字符数大于等于总长度，则不脱敏
|
||
if totalReservedLength >= length {
|
||
return text
|
||
}
|
||
|
||
// 确保保留位数不为负数
|
||
if startReserveCount < 0 {
|
||
startReserveCount = 0
|
||
}
|
||
if endReserveCount < 0 {
|
||
endReserveCount = 0
|
||
}
|
||
|
||
// 如果保留总长度为0，且字符串不为空，则全部脱敏
|
||
if totalReservedLength == 0 && length > 0 {
|
||
return strings.Repeat(string(asteriskChar), length)
|
||
}
|
||
|
||
// 确保保留的长度不超过总长度
|
||
if startReserveCount > length {
|
||
startReserveCount = length
|
||
}
|
||
if endReserveCount > length-startReserveCount {
|
||
endReserveCount = length - startReserveCount
|
||
}
|
||
|
||
// 构造脱敏字符串
|
||
var sb strings.Builder
|
||
// 写入前缀
|
||
if startReserveCount > 0 {
|
||
sb.WriteString(string(runes[0:startReserveCount]))
|
||
}
|
||
// 写入星号
|
||
asteriskCount := length - startReserveCount - endReserveCount
|
||
if asteriskCount > 0 {
|
||
sb.WriteString(strings.Repeat(string(asteriskChar), asteriskCount))
|
||
}
|
||
// 写入后缀
|
||
if endReserveCount > 0 && length-endReserveCount >= startReserveCount { // 确保后缀不与前缀重叠
|
||
sb.WriteString(string(runes[length-endReserveCount:]))
|
||
}
|
||
|
||
return sb.String()
|
||
}
|
||
|
||
func StringToDecimal(val string) decimal.Decimal {
|
||
cleanedNum := strings.TrimRight(val, "\x00") // 去除空字符
|
||
cleanedNum = strings.TrimSpace(cleanedNum) // 去除空格
|
||
cleanedNum = strings.ReplaceAll(cleanedNum, ",", "") // 去除逗号
|
||
d, err := decimal.NewFromString(cleanedNum)
|
||
if err != nil {
|
||
return decimal.Zero
|
||
}
|
||
return d
|
||
}
|