/
githubmirror
/
GoFrame
Обзор
Документация
Войти
/
githubmirror
/
GoFrame
Код
Запросы
0
Пакеты
0
Релизы
0
Аналитика
Безопасность
master
net/ghttp/ghttp_server_status.go
48 строк
2 KB
John Guo
replace char <xxx> to ;add GetWithEnv/GetWithCmd fuctions for package gcfg
21 окт 2021, 13:22
21 окт 2021, 13:22
fa54993
Код
Авторство
О чём код?
// Copyright GoFrame Author(https://goframe.org). 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://github.com/gogf/gf. package ghttp import ( "fmt" ) // getStatusHandler retrieves and returns the handler for given status code. func (s *Server) getStatusHandler(status int, r *Request) []HandlerFunc { domains := []string{r.GetHost(), DefaultDomainName} for _, domain := range domains { if f, ok := s.statusHandlerMap[s.statusHandlerKey(status, domain)]; ok { return f } } return nil } // addStatusHandler sets the handler for given status code. // The parameter `pattern` is like: domain#status func (s *Server) addStatusHandler(pattern string, handler HandlerFunc) { if s.statusHandlerMap[pattern] == nil { s.statusHandlerMap[pattern] = make([]HandlerFunc, 0) } s.statusHandlerMap[pattern] = append(s.statusHandlerMap[pattern], handler) } // statusHandlerKey creates and returns key for given status and domain. func (s *Server) statusHandlerKey(status int, domain string) string { return fmt.Sprintf("%s#%d", domain, status) } // BindStatusHandler registers handler for given status code. func (s *Server) BindStatusHandler(status int, handler HandlerFunc) { s.addStatusHandler(s.statusHandlerKey(status, DefaultDomainName), handler) } // BindStatusHandlerByMap registers handler for given status code using map. func (s *Server) BindStatusHandlerByMap(handlerMap map[int]HandlerFunc) { for k, v := range handlerMap { s.BindStatusHandler(k, v) } }