flat-records: throw exception when schema cannot be found instead of …

…npe (#684)
Netflix · Jul 2, 2024 · 63d369b · 63d369b
1 parent 67625e2
commit 63d369b
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/hollow/src/main/java/com/netflix/hollow/core/write/objectmapper/flatrecords/FlatRecord.java b/hollow/src/main/java/com/netflix/hollow/core/write/objectmapper/flatrecords/FlatRecord.java
@@ -50,6 +50,9 @@ public FlatRecord(ByteData recordData, HollowSchemaIdentifierMapper schemaIdMapp
 
         int topRecordSchemaId = VarInt.readVInt(recordData, dataStartByte + locationOfTopRecord);
         HollowSchema topRecordSchema = schemaIdMapper.getSchema(topRecordSchemaId);
+        if (topRecordSchema == null) {
+            throw new IllegalStateException("Could not find schema for schemaId " + topRecordSchemaId);
+        }
 
         if (topRecordSchema.getSchemaType() == SchemaType.OBJECT) {
             PrimaryKey primaryKey = ((HollowObjectSchema) topRecordSchema).getPrimaryKey();