如何在Kafka中测量Avro消息大小？

import org.apache.avro.Schema;
import org.apache.avro.generic.GenericData;
import org.apache.avro.generic.GenericRecord;
import org.apache.avro.io.BinaryEncoder;
import org.apache.avro.io.EncoderFactory;
import org.apache.avro.specific.SpecificDatumWriter;

import java.io.ByteArrayOutputStream;
import java.io.IOException;

public class AvroMessageSizeMeasurement {

    public static void main(String[] args) throws IOException {
        // 1. 定义Avro消息的Schema
        String schemaString = "{\"type\":\"record\",\"name\":\"Message\",\"fields\":[{\"name\":\"id\",\"type\":\"int\"},{\"name\":\"name\",\"type\":\"string\"}]}";
        Schema schema = new Schema.Parser().parse(schemaString);

        // 2. 创建一个Avro消息
        GenericRecord message = new GenericData.Record(schema);
        message.put("id", 1);
        message.put("name", "John Doe");

        // 3. 使用Avro的编码器将消息编码为Avro格式
        ByteArrayOutputStream outputStream = new ByteArrayOutputStream();
        BinaryEncoder encoder = EncoderFactory.get().binaryEncoder(outputStream, null);
        SpecificDatumWriter<GenericRecord> writer = new SpecificDatumWriter<>(schema);
        writer.write(message, encoder);
        encoder.flush();

        // 4. 获取编码后的Avro消息的字节大小
        int messageSize = outputStream.toByteArray().length;

        // 5. 打印消息大小
        System.out.println("Avro消息大小: " + messageSize + " 字节");
    }
}