mirror of
https://gitee.com/johng/gf
synced 2026-06-07 02:12:11 +08:00
105 lines
3.1 KiB
Go
105 lines
3.1 KiB
Go
// Copyright 2017-2018 gf Author(https://gitee.com/johng/gf). All Rights Reserved.
|
||
//
|
||
// This Source Code Form is subject to the terms of the MIT License.
|
||
// If a copy of the MIT was not distributed with this file,
|
||
// You can obtain one at https://gitee.com/johng/gf.
|
||
|
||
// Package gregex provides high performance API for regular expression functionality.
|
||
//
|
||
// 正则表达式.
|
||
package gregex
|
||
|
||
import (
|
||
"regexp"
|
||
)
|
||
|
||
// 转移正则规则字符串，例如：Quote(`[foo]`) 返回 `\[foo\]`
|
||
func Quote(s string) string {
|
||
return regexp.QuoteMeta(s)
|
||
}
|
||
|
||
// 校验所给定的正则表达式是否符合规范
|
||
func Validate(pattern string) error {
|
||
_, err := getRegexp(pattern)
|
||
return err
|
||
}
|
||
|
||
// 正则表达式是否匹配
|
||
func IsMatch(pattern string, src []byte) bool {
|
||
if r, err := getRegexp(pattern); err == nil {
|
||
return r.Match(src)
|
||
}
|
||
return false
|
||
}
|
||
|
||
func IsMatchString(pattern string, src string) bool {
|
||
return IsMatch(pattern, []byte(src))
|
||
}
|
||
|
||
// 正则匹配，并返回匹配的列表(参数[]byte)
|
||
func Match(pattern string, src []byte) ([][]byte, error) {
|
||
if r, err := getRegexp(pattern); err == nil {
|
||
return r.FindSubmatch(src), nil
|
||
} else {
|
||
return nil, err
|
||
}
|
||
}
|
||
|
||
// 正则匹配，并返回匹配的列表(参数[]string)
|
||
func MatchString(pattern string, src string) ([]string, error) {
|
||
if r, err := getRegexp(pattern); err == nil {
|
||
return r.FindStringSubmatch(src), nil
|
||
} else {
|
||
return nil, err
|
||
}
|
||
}
|
||
|
||
// 正则匹配，并返回所有匹配的列表(参数[]string)
|
||
func MatchAll(pattern string, src []byte) ([][][]byte, error) {
|
||
if r, err := getRegexp(pattern); err == nil {
|
||
return r.FindAllSubmatch(src, -1), nil
|
||
} else {
|
||
return nil, err
|
||
}
|
||
}
|
||
|
||
// 正则匹配，并返回所有匹配的列表(参数[][]string)
|
||
func MatchAllString(pattern string, src string) ([][]string, error) {
|
||
if r, err := getRegexp(pattern); err == nil {
|
||
return r.FindAllStringSubmatch(src, -1), nil
|
||
} else {
|
||
return nil, err
|
||
}
|
||
}
|
||
|
||
// 正则替换(全部替换)
|
||
func Replace(pattern string, replace, src []byte) ([]byte, error) {
|
||
if r, err := getRegexp(pattern); err == nil {
|
||
return r.ReplaceAll(src, replace), nil
|
||
} else {
|
||
return nil, err
|
||
}
|
||
}
|
||
|
||
// 正则替换(全部替换)，字符串
|
||
func ReplaceString(pattern, replace, src string) (string, error) {
|
||
r, e := Replace(pattern, []byte(replace), []byte(src))
|
||
return string(r), e
|
||
}
|
||
|
||
// 正则替换(全部替换)，给定自定义替换方法
|
||
func ReplaceFunc(pattern string, src []byte, replaceFunc func(b []byte) []byte) ([]byte, error) {
|
||
if r, err := getRegexp(pattern); err == nil {
|
||
return r.ReplaceAllFunc(src, replaceFunc), nil
|
||
} else {
|
||
return nil, err
|
||
}
|
||
}
|
||
|
||
// 正则替换(全部替换)，给定自定义替换方法
|
||
func ReplaceStringFunc(pattern string, src string, replaceFunc func(s string) string) (string, error) {
|
||
bytes, err := ReplaceFunc(pattern, []byte(src), func(bytes []byte) []byte {
|
||
return []byte(replaceFunc(string(bytes)))
|
||
})
|
||
return string(bytes), err
|
||
} |