optimize: add global descriptor release api (#249)

Co-authored-by: 吴骏逸 <[email protected]>
cloudwego · Dec 26, 2024 · f616652 · f616652
1 parent a8df881
commit f616652
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/thrift_reflection/descriptor_register.go b/thrift_reflection/descriptor_register.go
@@ -227,6 +227,21 @@ func RegisterAST(ast *parser.Thrift) (*GlobalDescriptor, *FileDescriptor) {
 	return gd, fd
 }
 
+// ReleaseGlobalDescriptors release the global descriptor
+func ReleaseGlobalDescriptors(gds ...*GlobalDescriptor) {
+	if len(gds) == 0 {
+		return
+	}
+	lock.Lock()
+	defer lock.Unlock()
+	for _, gd := range gds {
+		if gd == nil {
+			continue
+		}
+		delete(globalDescriptorMap, gd.uuid)
+	}
+}
+
 func generateShortUUID() string {
 	uuid := make([]byte, 4)
 	_, _ = rand.Read(uuid)