Merge pull request #25 from DaniElectra/hierarchy

PretendoNetwork · May 20, 2023 · 41568c0 · 41568c0
2 parents 9c78ebc + 4276ce2
commit 41568c0
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/stream_in.go b/stream_in.go
@@ -98,8 +98,8 @@ func (stream *StreamIn) ReadQBuffer() ([]byte, error) {
 func (stream *StreamIn) ReadStructure(structure StructureInterface) (StructureInterface, error) {
 	hierarchy := structure.Hierarchy()
 
-	for _, class := range hierarchy {
-		_, err := stream.ReadStructure(class)
+	for _, parent := range hierarchy {
+		_, err := stream.ReadStructure(parent)
 
 		if err != nil {
 			return structure, errors.New("[ReadStructure] " + err.Error())

diff --git a/stream_out.go b/stream_out.go
@@ -99,6 +99,12 @@ func (stream *StreamOut) WriteResult(result *Result) {
 
 // WriteStructure writes a nex Structure type
 func (stream *StreamOut) WriteStructure(structure StructureInterface) {
+	parents := structure.Hierarchy()
+
+	for _, parent := range parents {
+		stream.WriteStructure(parent)
+	}
+
 	content := structure.Bytes(NewStreamOut(stream.Server))
 
 	nexVersion := stream.Server.NEXVersion()