.withPartitionKey(partitionKey) .withSequenceNumber(r.getSequenceNumber()) .withApproximateArrivalTimestamp(aat < 0 ? null : new Date(aat)); result.add(new UserRecord(true, record, subSeqNum++, explicitHashKey));
rec.withPartitionKey("1").withSequenceNumber(r + "") .withData(ByteBuffer.wrap(line.getBytes())) .withApproximateArrivalTimestamp(approximateArrivalTimestamp);
private static List<Record> createRecords(int numRecords) { List<Record> records = new ArrayList<>(numRecords); Random rand = new Random(); for (int i = 0; i < numRecords; i++) { String dataStr = "testData-" + System.currentTimeMillis(); ByteBuffer data = ByteBuffer.wrap(dataStr.getBytes(StandardCharsets.UTF_8)); String key = String.format("partitionKey-%d", rand.nextLong()); String seqNum = String.format("%04d", 5 * i + 1); Record record = new Record() .withData(data) .withPartitionKey(key) .withSequenceNumber(seqNum) .withApproximateArrivalTimestamp(new Date()); records.add(record); } return records; } }
public Record convertToRecord() { return new Record() .withApproximateArrivalTimestamp(arrivalTimestamp.toDate()) .withData(ByteBuffer.wrap(data.getBytes(StandardCharsets.UTF_8))) .withSequenceNumber(sequenceNumber) .withPartitionKey(""); }
public static KinesisEvent createEvent(Class clazz, String resource) throws UnsupportedEncodingException, IOException { /* * Create a kinesis record from a sample JSON file */ String json = IOUtils.toString(new InputStreamReader(clazz.getResourceAsStream(resource), "UTF-8")); Date approximateArrivalTimestamp = new Date(); approximateArrivalTimestamp.setTime(1478737790000l); Record rec = new Record(); rec.withPartitionKey("1").withSequenceNumber("2").withData(ByteBuffer.wrap(json.getBytes())) .withApproximateArrivalTimestamp(approximateArrivalTimestamp); /* * Create a KinesisEventRecord and add single Record */ KinesisEventRecord krecord = new KinesisEventRecord(); krecord.setKinesis(rec); krecord.setEventSourceARN("arn:aws:kinesis:us-east-1:1234:stream/test-events-stream"); krecord.setEventID("shardId-000000000000:1234"); /* * Add single KinesisEventRecord to a KinesisEvent */ KinesisEvent kevent = new KinesisEvent(); List<KinesisEventRecord> events = new ArrayList<KinesisEventRecord>(1); events.add(krecord); kevent.setRecords(events); return kevent; }