AMEND: Add custom marshalling for _id

UTDNebula · Nov 7, 2023 · c8abcd5 · c8abcd5
1 parent 65eb619
commit c8abcd5
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/api/schema/objects.go b/api/schema/objects.go
@@ -1,7 +1,6 @@
 package schema
 
 import (
-	"bytes"
 	"time"
 
 	"go.mongodb.org/mongo-driver/bson/primitive"
@@ -14,8 +13,8 @@ type IdWrapper struct {
 
 // Custom JSON marshalling for ObjectID to marshal ObjectIDs correctly
 func (id IdWrapper) MarshalJSON() (data []byte, err error) {
-	jsonString := `{$oid:"` + id.String() + `"}`
-	return bytes.NewBufferString(jsonString).Bytes(), nil
+	jsonString := `{"$oid":"` + id.String() + `"}`
+	return []byte(jsonString), nil
 }
 
 type Course struct {