Go中的静态初始化？

Question

我目前正在研究Go Lang教程,但遇到了其中一个练习的问题:

https://tour.golang.org/methods/23

练习让我实现了一个ROT13密码.我决定使用从字节到旋转值的映射来实现密码,但我不确定初始化此映射的最佳方法.我不想使用文字初始化地图,但更喜欢通过循环遍历字母表并在循环内设置(键,值)对来编程.我还希望只能从Rot13Reader结构/对象访问该地图,并让所有实例(？)共享相同的地图(而不是每个Rot13Reader一个副本).

这是我目前正在进行的Go计划:

package main

import (
    "io"
    "os"
    "strings"
)

type rot13Reader struct {
    r io.Reader
}

var rot13Map = map[byte]byte{}

func (rotr *rot13Reader) Read(p []byte) (int, error) {
    n, err := rotr.r.Read(p)
    for i := 0; i < n; i++ {
        if sub := rot13Map[p[i]]; sub != byte(0) {
            p[i] = sub
        }
    }
    return n, err
}

func main() {
    func() {
        var uppers = []byte("ABCDEFGHIJKLMNOPQRSTUVWXYZ")
        var lowers = []byte("abcdefghijklmnopqrstuvwxyz")

        var init = func (alphabet []byte) {
            for i, char := range alphabet {
                rot13_i := (i + 13) % 26
                rot13Map[char] = alphabet[rot13_i]
            }
        }

        init(uppers)
        init(lowers)
    }()

    s := strings.NewReader("Lbh penpxrq gur pbqr!")
    r := rot13Reader{s}
    io.Copy(os.Stdout, &r)
}

以下是我遇到的问题:

• 我不希望有准备rot13Map的main()
• 我不想rot13Map在全球范围内.
• 我不希望每个副本rot13Reader都有一个单独的rot13Map

有没有办法在Go中实现我想要的东西？

Answer 1

为了做到这一点,我会做一个rot13包.您可以在init()函数中以编程方式创建映射,并将其作为包级别全局提供给所有rot13解码器.导入包时会运行init函数.

因为Rot13Reader是包中唯一的类型,所以它是唯一能够访问您的地图的类型.

警告:所有代码未经测试.

package rot13

import (
    "io"
)

var rot13Map = map[byte]byte{}

func init() {
    var uppers = []byte("ABCDEFGHIJKLMNOPQRSTUVWXYZ")
    var lowers = []byte("abcdefghijklmnopqrstuvwxyz")

    var init = func(alphabet []byte) {
        for i, char := range alphabet {
            rot13_i := (i + 13) % 26
            rot13Map[char] = alphabet[rot13_i]
        }
    }

    init(uppers)
    init(lowers)
}

type Reader struct {
    r io.Reader
}

func (rotr Reader) Read(p []byte) (int, error) {
    n, err := rotr.r.Read(p)
    for i := 0; i < n; i++ {
        if sub := rot13Map[p[i]]; sub != byte(0) {
            p[i] = sub
        }
    }
    return n, err
}

显然,你不能在旅行中制作另一个包.您可以通过main访问rot13Map.您需要在本地运行Go以获得所需的分离.

Answer 2

为了完整起见:对于除了init包中的函数之外的初始化工作sync.Once,它只运行一次提供的函数.

您创建一个Once对象并Do使用您的函数调用它.只要Once 对象的状态没有改变,所提供的函数只会被调用一次.

例:

import "sync"

var readerInitOnce sync.Once

func (rotr *rot13Reader) Read(p []byte) (int, error) {
    readerInitOnce.Do(initRot13Map)
    ...
}