update

Author: chenjie

cgrpc/mids/mids.go

@@ -19,6 +19,10 @@ func AllMids() map[string]cgrpc.OutsideHandler {
 	return all
 }
 
+//thread unsafe
+func RegMid(name string, handler cgrpc.OutsideHandler) {
+	all[name] = handler
+}
 func rate(ctx *cgrpc.Context) {
 	if !publicmids.GrpcRate(ctx.GetPath()) {
 		ctx.Abort(cerror.ErrLimit)

crpc/mids/mids.go

@@ -18,6 +18,12 @@ func init() {
 func AllMids() map[string]crpc.OutsideHandler {
 	return all
 }
+
+//thread unsafe
+func RegMid(name string, handler crpc.OutsideHandler) {
+	all[name] = handler
+}
+
 func rate(ctx *crpc.Context) {
 	if !publicmids.CrpcRate(ctx.GetPath()) {
 		ctx.Abort(cerror.ErrLimit)

web/mids/mids.go

@@ -20,6 +20,12 @@ func init() {
 func AllMids() map[string]web.OutsideHandler {
 	return all
 }
+
+//thread unsafe
+func RegMid(name string, handler web.OutsideHandler) {
+	all[name] = handler
+}
+
 func rate(ctx *web.Context) {
 	switch ctx.GetMethod() {
 	case http.MethodGet: