根据首尾字符串截取中间字符串

这段时间工作的需求多了不少，文章开始越来越水了。

实在抱歉，时间真的很仓促。

本次分享的是一个函数——根据首尾字符串截取中间的字符串。

虽然这个函数非常简单，但是真的很好用！也很常用！

比如 “我今天真的很高兴” 这句话，要把 今天 截取出来：

常规的方式是 strings.Index 出位置，然后根据字符串（今天）的长度，返回词的内容。

如果是 HTML 呢？比如获取链接地址：

<a href='https://m.baidu.com/'>跳转到百度</a>

这个时候你还想用 strings.Index 吗？

我是不想用的！太麻烦了。而用我的这个函数，就很方便了！

直接 StringCut(s, "href='", "'", false) 就可以轻松拿到链接地址！非常通用，无需关心越界问题。

func StringCut(s, begin, end string, withBegin bool) string {
    beginPos := strings.Index(s, begin)
    if beginPos == -1 {
        return ""
    }
    s = s[beginPos+len(begin):]
    endPos := strings.Index(s, end)
    if endPos == -1 {
        return ""
    }
    result := s[:endPos]
    if withBegin {
        return begin + result
    } else {
        return result
    }
}

附带单元测试，包含了很多例子，如果还有特殊情况，欢迎补充（找 bug 哈哈）：

func TestStringCut(t *testing.T) {
    type args struct {
        s         string
        begin     string
        end       string
        withBegin bool
    }
    tests := []struct {
        name string
        args args
        want string
    }{
        {"test", args{"", "a", "d", false}, ""},
        {"test", args{"abcd", "", "d", false}, "abc"},
        {"test", args{"abcd", "a", "", false}, ""},
        {"test", args{"abcd", "e", "d", false}, ""},
        {"test", args{"abcd", "a", "f", false}, ""},
        {"test", args{"abcd", "a", "d", false}, "bc"},
        {"test", args{"abcd", "a", "d", true}, "abc"},
        {"test", args{"abcd", "abcd", "", true}, "abcd"},
        {"test", args{"abcd", "", "abcd", true}, ""},
        {"test", args{"abcd", "ab", "cd", false}, ""},
        {"test", args{"abcd", "abcd", "e", false}, ""},
    }
    for _, tt := range tests {
        t.Run(tt.name, func(t *testing.T) {
            if got := StringCut(tt.args.s, tt.args.begin, tt.args.end, tt.args.withBegin); got != tt.want {
                t.Errorf("StringCut() = %v, want %v", got, tt.want)
            }
        })
    }
}

注：复制粘贴不要忘了 import "strings"。

时间不早了，得赶紧睡觉了，晚安。

文章来源于本人博客，发布于 2022-12-19，原文链接：https://imlht.com/archives/412/