gf/g/util/gregx/regx.go

// Copyright 2017 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 gregx

import (
    "regexp"
)

// 正则表达式是否匹配
func IsMatch(pattern string, src []byte) bool {
    match, err := regexp.Match(pattern, src)
    if err != nil {
        return false
    }
    return match
}

func IsMatchString(pattern string, src string) bool {
    return IsMatch(pattern, []byte(src))
}

// 正则匹配，并返回匹配的列表
func MatchString(pattern string, src string) ([]string, error) {
    reg, err := regexp.Compile(pattern)
    if err != nil {
        return nil, err
    }
    s := reg.FindStringSubmatch(src)
    return s, nil
}

func MatchAllString(pattern string, src string) ([][]string, error) {
    reg, err := regexp.Compile(pattern)
    if err != nil {
        return nil, err
    }
    s := reg.FindAllStringSubmatch(src, -1)
    return s, nil
}

// 正则替换(全部替换)
func Replace(pattern string, replace, src []byte) ([]byte, error) {
    reg, err := regexp.Compile(pattern)
    if err != nil {
        return src, err
    }
    return reg.ReplaceAll(src, replace), nil
}

// 正则替换(全部替换)，字符串
func ReplaceString(pattern, replace, src string) (string, error) {
    r, e := Replace(pattern, []byte(replace), []byte(src))
    return string(r), e
}