Java/Scala中的随机Avro数据生成器

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

import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.util.Random;

public class AvroDataGenerator {
    private static final String SCHEMA_STRING = "{\"type\":\"record\",\"name\":\"Person\",\"fields\":[{\"name\":\"name\",\"type\":\"string\"},{\"name\":\"age\",\"type\":\"int\"}]}";
    private static final Schema SCHEMA = new Schema.Parser().parse(SCHEMA_STRING);

    public static byte[] generateRandomAvroData() throws IOException {
        GenericRecord record = new GenericData.Record(SCHEMA);
        record.put("name", generateRandomString());
        record.put("age", generateRandomInt());

        DatumWriter<GenericRecord> writer = new SpecificDatumWriter<>(SCHEMA);
        ByteArrayOutputStream outputStream = new ByteArrayOutputStream();
        Encoder encoder = EncoderFactory.get().binaryEncoder(outputStream, null);
        writer.write(record, encoder);
        encoder.flush();

        return outputStream.toByteArray();
    }

    private static String generateRandomString() {
        String characters = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ";
        Random random = new Random();
        StringBuilder sb = new StringBuilder();
        int length = random.nextInt(10) + 1;
        for (int i = 0; i < length; i++) {
            int index = random.nextInt(characters.length());
            sb.append(characters.charAt(index));
        }
        return sb.toString();
    }

    private static int generateRandomInt() {
        Random random = new Random();
        return random.nextInt(100);
    }
}