🎨 代码完善

run_test.go

@@ -4,7 +4,9 @@ import (
 	"encoding/json"
 	"log"
 	"net/http"
+	"strings"
 	"testing"
+	"unicode/utf8"
 )
 
 // Response 返回值
@@ -29,6 +31,7 @@ type Data struct {
 	Name     string `json:"name"`
 	Phone    string `json:"phone" sen:"phone,*"`       // 使用脱敏标签，规则为手机号，替换占位字符为*
 	IdNumber string `json:"idNumber" sen:"idNumber,#"` // 使用脱敏标签，规则为身份证号码，替换占位字符为*
+	Email    string `json:"email" sen:"email,*"`       // 使用脱敏标签，规则为邮箱，替换占位字符为*
 }
 
 // 模拟测试一下
@@ -38,11 +41,13 @@ func TestDeal(t *testing.T) {
 		Name:     "张三",
 		Phone:    "13800138000",
 		IdNumber: "420102199010101010",
+		Email:    "zhangsan@gmail.com",
 	}, {
 		Id:       "234",
 		Name:     "李四",
 		Phone:    "13800138001",
 		IdNumber: "420102199010101011",
+		Email:    "lisi@gmail.com",
 	}}
 
 	pageData := PageData{
@@ -69,6 +74,14 @@ func TestDeal(t *testing.T) {
 	//bs, _ = json.Marshal(response)
 	//log.Printf("假设是管理员，需要跳过处理，脱敏后的数据: %v", string(bs))
 
+	AddHandler("email", func(src, p string) string {
+		// 将@符号后面的替换为*
+		idx := strings.Index(src, "@")
+		dst := src[:idx+1] + strings.Repeat(p, utf8.RuneCountInString(src)-idx-1)
+
+		return dst
+	})
+
 	if err := Desensitization(response, false); err != nil {
 		log.Println(err)
 	}