fix panic on GetServiceType

tcncloud · Apr 13, 2023 · af97c42 · af97c42
1 parent d4632b6
commit af97c42
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/generator/service.go b/generator/service.go
@@ -80,7 +80,12 @@ func (s *Service) GetTypeMapping() *persist.TypeMapping {
 func (s *Service) GetServiceType() *persist.PersistenceOptions {
 	if s.Desc.Options != nil && proto.HasExtension(s.Desc.Options, persist.E_ServiceType) {
 		ext := proto.GetExtension(s.Desc.Options, persist.E_ServiceType)
-		return ext.(*persist.PersistenceOptions)
+		switch e := ext.(type) {
+		case persist.PersistenceOptions:
+			return &e
+		default:
+			return nil
+		}
 	}
 	return nil
 }