覆蓋現有的 golang 結構並添加字段

Question

    func (c *client) Init(conf config.Config, topic string) (err error) {
    defer func() {
        p := recover()
        switch v := p.(type) {
        case error:
            err = v
        }
    }()
    c.reader = kafka.NewReader(kafka.ReaderConfig{
        Brokers:               conf.Brokers,
        GroupID:               conf.GroupID,
        Topic:                 topic,
        MaxWait:               1 * time.Second,    // maximum time to wait for new messages
        MinBytes:              1,                  // minimum message size
        MaxBytes:              10e6,               // maximum message size 1 MByte (= maximum size Kafka can handle)
        RetentionTime:         time.Hour * 24 * 7, // keep ConsumerGroup for 1 week
        WatchPartitionChanges: true,               // watch for changes to the partitions (e.g. increase of partitions)
    })
    if conf.TlsEnabled {
        d := &kafka.Dialer{
            TLS: &tls.Config{},
        }
    }
    return err
}

長話短說：如果 TlsEnabled 為真，我想做的是將字段Dialer: d添加到c.reader 。 c:reader 是 ReaderConfig 類型，在我的例子中已經包含 Dialer 字段：

    d := &kafka.Dialer{
    TLS: &tls.Config{},
    }

Answer 1

如果我正確理解了您的問題，那么當且僅當conf.TlsEnabled為真時，您才想在kafka.ReaderConfig上設置Dialer字段。 在這種情況下，您應該在調用kafka.NewReader之前移動您的if conf.TlsEnabled檢查，並將kafka.ReaderConfig分配給一個變量，如下所示：

rConf := kafka.ReaderConfig{
    Brokers:               conf.Brokers,
    GroupID:               conf.GroupID,
    Topic:                 topic,
    MaxWait:               1 * time.Second,    // maximum time to wait for new messages
    MinBytes:              1,                  // minimum message size
    MaxBytes:              10e6,               // maximum message size 1 MByte (= maximum size Kafka can handle)
    RetentionTime:         time.Hour * 24 * 7, // keep ConsumerGroup for 1 week
    WatchPartitionChanges: true,               // watch for changes to the partitions (e.g. increase of partitions)
}
if conf.TlsEnabled {
    rConf.Dialer = &kafka.Dialer{
        TLS: &tls.Config{},
    }
}
// now assign c.reader
c.reader = kafka.NewReader(rConf)

只是一個小小的挑剔：在 golang 中，首字母縮寫詞和其他首字母縮寫詞應該全部大寫。 您的配置類型不應該有一個名為TlsEnabled的字段，而應該是TLSEnabled 。

Answer 2

您不能將字段添加到現有類型，但可以將client嵌入自定義類型。 然后可以通過自定義類型直接訪問這些字段

type myType struct {
    something string
    foo       int
    bar       int
}

type myExtendedType struct {
    myType
    dialer *kafka.Dialer
}

func main() {
    ext := myExtendedType{
        myType: myType{
            something: "some",
            foo:       24,
            bar:       42,
        },
        dialer: &kafka.Dialer{
            TLS: &tls.Config{},
        },
    }
    println(ext.something, ext.foo, ext.bar, ext.dialer)
}