Avro是否支持将复杂对象或非字符串类型作为Map的键？

好问题！Avro确实在AVRO-680这个issue中添加了非字符串类型作为Map键的支持，这个特性从Avro 1.8.0版本正式引入，不过早期的文档和示例相对较少，容易让人找不到头绪。

一、Schema的正确写法 #

你期望的定义方式其实是完全可行的，下面给出具体的示例：

1. JSON Schema格式 #

假设你已经定义了MyKeyType（比如是一个自定义record）和MyValueType，对应的完整Schema如下：

{
  "name": "aThing",
  "type": "record",
  "fields": [
    {
      "name": "aMapOfThings",
      "type": {
        "type": "map",
        "keys": "MyKeyType",
        "values": "MyValueType"
      }
    }
  ],
  "types": [
    {
      "name": "MyKeyType",
      "type": "record",
      "fields": [
        {"name": "id", "type": "int"},
        {"name": "code", "type": "string"}
      ]
    },
    {
      "name": "MyValueType",
      "type": "record",
      "fields": [
        {"name": "value", "type": "long"},
        {"name": "description", "type": "string"}
      ]
    }
  ]
}

这里通过types字段定义了依赖的自定义类型，Map的keys字段直接指定为自定义的MyKeyType即可。

2. AVDL格式 #

对应的AVDL定义和你预想的几乎一致，非常直观：

@namespace("com.example")
protocol ExampleProtocol {
  record MyKeyType {
    int id;
    string code;
  }

  record MyValueType {
    long value;
    string description;
  }

  record aThing {
    map<MyKeyType, MyValueType> aMapOfThings = [];
  }
}

二、关键注意事项 #

• 键类型的限制：不是所有类型都能作为Map键，必须满足：
  • 类型是Avro支持的可哈希类型，包括基本类型（int、long、boolean、float、double、fixed、enum），以及自定义record类型（此时record的所有字段都会参与哈希值计算和相等性判断）。
  • 不能使用联合类型（union）作为Map键，因为联合类型的哈希和相等性判断逻辑不明确。
• 兼容性问题：虽然Avro 1.8.0+支持这个特性，但部分旧版本的Avro语言绑定（比如某些早期的Java、Python客户端）可能存在兼容性问题，使用前建议确认客户端版本是否支持。
• 序列化逻辑：当使用自定义record作为键时，Avro会将整个record序列化为字节数组作为Map的底层键，确保在反序列化时能准确还原出原始的键对象。

三、补充说明 #

AVRO-680的补丁主要是扩展了Avro的Schema解析逻辑，以及Map的序列化/反序列化实现，让非字符串类型可以作为键，你的需求和Protobuf中“Protobuf objects as Keys in Maps”的场景是类似的，Avro现在已经可以支持这类场景了。

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