mirror of
https://github.com/astaxie/beego.git
synced 2024-11-22 07:00:55 +00:00
Rewrite admin service by using multiple server feature
This commit is contained in:
parent
d455805a0a
commit
2473e69417
@ -17,6 +17,7 @@ package adapter
|
|||||||
import (
|
import (
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
|
_ "github.com/astaxie/beego/pkg/infrastructure/governor"
|
||||||
"github.com/astaxie/beego/pkg/server/web"
|
"github.com/astaxie/beego/pkg/server/web"
|
||||||
)
|
)
|
||||||
|
|
||||||
@ -38,11 +39,7 @@ import (
|
|||||||
// beego.FilterMonitorFunc = MyFilterMonitor.
|
// beego.FilterMonitorFunc = MyFilterMonitor.
|
||||||
var FilterMonitorFunc func(string, string, time.Duration, string, int) bool
|
var FilterMonitorFunc func(string, string, time.Duration, string, int) bool
|
||||||
|
|
||||||
func init() {
|
|
||||||
FilterMonitorFunc = web.FilterMonitorFunc
|
|
||||||
}
|
|
||||||
|
|
||||||
// PrintTree prints all registered routers.
|
// PrintTree prints all registered routers.
|
||||||
func PrintTree() M {
|
func PrintTree() M {
|
||||||
return (M)(web.PrintTree())
|
return (M)(web.BeeApp.PrintTree())
|
||||||
}
|
}
|
||||||
|
@ -15,24 +15,12 @@
|
|||||||
package web
|
package web
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"bytes"
|
|
||||||
context2 "context"
|
|
||||||
"encoding/json"
|
|
||||||
"fmt"
|
"fmt"
|
||||||
"net/http"
|
"net/http"
|
||||||
"os"
|
|
||||||
"reflect"
|
"reflect"
|
||||||
"strconv"
|
|
||||||
"text/template"
|
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"github.com/prometheus/client_golang/prometheus/promhttp"
|
|
||||||
|
|
||||||
"github.com/astaxie/beego/pkg/infrastructure/logs"
|
"github.com/astaxie/beego/pkg/infrastructure/logs"
|
||||||
|
|
||||||
"github.com/astaxie/beego/pkg/infrastructure/governor"
|
|
||||||
"github.com/astaxie/beego/pkg/infrastructure/utils"
|
|
||||||
"github.com/astaxie/beego/pkg/server/web/grace"
|
|
||||||
"github.com/astaxie/beego/pkg/task"
|
"github.com/astaxie/beego/pkg/task"
|
||||||
)
|
)
|
||||||
|
|
||||||
@ -58,127 +46,23 @@ var beeAdminApp *adminApp
|
|||||||
var FilterMonitorFunc func(string, string, time.Duration, string, int) bool
|
var FilterMonitorFunc func(string, string, time.Duration, string, int) bool
|
||||||
|
|
||||||
func init() {
|
func init() {
|
||||||
|
c := &adminController{
|
||||||
|
servers: make([]*HttpServer, 0, 2),
|
||||||
|
}
|
||||||
beeAdminApp = &adminApp{
|
beeAdminApp = &adminApp{
|
||||||
routers: make(map[string]http.HandlerFunc),
|
HttpServer: NewHttpServerWithCfg(*BConfig),
|
||||||
}
|
}
|
||||||
// keep in mind that all data should be html escaped to avoid XSS attack
|
// keep in mind that all data should be html escaped to avoid XSS attack
|
||||||
beeAdminApp.Route("/", adminIndex)
|
beeAdminApp.Router("/", c, "get:AdminIndex")
|
||||||
beeAdminApp.Route("/qps", qpsIndex)
|
beeAdminApp.Router("/qps", c, "get:QpsIndex")
|
||||||
beeAdminApp.Route("/prof", profIndex)
|
beeAdminApp.Router("/prof", c, "get:ProfIndex")
|
||||||
beeAdminApp.Route("/healthcheck", healthcheck)
|
beeAdminApp.Router("/healthcheck", c, "get:Healthcheck")
|
||||||
beeAdminApp.Route("/task", taskStatus)
|
beeAdminApp.Router("/task", c, "get:TaskStatus")
|
||||||
beeAdminApp.Route("/listconf", listConf)
|
beeAdminApp.Router("/listconf", c, "get:ListConf")
|
||||||
beeAdminApp.Route("/metrics", promhttp.Handler().ServeHTTP)
|
beeAdminApp.Router("/metrics", c, "get:PrometheusMetrics")
|
||||||
FilterMonitorFunc = func(string, string, time.Duration, string, int) bool { return true }
|
FilterMonitorFunc = func(string, string, time.Duration, string, int) bool { return true }
|
||||||
}
|
|
||||||
|
|
||||||
// AdminIndex is the default http.Handler for admin module.
|
beeAdminApp.Run()
|
||||||
// it matches url pattern "/".
|
|
||||||
func adminIndex(rw http.ResponseWriter, _ *http.Request) {
|
|
||||||
writeTemplate(rw, map[interface{}]interface{}{}, indexTpl, defaultScriptsTpl)
|
|
||||||
}
|
|
||||||
|
|
||||||
// QpsIndex is the http.Handler for writing qps statistics map result info in http.ResponseWriter.
|
|
||||||
// it's registered with url pattern "/qps" in admin module.
|
|
||||||
func qpsIndex(rw http.ResponseWriter, _ *http.Request) {
|
|
||||||
data := make(map[interface{}]interface{})
|
|
||||||
data["Content"] = StatisticsMap.GetMap()
|
|
||||||
|
|
||||||
// do html escape before display path, avoid xss
|
|
||||||
if content, ok := (data["Content"]).(M); ok {
|
|
||||||
if resultLists, ok := (content["Data"]).([][]string); ok {
|
|
||||||
for i := range resultLists {
|
|
||||||
if len(resultLists[i]) > 0 {
|
|
||||||
resultLists[i][0] = template.HTMLEscapeString(resultLists[i][0])
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
writeTemplate(rw, data, qpsTpl, defaultScriptsTpl)
|
|
||||||
}
|
|
||||||
|
|
||||||
// ListConf is the http.Handler of displaying all beego configuration values as key/value pair.
|
|
||||||
// it's registered with url pattern "/listconf" in admin module.
|
|
||||||
func listConf(rw http.ResponseWriter, r *http.Request) {
|
|
||||||
r.ParseForm()
|
|
||||||
command := r.Form.Get("command")
|
|
||||||
if command == "" {
|
|
||||||
rw.Write([]byte("command not support"))
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
data := make(map[interface{}]interface{})
|
|
||||||
switch command {
|
|
||||||
case "conf":
|
|
||||||
m := make(M)
|
|
||||||
list("BConfig", BConfig, m)
|
|
||||||
m["AppConfigPath"] = template.HTMLEscapeString(appConfigPath)
|
|
||||||
m["AppConfigProvider"] = template.HTMLEscapeString(appConfigProvider)
|
|
||||||
tmpl := template.Must(template.New("dashboard").Parse(dashboardTpl))
|
|
||||||
tmpl = template.Must(tmpl.Parse(configTpl))
|
|
||||||
tmpl = template.Must(tmpl.Parse(defaultScriptsTpl))
|
|
||||||
|
|
||||||
data["Content"] = m
|
|
||||||
|
|
||||||
tmpl.Execute(rw, data)
|
|
||||||
|
|
||||||
case "router":
|
|
||||||
content := PrintTree()
|
|
||||||
content["Fields"] = []string{
|
|
||||||
"Router Pattern",
|
|
||||||
"Methods",
|
|
||||||
"Controller",
|
|
||||||
}
|
|
||||||
data["Content"] = content
|
|
||||||
data["Title"] = "Routers"
|
|
||||||
writeTemplate(rw, data, routerAndFilterTpl, defaultScriptsTpl)
|
|
||||||
case "filter":
|
|
||||||
var (
|
|
||||||
content = M{
|
|
||||||
"Fields": []string{
|
|
||||||
"Router Pattern",
|
|
||||||
"Filter Function",
|
|
||||||
},
|
|
||||||
}
|
|
||||||
filterTypes = []string{}
|
|
||||||
filterTypeData = make(M)
|
|
||||||
)
|
|
||||||
|
|
||||||
if BeeApp.Handlers.enableFilter {
|
|
||||||
var filterType string
|
|
||||||
for k, fr := range map[int]string{
|
|
||||||
BeforeStatic: "Before Static",
|
|
||||||
BeforeRouter: "Before Router",
|
|
||||||
BeforeExec: "Before Exec",
|
|
||||||
AfterExec: "After Exec",
|
|
||||||
FinishRouter: "Finish Router"} {
|
|
||||||
if bf := BeeApp.Handlers.filters[k]; len(bf) > 0 {
|
|
||||||
filterType = fr
|
|
||||||
filterTypes = append(filterTypes, filterType)
|
|
||||||
resultList := new([][]string)
|
|
||||||
for _, f := range bf {
|
|
||||||
var result = []string{
|
|
||||||
// void xss
|
|
||||||
template.HTMLEscapeString(f.pattern),
|
|
||||||
template.HTMLEscapeString(utils.GetFuncName(f.filterFunc)),
|
|
||||||
}
|
|
||||||
*resultList = append(*resultList, result)
|
|
||||||
}
|
|
||||||
filterTypeData[filterType] = resultList
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
content["Data"] = filterTypeData
|
|
||||||
content["Methods"] = filterTypes
|
|
||||||
|
|
||||||
data["Content"] = content
|
|
||||||
data["Title"] = "Filters"
|
|
||||||
writeTemplate(rw, data, routerAndFilterTpl, defaultScriptsTpl)
|
|
||||||
default:
|
|
||||||
rw.Write([]byte("command not support"))
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func list(root string, p interface{}, m M) {
|
func list(root string, p interface{}, m M) {
|
||||||
@ -203,239 +87,19 @@ func list(root string, p interface{}, m M) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// PrintTree prints all registered routers.
|
|
||||||
func PrintTree() M {
|
|
||||||
var (
|
|
||||||
content = M{}
|
|
||||||
methods = []string{}
|
|
||||||
methodsData = make(M)
|
|
||||||
)
|
|
||||||
for method, t := range BeeApp.Handlers.routers {
|
|
||||||
|
|
||||||
resultList := new([][]string)
|
|
||||||
|
|
||||||
printTree(resultList, t)
|
|
||||||
|
|
||||||
methods = append(methods, template.HTMLEscapeString(method))
|
|
||||||
methodsData[template.HTMLEscapeString(method)] = resultList
|
|
||||||
}
|
|
||||||
|
|
||||||
content["Data"] = methodsData
|
|
||||||
content["Methods"] = methods
|
|
||||||
return content
|
|
||||||
}
|
|
||||||
|
|
||||||
func printTree(resultList *[][]string, t *Tree) {
|
|
||||||
for _, tr := range t.fixrouters {
|
|
||||||
printTree(resultList, tr)
|
|
||||||
}
|
|
||||||
if t.wildcard != nil {
|
|
||||||
printTree(resultList, t.wildcard)
|
|
||||||
}
|
|
||||||
for _, l := range t.leaves {
|
|
||||||
if v, ok := l.runObject.(*ControllerInfo); ok {
|
|
||||||
if v.routerType == routerTypeBeego {
|
|
||||||
var result = []string{
|
|
||||||
template.HTMLEscapeString(v.pattern),
|
|
||||||
template.HTMLEscapeString(fmt.Sprintf("%s", v.methods)),
|
|
||||||
template.HTMLEscapeString(v.controllerType.String()),
|
|
||||||
}
|
|
||||||
*resultList = append(*resultList, result)
|
|
||||||
} else if v.routerType == routerTypeRESTFul {
|
|
||||||
var result = []string{
|
|
||||||
template.HTMLEscapeString(v.pattern),
|
|
||||||
template.HTMLEscapeString(fmt.Sprintf("%s", v.methods)),
|
|
||||||
"",
|
|
||||||
}
|
|
||||||
*resultList = append(*resultList, result)
|
|
||||||
} else if v.routerType == routerTypeHandler {
|
|
||||||
var result = []string{
|
|
||||||
template.HTMLEscapeString(v.pattern),
|
|
||||||
"",
|
|
||||||
"",
|
|
||||||
}
|
|
||||||
*resultList = append(*resultList, result)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// ProfIndex is a http.Handler for showing profile command.
|
|
||||||
// it's in url pattern "/prof" in admin module.
|
|
||||||
func profIndex(rw http.ResponseWriter, r *http.Request) {
|
|
||||||
r.ParseForm()
|
|
||||||
command := r.Form.Get("command")
|
|
||||||
if command == "" {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
var (
|
|
||||||
format = r.Form.Get("format")
|
|
||||||
data = make(map[interface{}]interface{})
|
|
||||||
result bytes.Buffer
|
|
||||||
)
|
|
||||||
governor.ProcessInput(command, &result)
|
|
||||||
data["Content"] = template.HTMLEscapeString(result.String())
|
|
||||||
|
|
||||||
if format == "json" && command == "gc summary" {
|
|
||||||
dataJSON, err := json.Marshal(data)
|
|
||||||
if err != nil {
|
|
||||||
http.Error(rw, err.Error(), http.StatusInternalServerError)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
writeJSON(rw, dataJSON)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
data["Title"] = template.HTMLEscapeString(command)
|
|
||||||
defaultTpl := defaultScriptsTpl
|
|
||||||
if command == "gc summary" {
|
|
||||||
defaultTpl = gcAjaxTpl
|
|
||||||
}
|
|
||||||
writeTemplate(rw, data, profillingTpl, defaultTpl)
|
|
||||||
}
|
|
||||||
|
|
||||||
// Healthcheck is a http.Handler calling health checking and showing the result.
|
|
||||||
// it's in "/healthcheck" pattern in admin module.
|
|
||||||
func healthcheck(rw http.ResponseWriter, r *http.Request) {
|
|
||||||
var (
|
|
||||||
result []string
|
|
||||||
data = make(map[interface{}]interface{})
|
|
||||||
resultList = new([][]string)
|
|
||||||
content = M{
|
|
||||||
"Fields": []string{"Name", "Message", "Status"},
|
|
||||||
}
|
|
||||||
)
|
|
||||||
|
|
||||||
for name, h := range governor.AdminCheckList {
|
|
||||||
if err := h.Check(); err != nil {
|
|
||||||
result = []string{
|
|
||||||
"error",
|
|
||||||
template.HTMLEscapeString(name),
|
|
||||||
template.HTMLEscapeString(err.Error()),
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
result = []string{
|
|
||||||
"success",
|
|
||||||
template.HTMLEscapeString(name),
|
|
||||||
"OK",
|
|
||||||
}
|
|
||||||
}
|
|
||||||
*resultList = append(*resultList, result)
|
|
||||||
}
|
|
||||||
|
|
||||||
queryParams := r.URL.Query()
|
|
||||||
jsonFlag := queryParams.Get("json")
|
|
||||||
shouldReturnJSON, _ := strconv.ParseBool(jsonFlag)
|
|
||||||
|
|
||||||
if shouldReturnJSON {
|
|
||||||
response := buildHealthCheckResponseList(resultList)
|
|
||||||
jsonResponse, err := json.Marshal(response)
|
|
||||||
|
|
||||||
if err != nil {
|
|
||||||
http.Error(rw, err.Error(), http.StatusInternalServerError)
|
|
||||||
} else {
|
|
||||||
writeJSON(rw, jsonResponse)
|
|
||||||
}
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
content["Data"] = resultList
|
|
||||||
data["Content"] = content
|
|
||||||
data["Title"] = "Health Check"
|
|
||||||
|
|
||||||
writeTemplate(rw, data, healthCheckTpl, defaultScriptsTpl)
|
|
||||||
}
|
|
||||||
|
|
||||||
func buildHealthCheckResponseList(healthCheckResults *[][]string) []map[string]interface{} {
|
|
||||||
response := make([]map[string]interface{}, len(*healthCheckResults))
|
|
||||||
|
|
||||||
for i, healthCheckResult := range *healthCheckResults {
|
|
||||||
currentResultMap := make(map[string]interface{})
|
|
||||||
|
|
||||||
currentResultMap["name"] = healthCheckResult[0]
|
|
||||||
currentResultMap["message"] = healthCheckResult[1]
|
|
||||||
currentResultMap["status"] = healthCheckResult[2]
|
|
||||||
|
|
||||||
response[i] = currentResultMap
|
|
||||||
}
|
|
||||||
|
|
||||||
return response
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
func writeJSON(rw http.ResponseWriter, jsonData []byte) {
|
func writeJSON(rw http.ResponseWriter, jsonData []byte) {
|
||||||
rw.Header().Set("Content-Type", "application/json")
|
rw.Header().Set("Content-Type", "application/json")
|
||||||
rw.Write(jsonData)
|
rw.Write(jsonData)
|
||||||
}
|
}
|
||||||
|
|
||||||
// TaskStatus is a http.Handler with running task status (task name, status and the last execution).
|
|
||||||
// it's in "/task" pattern in admin module.
|
|
||||||
func taskStatus(rw http.ResponseWriter, req *http.Request) {
|
|
||||||
data := make(map[interface{}]interface{})
|
|
||||||
|
|
||||||
// Run Task
|
|
||||||
req.ParseForm()
|
|
||||||
taskname := req.Form.Get("taskname")
|
|
||||||
if taskname != "" {
|
|
||||||
if t, ok := task.AdminTaskList[taskname]; ok {
|
|
||||||
if err := t.Run(nil); err != nil {
|
|
||||||
data["Message"] = []string{"error", template.HTMLEscapeString(fmt.Sprintf("%s", err))}
|
|
||||||
}
|
|
||||||
data["Message"] = []string{"success", template.HTMLEscapeString(fmt.Sprintf("%s run success,Now the Status is <br>%s", taskname, t.GetStatus(nil)))}
|
|
||||||
} else {
|
|
||||||
data["Message"] = []string{"warning", template.HTMLEscapeString(fmt.Sprintf("there's no task which named: %s", taskname))}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// List Tasks
|
|
||||||
content := make(M)
|
|
||||||
resultList := new([][]string)
|
|
||||||
var fields = []string{
|
|
||||||
"Task Name",
|
|
||||||
"Task Spec",
|
|
||||||
"Task Status",
|
|
||||||
"Last Time",
|
|
||||||
"",
|
|
||||||
}
|
|
||||||
for tname, tk := range task.AdminTaskList {
|
|
||||||
result := []string{
|
|
||||||
template.HTMLEscapeString(tname),
|
|
||||||
template.HTMLEscapeString(tk.GetSpec(nil)),
|
|
||||||
template.HTMLEscapeString(tk.GetStatus(nil)),
|
|
||||||
template.HTMLEscapeString(tk.GetPrev(context2.Background()).String()),
|
|
||||||
}
|
|
||||||
*resultList = append(*resultList, result)
|
|
||||||
}
|
|
||||||
|
|
||||||
content["Fields"] = fields
|
|
||||||
content["Data"] = resultList
|
|
||||||
data["Content"] = content
|
|
||||||
data["Title"] = "Tasks"
|
|
||||||
writeTemplate(rw, data, tasksTpl, defaultScriptsTpl)
|
|
||||||
}
|
|
||||||
|
|
||||||
func writeTemplate(rw http.ResponseWriter, data map[interface{}]interface{}, tpls ...string) {
|
|
||||||
tmpl := template.Must(template.New("dashboard").Parse(dashboardTpl))
|
|
||||||
for _, tpl := range tpls {
|
|
||||||
tmpl = template.Must(tmpl.Parse(tpl))
|
|
||||||
}
|
|
||||||
tmpl.Execute(rw, data)
|
|
||||||
}
|
|
||||||
|
|
||||||
// adminApp is an http.HandlerFunc map used as beeAdminApp.
|
// adminApp is an http.HandlerFunc map used as beeAdminApp.
|
||||||
type adminApp struct {
|
type adminApp struct {
|
||||||
routers map[string]http.HandlerFunc
|
*HttpServer
|
||||||
}
|
}
|
||||||
|
|
||||||
// Route adds http.HandlerFunc to adminApp with url pattern.
|
// Route adds http.HandlerFunc to adminApp with url pattern.
|
||||||
func (admin *adminApp) Route(pattern string, f http.HandlerFunc) {
|
|
||||||
admin.routers[pattern] = f
|
|
||||||
}
|
|
||||||
|
|
||||||
// Run adminApp http server.
|
|
||||||
// Its addr is defined in configuration file as adminhttpaddr and adminhttpport.
|
|
||||||
func (admin *adminApp) Run() {
|
func (admin *adminApp) Run() {
|
||||||
|
|
||||||
if len(task.AdminTaskList) > 0 {
|
if len(task.AdminTaskList) > 0 {
|
||||||
task.StartTask()
|
task.StartTask()
|
||||||
}
|
}
|
||||||
@ -444,18 +108,14 @@ func (admin *adminApp) Run() {
|
|||||||
if BConfig.Listen.AdminPort != 0 {
|
if BConfig.Listen.AdminPort != 0 {
|
||||||
addr = fmt.Sprintf("%s:%d", BConfig.Listen.AdminAddr, BConfig.Listen.AdminPort)
|
addr = fmt.Sprintf("%s:%d", BConfig.Listen.AdminAddr, BConfig.Listen.AdminPort)
|
||||||
}
|
}
|
||||||
for p, f := range admin.routers {
|
|
||||||
http.Handle(p, f)
|
|
||||||
}
|
|
||||||
logs.Info("Admin server Running on %s", addr)
|
|
||||||
|
|
||||||
var err error
|
logs.Info("Admin server Running on %s", addr)
|
||||||
if BConfig.Listen.Graceful {
|
admin.HttpServer.Run(addr)
|
||||||
err = grace.ListenAndServe(addr, nil)
|
}
|
||||||
} else {
|
|
||||||
err = http.ListenAndServe(addr, nil)
|
func registerAdmin() error {
|
||||||
}
|
if BConfig.Listen.EnableAdmin {
|
||||||
if err != nil {
|
go beeAdminApp.Run()
|
||||||
logs.Critical("Admin ListenAndServe: ", err, fmt.Sprintf("%d", os.Getpid()))
|
}
|
||||||
}
|
return nil
|
||||||
}
|
}
|
||||||
|
305
pkg/server/web/admin_controller.go
Normal file
305
pkg/server/web/admin_controller.go
Normal file
@ -0,0 +1,305 @@
|
|||||||
|
// Copyright 2020
|
||||||
|
//
|
||||||
|
// Licensed 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 web
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bytes"
|
||||||
|
context2 "context"
|
||||||
|
"encoding/json"
|
||||||
|
"fmt"
|
||||||
|
"net/http"
|
||||||
|
"strconv"
|
||||||
|
"text/template"
|
||||||
|
|
||||||
|
"github.com/prometheus/client_golang/prometheus/promhttp"
|
||||||
|
|
||||||
|
"github.com/astaxie/beego/pkg/infrastructure/governor"
|
||||||
|
"github.com/astaxie/beego/pkg/task"
|
||||||
|
)
|
||||||
|
|
||||||
|
type adminController struct {
|
||||||
|
Controller
|
||||||
|
servers []*HttpServer
|
||||||
|
}
|
||||||
|
|
||||||
|
func (a *adminController) registerHttpServer(svr *HttpServer) {
|
||||||
|
a.servers = append(a.servers, svr)
|
||||||
|
}
|
||||||
|
|
||||||
|
// ProfIndex is a http.Handler for showing profile command.
|
||||||
|
// it's in url pattern "/prof" in admin module.
|
||||||
|
func (a *adminController) ProfIndex() {
|
||||||
|
rw, r := a.Ctx.ResponseWriter, a.Ctx.Request
|
||||||
|
r.ParseForm()
|
||||||
|
command := r.Form.Get("command")
|
||||||
|
if command == "" {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
var (
|
||||||
|
format = r.Form.Get("format")
|
||||||
|
data = make(map[interface{}]interface{})
|
||||||
|
result bytes.Buffer
|
||||||
|
)
|
||||||
|
governor.ProcessInput(command, &result)
|
||||||
|
data["Content"] = template.HTMLEscapeString(result.String())
|
||||||
|
|
||||||
|
if format == "json" && command == "gc summary" {
|
||||||
|
dataJSON, err := json.Marshal(data)
|
||||||
|
if err != nil {
|
||||||
|
http.Error(rw, err.Error(), http.StatusInternalServerError)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
writeJSON(rw, dataJSON)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
data["Title"] = template.HTMLEscapeString(command)
|
||||||
|
defaultTpl := defaultScriptsTpl
|
||||||
|
if command == "gc summary" {
|
||||||
|
defaultTpl = gcAjaxTpl
|
||||||
|
}
|
||||||
|
writeTemplate(rw, data, profillingTpl, defaultTpl)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (a *adminController) PrometheusMetrics() {
|
||||||
|
promhttp.Handler().ServeHTTP(a.Ctx.ResponseWriter, a.Ctx.Request)
|
||||||
|
}
|
||||||
|
|
||||||
|
// TaskStatus is a http.Handler with running task status (task name, status and the last execution).
|
||||||
|
// it's in "/task" pattern in admin module.
|
||||||
|
func (a *adminController) TaskStatus() {
|
||||||
|
|
||||||
|
rw, req := a.Ctx.ResponseWriter, a.Ctx.Request
|
||||||
|
|
||||||
|
data := make(map[interface{}]interface{})
|
||||||
|
|
||||||
|
// Run Task
|
||||||
|
req.ParseForm()
|
||||||
|
taskname := req.Form.Get("taskname")
|
||||||
|
if taskname != "" {
|
||||||
|
if t, ok := task.AdminTaskList[taskname]; ok {
|
||||||
|
if err := t.Run(nil); err != nil {
|
||||||
|
data["Message"] = []string{"error", template.HTMLEscapeString(fmt.Sprintf("%s", err))}
|
||||||
|
}
|
||||||
|
data["Message"] = []string{"success", template.HTMLEscapeString(fmt.Sprintf("%s run success,Now the Status is <br>%s", taskname, t.GetStatus(nil)))}
|
||||||
|
} else {
|
||||||
|
data["Message"] = []string{"warning", template.HTMLEscapeString(fmt.Sprintf("there's no task which named: %s", taskname))}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// List Tasks
|
||||||
|
content := make(M)
|
||||||
|
resultList := new([][]string)
|
||||||
|
var fields = []string{
|
||||||
|
"Task Name",
|
||||||
|
"Task Spec",
|
||||||
|
"Task Status",
|
||||||
|
"Last Time",
|
||||||
|
"",
|
||||||
|
}
|
||||||
|
for tname, tk := range task.AdminTaskList {
|
||||||
|
result := []string{
|
||||||
|
template.HTMLEscapeString(tname),
|
||||||
|
template.HTMLEscapeString(tk.GetSpec(nil)),
|
||||||
|
template.HTMLEscapeString(tk.GetStatus(nil)),
|
||||||
|
template.HTMLEscapeString(tk.GetPrev(context2.Background()).String()),
|
||||||
|
}
|
||||||
|
*resultList = append(*resultList, result)
|
||||||
|
}
|
||||||
|
|
||||||
|
content["Fields"] = fields
|
||||||
|
content["Data"] = resultList
|
||||||
|
data["Content"] = content
|
||||||
|
data["Title"] = "Tasks"
|
||||||
|
writeTemplate(rw, data, tasksTpl, defaultScriptsTpl)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (a *adminController) AdminIndex() {
|
||||||
|
// AdminIndex is the default http.Handler for admin module.
|
||||||
|
// it matches url pattern "/".
|
||||||
|
writeTemplate(a.Ctx.ResponseWriter, map[interface{}]interface{}{}, indexTpl, defaultScriptsTpl)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Healthcheck is a http.Handler calling health checking and showing the result.
|
||||||
|
// it's in "/healthcheck" pattern in admin module.
|
||||||
|
func (a *adminController) Healthcheck() {
|
||||||
|
heathCheck(a.Ctx.ResponseWriter, a.Ctx.Request)
|
||||||
|
}
|
||||||
|
|
||||||
|
func heathCheck(rw http.ResponseWriter, r *http.Request) {
|
||||||
|
var (
|
||||||
|
result []string
|
||||||
|
data = make(map[interface{}]interface{})
|
||||||
|
resultList = new([][]string)
|
||||||
|
content = M{
|
||||||
|
"Fields": []string{"Name", "Message", "Status"},
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
|
for name, h := range governor.AdminCheckList {
|
||||||
|
if err := h.Check(); err != nil {
|
||||||
|
result = []string{
|
||||||
|
"error",
|
||||||
|
template.HTMLEscapeString(name),
|
||||||
|
template.HTMLEscapeString(err.Error()),
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
result = []string{
|
||||||
|
"success",
|
||||||
|
template.HTMLEscapeString(name),
|
||||||
|
"OK",
|
||||||
|
}
|
||||||
|
}
|
||||||
|
*resultList = append(*resultList, result)
|
||||||
|
}
|
||||||
|
|
||||||
|
queryParams := r.URL.Query()
|
||||||
|
jsonFlag := queryParams.Get("json")
|
||||||
|
shouldReturnJSON, _ := strconv.ParseBool(jsonFlag)
|
||||||
|
|
||||||
|
if shouldReturnJSON {
|
||||||
|
response := buildHealthCheckResponseList(resultList)
|
||||||
|
jsonResponse, err := json.Marshal(response)
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
http.Error(rw, err.Error(), http.StatusInternalServerError)
|
||||||
|
} else {
|
||||||
|
writeJSON(rw, jsonResponse)
|
||||||
|
}
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
content["Data"] = resultList
|
||||||
|
data["Content"] = content
|
||||||
|
data["Title"] = "Health Check"
|
||||||
|
|
||||||
|
writeTemplate(rw, data, healthCheckTpl, defaultScriptsTpl)
|
||||||
|
}
|
||||||
|
|
||||||
|
// QpsIndex is the http.Handler for writing qps statistics map result info in http.ResponseWriter.
|
||||||
|
// it's registered with url pattern "/qps" in admin module.
|
||||||
|
func (a *adminController) QpsIndex() {
|
||||||
|
data := make(map[interface{}]interface{})
|
||||||
|
data["Content"] = StatisticsMap.GetMap()
|
||||||
|
|
||||||
|
// do html escape before display path, avoid xss
|
||||||
|
if content, ok := (data["Content"]).(M); ok {
|
||||||
|
if resultLists, ok := (content["Data"]).([][]string); ok {
|
||||||
|
for i := range resultLists {
|
||||||
|
if len(resultLists[i]) > 0 {
|
||||||
|
resultLists[i][0] = template.HTMLEscapeString(resultLists[i][0])
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
writeTemplate(a.Ctx.ResponseWriter, data, qpsTpl, defaultScriptsTpl)
|
||||||
|
}
|
||||||
|
|
||||||
|
// ListConf is the http.Handler of displaying all beego configuration values as key/value pair.
|
||||||
|
// it's registered with url pattern "/listconf" in admin module.
|
||||||
|
func (a *adminController) ListConf() {
|
||||||
|
rw := a.Ctx.ResponseWriter
|
||||||
|
r := a.Ctx.Request
|
||||||
|
r.ParseForm()
|
||||||
|
command := r.Form.Get("command")
|
||||||
|
if command == "" {
|
||||||
|
rw.Write([]byte("command not support"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
data := make(map[interface{}]interface{})
|
||||||
|
switch command {
|
||||||
|
case "conf":
|
||||||
|
m := make(M)
|
||||||
|
list("BConfig", BConfig, m)
|
||||||
|
m["appConfigPath"] = template.HTMLEscapeString(appConfigPath)
|
||||||
|
m["appConfigProvider"] = template.HTMLEscapeString(appConfigProvider)
|
||||||
|
tmpl := template.Must(template.New("dashboard").Parse(dashboardTpl))
|
||||||
|
tmpl = template.Must(tmpl.Parse(configTpl))
|
||||||
|
tmpl = template.Must(tmpl.Parse(defaultScriptsTpl))
|
||||||
|
|
||||||
|
data["Content"] = m
|
||||||
|
|
||||||
|
tmpl.Execute(rw, data)
|
||||||
|
|
||||||
|
case "router":
|
||||||
|
content := BeeApp.PrintTree()
|
||||||
|
content["Fields"] = []string{
|
||||||
|
"Router Pattern",
|
||||||
|
"Methods",
|
||||||
|
"Controller",
|
||||||
|
}
|
||||||
|
data["Content"] = content
|
||||||
|
data["Title"] = "Routers"
|
||||||
|
writeTemplate(rw, data, routerAndFilterTpl, defaultScriptsTpl)
|
||||||
|
case "filter":
|
||||||
|
var (
|
||||||
|
content = M{
|
||||||
|
"Fields": []string{
|
||||||
|
"Router Pattern",
|
||||||
|
"Filter Function",
|
||||||
|
},
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
|
filterTypeData := BeeApp.reportFilter()
|
||||||
|
|
||||||
|
filterTypes := make([]string, 0, len(filterTypeData))
|
||||||
|
for k, _ := range filterTypeData {
|
||||||
|
filterTypes = append(filterTypes, k)
|
||||||
|
}
|
||||||
|
|
||||||
|
content["Data"] = filterTypeData
|
||||||
|
content["Methods"] = filterTypes
|
||||||
|
|
||||||
|
data["Content"] = content
|
||||||
|
data["Title"] = "Filters"
|
||||||
|
writeTemplate(rw, data, routerAndFilterTpl, defaultScriptsTpl)
|
||||||
|
default:
|
||||||
|
rw.Write([]byte("command not support"))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func writeTemplate(rw http.ResponseWriter, data map[interface{}]interface{}, tpls ...string) {
|
||||||
|
tmpl := template.Must(template.New("dashboard").Parse(dashboardTpl))
|
||||||
|
for _, tpl := range tpls {
|
||||||
|
tmpl = template.Must(tmpl.Parse(tpl))
|
||||||
|
}
|
||||||
|
tmpl.Execute(rw, data)
|
||||||
|
}
|
||||||
|
|
||||||
|
func buildHealthCheckResponseList(healthCheckResults *[][]string) []map[string]interface{} {
|
||||||
|
response := make([]map[string]interface{}, len(*healthCheckResults))
|
||||||
|
|
||||||
|
for i, healthCheckResult := range *healthCheckResults {
|
||||||
|
currentResultMap := make(map[string]interface{})
|
||||||
|
|
||||||
|
currentResultMap["name"] = healthCheckResult[0]
|
||||||
|
currentResultMap["message"] = healthCheckResult[1]
|
||||||
|
currentResultMap["status"] = healthCheckResult[2]
|
||||||
|
|
||||||
|
response[i] = currentResultMap
|
||||||
|
}
|
||||||
|
|
||||||
|
return response
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
// PrintTree print all routers
|
||||||
|
// Deprecated using BeeApp directly
|
||||||
|
func PrintTree() M {
|
||||||
|
return BeeApp.PrintTree()
|
||||||
|
}
|
@ -136,7 +136,7 @@ func TestHealthCheckHandlerDefault(t *testing.T) {
|
|||||||
|
|
||||||
w := httptest.NewRecorder()
|
w := httptest.NewRecorder()
|
||||||
|
|
||||||
handler := http.HandlerFunc(healthcheck)
|
handler := http.HandlerFunc(heathCheck)
|
||||||
|
|
||||||
handler.ServeHTTP(w, req)
|
handler.ServeHTTP(w, req)
|
||||||
|
|
||||||
@ -197,7 +197,7 @@ func TestHealthCheckHandlerReturnsJSON(t *testing.T) {
|
|||||||
|
|
||||||
w := httptest.NewRecorder()
|
w := httptest.NewRecorder()
|
||||||
|
|
||||||
handler := http.HandlerFunc(healthcheck)
|
handler := http.HandlerFunc(heathCheck)
|
||||||
|
|
||||||
handler.ServeHTTP(w, req)
|
handler.ServeHTTP(w, req)
|
||||||
if status := w.Code; status != http.StatusOK {
|
if status := w.Code; status != http.StatusOK {
|
||||||
|
@ -9,7 +9,6 @@ import (
|
|||||||
|
|
||||||
"github.com/astaxie/beego/pkg/infrastructure/logs"
|
"github.com/astaxie/beego/pkg/infrastructure/logs"
|
||||||
"github.com/astaxie/beego/pkg/infrastructure/session"
|
"github.com/astaxie/beego/pkg/infrastructure/session"
|
||||||
|
|
||||||
"github.com/astaxie/beego/pkg/server/web/context"
|
"github.com/astaxie/beego/pkg/server/web/context"
|
||||||
)
|
)
|
||||||
|
|
||||||
@ -87,13 +86,6 @@ func registerTemplate() error {
|
|||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func registerAdmin() error {
|
|
||||||
if BConfig.Listen.EnableAdmin {
|
|
||||||
go beeAdminApp.Run()
|
|
||||||
}
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func registerGzip() error {
|
func registerGzip() error {
|
||||||
if BConfig.EnableGzip {
|
if BConfig.EnableGzip {
|
||||||
context.InitGzip(
|
context.InitGzip(
|
||||||
|
@ -26,6 +26,7 @@ import (
|
|||||||
"path"
|
"path"
|
||||||
"strconv"
|
"strconv"
|
||||||
"strings"
|
"strings"
|
||||||
|
"text/template"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"golang.org/x/crypto/acme/autocert"
|
"golang.org/x/crypto/acme/autocert"
|
||||||
@ -72,6 +73,7 @@ func NewHttpServerWithCfg(cfg Config) *HttpServer {
|
|||||||
Server: &http.Server{},
|
Server: &http.Server{},
|
||||||
Cfg: cfgPtr,
|
Cfg: cfgPtr,
|
||||||
}
|
}
|
||||||
|
|
||||||
return app
|
return app
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -665,3 +667,90 @@ func (app *HttpServer) LogAccess(ctx *beecontext.Context, startTime *time.Time,
|
|||||||
}
|
}
|
||||||
logs.AccessLog(record, app.Cfg.Log.AccessLogsFormat)
|
logs.AccessLog(record, app.Cfg.Log.AccessLogsFormat)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// PrintTree prints all registered routers.
|
||||||
|
func (app *HttpServer) PrintTree() M {
|
||||||
|
var (
|
||||||
|
content = M{}
|
||||||
|
methods = []string{}
|
||||||
|
methodsData = make(M)
|
||||||
|
)
|
||||||
|
for method, t := range app.Handlers.routers {
|
||||||
|
|
||||||
|
resultList := new([][]string)
|
||||||
|
|
||||||
|
printTree(resultList, t)
|
||||||
|
|
||||||
|
methods = append(methods, template.HTMLEscapeString(method))
|
||||||
|
methodsData[template.HTMLEscapeString(method)] = resultList
|
||||||
|
}
|
||||||
|
|
||||||
|
content["Data"] = methodsData
|
||||||
|
content["Methods"] = methods
|
||||||
|
return content
|
||||||
|
}
|
||||||
|
|
||||||
|
func printTree(resultList *[][]string, t *Tree) {
|
||||||
|
for _, tr := range t.fixrouters {
|
||||||
|
printTree(resultList, tr)
|
||||||
|
}
|
||||||
|
if t.wildcard != nil {
|
||||||
|
printTree(resultList, t.wildcard)
|
||||||
|
}
|
||||||
|
for _, l := range t.leaves {
|
||||||
|
if v, ok := l.runObject.(*ControllerInfo); ok {
|
||||||
|
if v.routerType == routerTypeBeego {
|
||||||
|
var result = []string{
|
||||||
|
template.HTMLEscapeString(v.pattern),
|
||||||
|
template.HTMLEscapeString(fmt.Sprintf("%s", v.methods)),
|
||||||
|
template.HTMLEscapeString(v.controllerType.String()),
|
||||||
|
}
|
||||||
|
*resultList = append(*resultList, result)
|
||||||
|
} else if v.routerType == routerTypeRESTFul {
|
||||||
|
var result = []string{
|
||||||
|
template.HTMLEscapeString(v.pattern),
|
||||||
|
template.HTMLEscapeString(fmt.Sprintf("%s", v.methods)),
|
||||||
|
"",
|
||||||
|
}
|
||||||
|
*resultList = append(*resultList, result)
|
||||||
|
} else if v.routerType == routerTypeHandler {
|
||||||
|
var result = []string{
|
||||||
|
template.HTMLEscapeString(v.pattern),
|
||||||
|
"",
|
||||||
|
"",
|
||||||
|
}
|
||||||
|
*resultList = append(*resultList, result)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (app *HttpServer) reportFilter() M {
|
||||||
|
filterTypeData := make(M)
|
||||||
|
// filterTypes := []string{}
|
||||||
|
if app.Handlers.enableFilter {
|
||||||
|
// var filterType string
|
||||||
|
for k, fr := range map[int]string{
|
||||||
|
BeforeStatic: "Before Static",
|
||||||
|
BeforeRouter: "Before Router",
|
||||||
|
BeforeExec: "Before Exec",
|
||||||
|
AfterExec: "After Exec",
|
||||||
|
FinishRouter: "Finish Router",
|
||||||
|
} {
|
||||||
|
if bf := app.Handlers.filters[k]; len(bf) > 0 {
|
||||||
|
resultList := new([][]string)
|
||||||
|
for _, f := range bf {
|
||||||
|
var result = []string{
|
||||||
|
// void xss
|
||||||
|
template.HTMLEscapeString(f.pattern),
|
||||||
|
template.HTMLEscapeString(utils.GetFuncName(f.filterFunc)),
|
||||||
|
}
|
||||||
|
*resultList = append(*resultList, result)
|
||||||
|
}
|
||||||
|
filterTypeData[fr] = resultList
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return filterTypeData
|
||||||
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user