Go语言中修改KSUID生成函数使其返回字符串类型

错误原因 #

你之前的报错是因为函数返回类型和实际返回值类型不匹配：原函数genKsuid()的返回类型是自定义的id（本质是[]byte），但你尝试返回id.String()（string类型），两种类型无法直接转换，所以编译器报错。

修正方案 #

要实现var id_gen string = genKsuid()的调用方式，只需要调整函数的返回类型，并返回KSUID的字符串形式即可，具体修改如下：

简化版代码（移除不必要的自定义类型）

import (
    "fmt"
    "log"
    "time"
    "github.com/segmentio/ksuid"
)

func genKsuid() string {
    id, err := ksuid.NewRandomWithTime(time.Now())
    if err != nil {
        log.Fatal(err)
    }
    fmt.Printf("ksuid:     %s\n", id.String())
    // 直接返回KSUID的字符串形式
    return id.String()
}

// 现在可以直接赋值给string类型变量
var id_gen string = genKsuid()

保留自定义类型的版本（如果需要保留id_return）

如果你想保留自定义的id_return类型，需要把函数返回类型改成id_return，同时将id.String()转换为该类型：

import (
    "fmt"
    "log"
    "time"
    "github.com/segmentio/ksuid"
)

type id_return string

func genKsuid() id_return {
    id, err := ksuid.NewRandomWithTime(time.Now())
    if err != nil {
        log.Fatal(err)
    }
    fmt.Printf("ksuid:     %s\n", id.String())
    // 将string转换为自定义的id_return类型
    return id_return(id.String())
}

// 赋值时可以直接用string接收，因为类型兼容
var id_gen string = genKsuid()

关键修改点 #

• 把genKsuid()的返回类型从id（[]byte）改成string或id_return
• 返回语句从return id.Bytes()改为return id.String()（或转换为自定义类型）

内容的提问来源于stack exchange，提问作者code_recall