@@ -17,7 +17,6 @@ import scala.collection.mutable
1717import java .util .concurrent
1818
1919class WriterContext (val fileChannel : FileChannel , val executor : concurrent.ExecutorService ) {
20-
2120 val compressCtx = new ZstdWrapper .ZstdCtx
2221 var fileOffset : Long = 16L
2322 val compressQueue = mutable.ArrayDeque [concurrent.Future [(OutlineStorage , ByteBuffer )]]()
@@ -220,7 +219,7 @@ class WriterContext(val fileChannel: FileChannel, val executor: concurrent.Execu
220219
221220object Serialization {
222221 val logger = LoggerFactory .getLogger(getClass)
223- case class Counts (nodes : Int , edges : Int , properties : Int )
222+ case class Counts (nodeKinds : Int , edgeKinds : Int , propertyKinds : Int )
224223
225224 def writeGraph (graph : Graph , storagePath : Path , requestedExecutor : Option [concurrent.ExecutorService ] = None ): Counts = {
226225 logger.info(s " writing to storage at ` $storagePath` " )
@@ -241,9 +240,9 @@ object Serialization {
241240 }
242241
243242 private def innerWriteGraph (graph : Graph , writer : WriterContext ): Counts = {
244- val nodes = mutable.ArrayBuffer .empty[NodeItem ]
245- val edges = mutable.ArrayBuffer .empty[EdgeItem ]
246- val properties = mutable.ArrayBuffer .empty[PropertyItem ]
243+ val nodeKinds = mutable.ArrayBuffer .empty[NodeItem ]
244+ val edgeKinds = mutable.ArrayBuffer .empty[EdgeItem ]
245+ val propertyKinds = mutable.ArrayBuffer .empty[PropertyItem ]
247246 for (nodeKind <- graph.schema.nodeKinds) {
248247 val nodeLabel = graph.schema.getNodeLabel(nodeKind)
249248 val deletions = graph
@@ -252,7 +251,7 @@ object Serialization {
252251 case deleted : GNode if AccessHelpers .isDeleted(deleted) => deleted.seq()
253252 }
254253 val size = graph.nodeCountByKind(nodeKind)
255- nodes .addOne(new Manifest .NodeItem (nodeLabel, size, deletions))
254+ nodeKinds .addOne(new Manifest .NodeItem (nodeLabel, size, deletions))
256255 }
257256 for {
258257 nodeKind <- graph.schema.nodeKinds
@@ -264,7 +263,7 @@ object Serialization {
264263 val nodeLabel = graph.schema.getNodeLabel(nodeKind)
265264 val edgeLabel = graph.schema.getEdgeLabel(nodeKind, edgeKind)
266265 val edgeItem = new Manifest .EdgeItem (nodeLabel, edgeLabel, direction.encoding)
267- edges .addOne(edgeItem)
266+ edgeKinds .addOne(edgeItem)
268267 writer.encodeAny(graph.neighbors(pos), edgeItem.qty, delta = graph.nodeCountByKind(nodeKind))
269268 writer.encodeAny(graph.neighbors(pos + 1 ), edgeItem.neighbors)
270269 writer.encodeAny(graph.neighbors(pos + 2 ), edgeItem.property)
@@ -279,16 +278,16 @@ object Serialization {
279278 val nodeLabel = graph.schema.getNodeLabel(nodeKind)
280279 val propertyLabel = graph.schema.getPropertyLabel(nodeKind, propertyKind)
281280 val propertyItem = new Manifest .PropertyItem (nodeLabel, propertyLabel)
282- properties .addOne(propertyItem)
281+ propertyKinds .addOne(propertyItem)
283282 writer.encodeAny(graph.properties(pos).asInstanceOf [Array [Int ]], propertyItem.qty, delta = graph.nodeCountByKind(nodeKind))
284283 writer.encodeAny(graph.properties(pos + 1 ), propertyItem.property)
285284 }
286285 }
287286
288- val manifest = new GraphItem (nodes .toArray, edges .toArray, properties .toArray)
287+ val manifest = new GraphItem (nodeKinds .toArray, edgeKinds .toArray, propertyKinds .toArray)
289288 writer.finish(manifest)
290- logger.debug(s " wrote ${nodes .size} nodes with $ {edges.size} edges and ${properties .size} properties " )
291- Counts (nodes .size, edges .size, properties .size)
289+ logger.debug(s " wrote ${nodeKinds .size} node kinds, {edges.size} edge kinds and ${propertyKinds .size} property kinds " )
290+ Counts (nodeKinds .size, edgeKinds .size, propertyKinds .size)
292291 }
293292
294293 private [flatgraph] def write (bytes : Array [Byte ], res : OutlineStorage , filePtr : AtomicLong , fileChannel : FileChannel ): OutlineStorage = {
0 commit comments