在 Go 中对正则表达式进行预编译缓存

2024-04-04 23:20

短信预约 -IT技能 免费直播动态提醒

编程网今天将给大家带来《在 Go 中对正则表达式进行预编译缓存》，感兴趣的朋友请继续看下去吧！以下内容将会涉及到等等知识点，如果你是正在学习Golang或者已经是大佬级别了，都非常欢迎也希望大家都能给我建议评论哈~希望能帮助到大家！

问题内容

下面是我的 golang 代码。每次调用验证方法时，我的编译方法都会被执行。我只想编译一次，而不是每次调用验证时都编译。

1）如何做？ 2）我的想法是创建一个实例变量，该变量在开始时为零。它将在验证中延迟初始化。

if (a != nil) {
  a, err := regexp.compile(rras.cfg.whitelist)
}

但是，如果我将变量声明为实例变量，

var a *regexp; // regexp.compile returns *regexp

我的编译器用红色下划线。如何修复它？

type RRAS struct {
    Cfg       *RRAPIConfig
}

type RRAPIConfig struct {
    WhiteList               string
}

func (rras *RRAS) validate(ctx context.Context) error {
        a, err := regexp.Compile(rras.Cfg.WhiteList)
}

解决方案

静态初始化

var whitelistregexp = regexp.mustcompile(cfg.whitelist)

func (rras *rras) validate(ctx context.context) error {
  if !whitelistregexp.match(...) {...}
}

这将在导入包后立即编译正则表达式，这通常是在程序启动时，在主方法中的任何代码执行之前。

好处

如果正则表达式损坏，您的程序将立即崩溃，这有助于快速找到错误。
代码非常小、干净，没有任何陷阱
无需担心 go 例程

缺点

潜在的缓慢编译可能会减慢整个程序（或服务器）的启动
仅当正则表达式是静态的且在启动时存在时才有效
仅当单个正则表达式（或几个静态正则表达式）用于所有情况时才有效

同步和缓存

var whitelistR struct{
  rex *regexp.Regexp
  once sync.Once
  err error
}

func (rras *RRAS) validate(ctx context.Context) error {
  whitelistR.once.Do(func() {
    whitelistR.ex, whitelistR.err = regexp.Compile(rras.Cfg.WhiteList)
  })

  if whitelistR.err != nil {
    return fmt.Errorf("could not compile regex: %w", err)
  }

  if !whitelistR.rex.Match(...) {...}
}

这将在第一次调用该方法时轻松编译正则表达式。 sync.once 非常重要，因为它是一个同步点，它保证对正则表达式的访问不会出现竞争条件。每次调用该方法都必须等到 regexp 第一次编译。之后同步非常快，因为它仅使用原子加载。

您还可以在主方法中调用一次 go.do(...) 来并行初始化正则表达式，以加快首次调用的速度，而不会阻塞其他方法。