mirror of
https://gitee.com/johng/gf
synced 2026-07-05 05:13:14 +08:00
50 lines
1.5 KiB
Go
50 lines
1.5 KiB
Go
|
// Copyright 2018 gf Author(https://gitee.com/johng/gf). All Rights Reserved.
|
|||
|
|
//
|
|||
|
|
// This Source Code Form is subject to the terms of the MIT License.
|
|||
|
|
// If a copy of the MIT was not distributed with this file,
|
|||
|
|
// You can obtain one at https://gitee.com/johng/gf.
|
|||
|
|
// 状态码回调函数注册.
|
|||
|
|
|
|||
|
|
package ghttp
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"fmt"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// 查询状态码回调函数
|
|||
|
|
func (s *Server)getStatusHandler(status int, r *Request) HandlerFunc {
|
|||
|
|
domains := []string{r.GetHost(), gDEFAULT_DOMAIN}
|
|||
|
|
s.hsmu.RLock()
|
|||
|
|
defer s.hsmu.RUnlock()
|
|||
|
|
for _, domain := range domains {
|
|||
|
|
if f, ok := s.statusHandlerMap[s.statusHandlerKey(status, domain)]; ok {
|
|||
|
|
return f
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
return nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 不同状态码下的回调方法处理
|
|||
|
|
// pattern格式：domain#status
|
|||
|
|
func (s *Server)setStatusHandler(pattern string, handler HandlerFunc) {
|
|||
|
|
s.hsmu.Lock()
|
|||
|
|
s.statusHandlerMap[pattern] = handler
|
|||
|
|
s.hsmu.Unlock()
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 生成状态码回调函数map存储键名
|
|||
|
|
func (s *Server)statusHandlerKey(status int, domain string) string {
|
|||
|
|
return fmt.Sprintf("%s#%d", domain, status)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 绑定指定的状态码回调函数
|
|||
|
|
func (s *Server)BindStatusHandler(status int, handler HandlerFunc) {
|
|||
|
|
s.setStatusHandler(s.statusHandlerKey(status, gDEFAULT_DOMAIN), handler)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 通过map批量绑定状态码回调函数
|
|||
|
|
func (s *Server)BindStatusHandlerByMap(handlerMap map[int]HandlerFunc) {
|
|||
|
|
for k, v := range handlerMap {
|
|||
|
|
s.BindStatusHandler(k, v)
|
|||
|
|
}
|
|||
|
|
}
|