From e00598e1f622409f9ff2ea1910d575f36586cfd0 Mon Sep 17 00:00:00 2001
From: leiliang <1354370192@qq.com>
Date: Mon, 15 Apr 2024 14:47:36 +0800
Subject: [PATCH] feat(p2) Added hessian2 enumeration support for dubbo (#88)

---
 pkg/hessian2/enumeration.go | 26 ++++++++++++++++++++++++++
 pkg/hessian2/hessian2.go    |  7 ++++++-
 2 files changed, 32 insertions(+), 1 deletion(-)
 create mode 100644 pkg/hessian2/enumeration.go

diff --git a/pkg/hessian2/enumeration.go b/pkg/hessian2/enumeration.go
new file mode 100644
index 00000000..b56a6410
--- /dev/null
+++ b/pkg/hessian2/enumeration.go
@@ -0,0 +1,26 @@
+/*
+ * Copyright 2024 CloudWeGo Authors
+ *
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements.  See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License.  You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package hessian2
+
+import (
+	hessian "github.com/apache/dubbo-go-hessian2"
+)
+
+type JavaEnum hessian.JavaEnum
diff --git a/pkg/hessian2/hessian2.go b/pkg/hessian2/hessian2.go
index c3b0d9c0..cce42460 100644
--- a/pkg/hessian2/hessian2.go
+++ b/pkg/hessian2/hessian2.go
@@ -43,6 +43,11 @@ type (
 
 func Register(pojos []interface{}) {
 	for _, i := range pojos {
-		hessian.RegisterPOJO(i.(hessian.POJO))
+		pojo, ok := i.(hessian.POJOEnum)
+		if ok {
+			hessian.RegisterJavaEnum(pojo)
+		} else {
+			hessian.RegisterPOJO(pojo)
+		}
 	}
 }