增加 RegisterStatuses 函数

Author: FishGoddess

HISTORY.md

@@ -1,5 +1,11 @@
 ## ✒ 历史版本的特性介绍 (Features in old versions)
 
+### v0.5.1
+
+> 此版本发布于 2023-06-19
+
+* 增加 RegisterStatuses 函数
+
 ### v0.5.0
 
 > 此版本发布于 2023-06-19

_examples/status.go

@@ -53,4 +53,10 @@ func main() {
 	// Try to remove errors.WithMsg("hello") above and run again.
 	code, msg = status.Parse(err)
 	fmt.Println(code, msg)
+
+	// Also, you can register many statuses at one time by RegisterStatuses.
+	status.RegisterStatuses(
+		status.New(123, "abc", nil),
+		status.New(666, "xxx", nil),
+	)
 }

_icons/coverage.svg

@@ -70,6 +70,16 @@ func RegisterStatus(status *Status) {
 	statusesLock.Unlock()
 }
 
+// RegisterStatuses registers some statuses to local statuses so that we can use it in Parse.
+func RegisterStatuses(registerStatus ...*Status) {
+	statusesLock.Lock()
+	defer statusesLock.Unlock()
+
+	for _, status := range registerStatus {
+		statuses = append(statuses, status)
+	}
+}
+
 // Parse parses err and returns its code and error created from msg.
 func Parse(err error) (int32, string) {
 	if err == nil {

status/status.go

@@ -25,6 +25,30 @@ func TestRegisterStatus(t *testing.T) {
 	}
 }
 
+// go test -v -cover -run=^TestRegisterStatuses$
+func TestRegisterStatuses(t *testing.T) {
+	registered := []*Status{
+		New(123, "abc", nil),
+		New(666, "xxx", nil),
+	}
+
+	RegisterStatuses(registered...)
+
+	for i := range registered {
+		found := false
+		for _, status := range statuses {
+			if status == registered[i] {
+				found = true
+				break
+			}
+		}
+
+		if !found {
+			t.Error("registered status not found", statuses)
+		}
+	}
+}
+
 func TestParse(t *testing.T) {
 	codeBadRequest := int32(http.StatusBadRequest)
 	codeForbidden := int32(http.StatusForbidden)