Go语言中如何优雅选择base64.StdEncoding与RawStdEncoding解码字符串

众所周知，Go语言中解码标准base64字符串通常涉及两个方法：base64.StdEncoding和base64.RawStdEncoding，如果调用的解码器和输入编码规则不匹配，会抛出illegal base64 data at input byte xxx错误。

两者的核心差异如下：

const (
    StdPadding rune = '=' // 标准填充字符
    NoPadding  rune = -1  // 无填充
)

RawStdEncoding是RFC 4648第3.2节定义的标准无填充原始base64编码，与StdEncoding的区别仅为省略了填充字符。

末尾填充符判断方案的问题 #

通过判断末尾是否为=选择解码器的方案存在明显缺陷：

• 空输入直接崩溃：代码直接取s[len(s)-1:]，如果输入是空字符串会触发下标越界，缺少基础合法性校验
• 逻辑覆盖不全：当原始输入字节长度刚好是3的倍数时，StdEncoding编码的结果也不会带末尾填充符，此时判断逻辑会错误选择RawStdEncoding，虽然该场景下大多能正常解码，但本质是逻辑判断失效
• 无容错能力：如果输入字符串被截断、夹杂非法字符，该逻辑无法处理异常

错误重试方案的优劣 #

错误重试方案的优缺点都很明确：

• 优势：兼容性极强，无需提前判断输入的填充规则，只要是符合RFC 4648标准的Std/RawStd编码字符串都能正确解码，逻辑简单不易出错
• 劣势：仅当输入为无填充字符串时会多执行一次错误判定流程，不过常规业务场景下base64解码的错误率极低，这点性能损耗几乎可以忽略

最优方案选型 #

根据不同场景可以选择不同的方案：

1. 输入规则明确场景：如果上游已经明确输出的base64是否带填充，直接调用对应的base64.StdEncoding.DecodeString或者base64.RawStdEncoding.DecodeString即可，性能最高
2. 需要兼容两种填充规则的场景：
   • 追求代码易维护：直接用错误重试方案即可，代码示例：

     rawByte, err := base64.StdEncoding.DecodeString(s)
     if err != nil {
         rawByte, err = base64.RawStdEncoding.DecodeString(s)
     }
     

   • 追求更高性能：可以先移除所有填充符再用无填充解码器解码，无需错误重试，性能更好，代码示例：

     import "strings"
     
     s = strings.TrimRight(s, "=")
     rawByte, err := base64.RawStdEncoding.DecodeString(s)
     

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