Go语言实现字符串的Python式分区功能

package main

import (
    "fmt"
    "strings"
)

// Partition 函数根据第一个出现的分隔符将字符串 s 分割为三部分。
// 返回值依次是：分隔符之前的部分、分隔符本身、分隔符之后的部分。
// 如果分隔符未找到，则返回 (s, "", "")。
func Partition(s string, sep string) (before string, separator string, after string) {
    parts := strings.SplitN(s, sep, 2)
    if len(parts) == 1 {
        // 分隔符未找到，返回原始字符串和两个空字符串
        return parts[0], "", ""
    }
    // 分隔符找到，返回分隔符之前、分隔符本身、分隔符之后的部分
    return parts[0], sep, parts[1]
}

func main() {
    // 示例 1: 分隔符存在且只出现一次
    user, sep, domain := Partition("foo@example.com", "@")
    fmt.Printf("Input: \"foo@example.com\", Separator: \"@\"\n")
    fmt.Printf("Before: \"%s\", Separator: \"%s\", After: \"%s\"\n", user, sep, domain)
    // 预期输出: Before: "foo", Separator: "@", After: "example.com"

    fmt.Println("---")

    // 示例 2: 分隔符存在且出现多次
    pathBefore, sep2, pathAfter := Partition("/usr/local/bin/go", "/")
    fmt.Printf("Input: \"/usr/local/bin/go\", Separator: \"/\"\n")
    fmt.Printf("Before: \"%s\", Separator: \"%s\", After: \"%s\"\n", pathBefore, sep2, pathAfter)
    // 预期输出: Before: "", Separator: "/", After: "usr/local/bin/go"
    // 注意：由于第一个字符就是'/'，所以Before是空字符串。

    fmt.Println("---")

    // 示例 3: 分隔符不存在
    textBefore, sep3, textAfter := Partition("Hello_World", "-")
    fmt.Printf("Input: \"Hello_World\", Separator: \"-\"\n")
    fmt.Printf("Before: \"%s\", Separator: \"%s\", After: \"%s\"\n", textBefore, sep3, textAfter)
    // 预期输出: Before: "Hello_World", Separator: "", After: ""

    fmt.Println("---")

    // 示例 4: 字符串为空
    emptyBefore, sep4, emptyAfter := Partition("", "@")
    fmt.Printf("Input: \"\", Separator: \"@\"\n")
    fmt.Printf("Before: \"%s\", Separator: \"%s\", After: \"%s\"\n", emptyBefore, sep4, emptyAfter)
    // 预期输出: Before: "", Separator: "", After: ""

    fmt.Println("---")

    // 示例 5: 分隔符为空字符串
    // 这种情况下，SplitN的行为是返回原始字符串本身。
    // 因此，Partition函数会将其视为未找到分隔符。
    emptySepBefore, emptySepSep, emptySepAfter := Partition("abc", "")
    fmt.Printf("Input: \"abc\", Separator: \"\"\n")
    fmt.Printf("Before: \"%s\", Separator: \"%s\", After: \"%s\"\n", emptySepBefore, emptySepSep, emptySepAfter)
    // 预期输出: Before: "abc", Separator: "", After: ""
}