goka PARTITION 示范

这是一个使用 goka PARTITION 函数的示例，假设我们有一个 Kafka 主题名为 "test-topic"，其消息的键（key）为字符串类型，值（value）为整数类型。我们希望将消息按照键的哈希值分配到 4 个分区中：

func main() {
    // 创建一个 goka 处理器
    p, err := goka.NewProcessor([]string{"localhost:9092"}, goka.DefineGroup(
        goka.Group("test-group"),
        goka.Input("test-topic", new(codec.String), new(codec.Int)),
        goka.Persist(new(codec.String), new(codec.Int)),
    ))
    if err != nil {
        panic(err)
    }
    defer p.Close()

    // 注册处理器函数
    p.HandleFunc(new(codec.String), func(ctx goka.Context, key string, value int) error {
        // 处理消息
        fmt.Printf("Received message: key=%s, value=%d\n", key, value)
        return nil
    })

    // 使用 goka PARTITION 函数将消息分配到不同的分区中
    p.Partition(func(key interface{}, _ interface{}, numPartitions int) int {
        s := key.(string)
        h := fnv.New32a()
        h.Write([]byte(s))
        return int(h.Sum32() % uint32(numPartitions))
    })

    // 启动处理器
    if err := p.Run(context.Background()); err != nil {
        panic(err)
    }
}

在这个示例中，我们使用了 goka PARTITION 函数将消息按照键的哈希值分配到不同的分区中。在 Partition 函数中，我们首先将键转换为字符串类型，然后计算其哈希值，并将哈希值对分区数取模，得到分配的分区编号。这样，相同键的消息将被分配到同一个分区中。在处理器函数中，我们可以根据需要对分区进行处理。