78 lines
2.0 KiB
Go
78 lines
2.0 KiB
Go
|
package sensitive
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"reflect"
|
|||
|
|
"strings"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// Desensitization
|
|||
|
|
// @description: 脱敏
|
|||
|
|
// @param r any: 要脱敏的数据(只支持结构体)
|
|||
|
|
// @param skip bool: 是否跳过脱敏
|
|||
|
|
// @return err error: 错误信息
|
|||
|
|
func Desensitization(r any, skip bool) (err error) {
|
|||
|
|
// 处理Data的值，如果也是结构体，就处理每个字段的标签，如果包含sen，就把sen改为sen_id
|
|||
|
|
if r == nil || skip {
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
// 判断是不是结构体
|
|||
|
|
isPointer := reflect.TypeOf(r).Kind() == reflect.Ptr
|
|||
|
|
dataType := reflect.TypeOf(r).Kind()
|
|||
|
|
if isPointer {
|
|||
|
|
dataType = reflect.TypeOf(r).Elem().Kind()
|
|||
|
|
}
|
|||
|
|
//log.Printf("数据类型: %v，是否为指针: %v", dataType, isPointer)
|
|||
|
|
|
|||
|
|
// 处理是数组的情况
|
|||
|
|
if dataType == reflect.Slice || dataType == reflect.Array {
|
|||
|
|
//log.Println("传入类型为数组")
|
|||
|
|
rs := reflect.ValueOf(r)
|
|||
|
|
if isPointer {
|
|||
|
|
rs = rs.Elem()
|
|||
|
|
}
|
|||
|
|
for i := 0; i < rs.Len(); i++ {
|
|||
|
|
id := rs.Index(i)
|
|||
|
|
err = Desensitization(id.Addr().Interface(), skip)
|
|||
|
|
if err != nil {
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 如果是指针结构体，处理脱敏
|
|||
|
|
if dataType == reflect.Struct {
|
|||
|
|
val := reflect.ValueOf(r)
|
|||
|
|
if isPointer {
|
|||
|
|
val = val.Elem()
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
for i := 0; i < val.NumField(); i++ {
|
|||
|
|
f := val.Field(i)
|
|||
|
|
tag := val.Type().Field(i).Tag.Get("sen")
|
|||
|
|
|
|||
|
|
//log.Printf("类型: %v -> %v 值: %v 脱敏标签是否存在: %v", f.Type(), f.Kind(), f.Interface(), tag != "")
|
|||
|
|
// 如果是结构体，递归调用
|
|||
|
|
if f.Kind() == reflect.Interface {
|
|||
|
|
//log.Println("开始处理子级")
|
|||
|
|
err = Desensitization(f.Interface(), skip)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if tag != "" {
|
|||
|
|
// 脱敏标签存在，处理一下，取出规则Id和占位符
|
|||
|
|
ruleId := strings.Split(tag, ",")[0]
|
|||
|
|
placeholder := strings.Split(tag, ",")[1]
|
|||
|
|
//log.Printf("脱敏规则Id: %v, 占位符: %v", ruleId, placeholder)
|
|||
|
|
|
|||
|
|
// 处理脱敏
|
|||
|
|
if handle, ok := senRuleMap[ruleId]; ok {
|
|||
|
|
newData := handle(f.Interface().(string), placeholder)
|
|||
|
|
//log.Printf("脱敏后的值: %v", newData)
|
|||
|
|
f.SetString(newData)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return
|
|||
|
|
}
|