添加自增控制

utils/auth/user.go

@@ -172,3 +172,12 @@ func getJwtClaims(AuthKey string, AccessSecret *string) (jwt.MapClaims, error) {
 
 	return nil, errors.New(fmt.Sprint("Invalid token", err))
 }
+
+func CheckValueRange[T comparable](v T, rangevalues ...T) bool {
+	for _, rv := range rangevalues {
+		if v == rv {
+			return true
+		}
+	}
+	return false
+}

utils/auth/user_test.go

@@ -28,3 +28,29 @@ func TestGenJwt(t *testing.T) {
 	}
 	// log.Println(claims)
 }
+
+func TestCheckValueRange(t *testing.T) {
+	v := 1
+	rv1 := []int{1, 3, 4}
+
+	if !CheckValueRange[int](v, rv1...) {
+		t.Error("1 in [1 3 4], but is false")
+	}
+
+	v = 2
+	if CheckValueRange(v, rv1...) {
+		t.Error("2 not in [1 3 4], but is true")
+	}
+
+	vstr := "funsen123"
+	rvstr := []string{"funsun123", "funsen123", "funsen23"}
+	if !CheckValueRange(vstr, rvstr...) {
+		t.Error("vstr in rvstr, but is false")
+	}
+
+	vstr = "funsen1234"
+	if CheckValueRange(vstr, rvstr...) {
+		t.Error("vstr not in rvstr, but is true")
+	}
+
+}