goregen - 基于正则表达式的随机字符串生成库

goregen 是一个 Go 语言库，可以根据给定的正则表达式模式生成匹配该模式的随机字符串。这在测试、模拟数据生成等场景中非常有用。

安装

go get github.com/asaskevich/goregen

基本用法

1. 简单示例

package main

import (
	"fmt"
	"github.com/asaskevich/goregen"
)

func main() {
	// 生成一个随机的字母数字字符串，长度5-10个字符
	str, err := goregen.Generate(`[a-zA-Z0-9]{5,10}`)
	if err != nil {
		panic(err)
	}
	fmt.Println("随机字符串:", str)
}

2. 常用模式示例

func main() {
	// 生成随机邮箱
	email, _ := goregen.Generate(`[a-z]{5,10}\.[a-z]{5,10}@(gmail|yahoo|outlook)\.com`)
	fmt.Println("随机邮箱:", email)

	// 生成随机IP地址
	ip, _ := goregen.Generate(`(25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)\.(25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)\.(25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)\.(25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)`)
	fmt.Println("随机IP:", ip)

	// 生成随机日期
	date, _ := goregen.Generate(`(19|20)\d\d-(0[1-9]|1[012])-(0[1-9]|[12][0-9]|3[01])`)
	fmt.Println("随机日期:", date)
}

3. 自定义随机生成器

func main() {
	// 创建自定义生成器
	generator := goregen.NewGenerator("[A-Z]{2}[0-9]{4}")

	// 生成多个匹配模式的字符串
	for i := 0; i < 5; i++ {
		str, err := generator.Generate()
		if err != nil {
			panic(err)
		}
		fmt.Println("生成的代码:", str)
	}
}

高级功能

1. 设置随机种子

func main() {
	// 设置随机种子以确保可重复的结果
	goregen.SetSeed(42) // 使用固定种子

	str1, _ := goregen.Generate(`[a-z]{5}`)
	str2, _ := goregen.Generate(`[a-z]{5}`)
	
	fmt.Println("固定种子结果1:", str1)
	fmt.Println("固定种子结果2:", str2)
	
	// 重置为随机种子
	goregen.SetSeed(0)
}

2. 自定义字符集

func main() {
	// 自定义字符集
	customSet := map[rune][]rune{
		'X': []rune("ABCDE"), // X 只生成A-E
		'Y': []rune("12345"), // Y 只生成1-5
	}

	// 使用自定义字符集生成
	str, err := goregen.GenerateWithArgs(`X{3}-Y{2}`, &goregen.GeneratorArgs{
		CustomRunes: customSet,
	})
	if err != nil {
		panic(err)
	}
	fmt.Println("自定义字符集结果:", str)
}

注意事项

1. goregen 不支持所有正则表达式特性，主要支持：

   • 字符类 [a-z], [A-Z0-9] 等
   • 量词 {n}, {n,m}, *, +, ?
   • 分组 (...)
   • 或操作 |

2. 不支持的特性包括：

   • 反向引用
   • 复杂的断言
   • 非贪婪匹配

3. 对于复杂的正则表达式，生成可能会失败或效率较低

实际应用示例

package main

import (
	"fmt"
	"github.com/asaskevich/goregen"
)

// 生成测试用户数据
func generateTestUsers(count int) []map[string]string {
	var users []map[string]string
	
	for i := 0; i < count; i++ {
		username, _ := goregen.Generate(`user_[a-z]{5,8}_[0-9]{3}`)
		email, _ := goregen.Generate(`[a-z]{5,10}\.[a-z]{5,10}@test\.com`)
		phone, _ := goregen.Generate(`1[3-9][0-9]{9}`) // 中国手机号格式
		
		users = append(users, map[string]string{
			"username": username,
			"email":    email,
			"phone":    phone,
		})
	}
	
	return users
}

func main() {
	users := generateTestUsers(3)
	for i, user := range users {
		fmt.Printf("用户%d: %v\n", i+1, user)
	}
}

goregen 是一个简单但强大的工具，特别适合在测试和开发阶段生成各种格式的随机数据。通过合理设计正则表达式，可以生成符合特定业务规则的测试数据，提高测试覆盖率和开发效率。