2013-11-13 13:11:03 +00:00
|
|
|
package beego
|
|
|
|
|
|
|
|
import (
|
|
|
|
"fmt"
|
2013-11-15 10:08:53 +00:00
|
|
|
"github.com/astaxie/beego/admin"
|
2013-11-13 13:11:03 +00:00
|
|
|
"net/http"
|
2013-11-15 10:08:53 +00:00
|
|
|
"time"
|
2013-11-13 13:11:03 +00:00
|
|
|
)
|
|
|
|
|
|
|
|
var BeeAdminApp *AdminApp
|
|
|
|
|
2013-11-15 10:08:53 +00:00
|
|
|
//func MyFilterMonitor(method, requestPath string, t time.Duration) bool {
|
|
|
|
// if method == "POST" {
|
|
|
|
// return false
|
|
|
|
// }
|
|
|
|
// if t.Nanoseconds() < 100 {
|
|
|
|
// return false
|
|
|
|
// }
|
|
|
|
// if strings.HasPrefix(requestPath, "/astaxie") {
|
|
|
|
// return false
|
|
|
|
// }
|
|
|
|
// return true
|
|
|
|
//}
|
|
|
|
|
|
|
|
//beego.FilterMonitorFunc = MyFilterMonitor
|
|
|
|
var FilterMonitorFunc func(string, string, time.Duration) bool
|
|
|
|
|
2013-11-13 13:11:03 +00:00
|
|
|
func init() {
|
|
|
|
BeeAdminApp = &AdminApp{
|
|
|
|
routers: make(map[string]http.HandlerFunc),
|
|
|
|
}
|
|
|
|
BeeAdminApp.Route("/", AdminIndex)
|
2013-11-15 10:08:53 +00:00
|
|
|
BeeAdminApp.Route("/qps", QpsIndex)
|
|
|
|
BeeAdminApp.Route("/prof", ProfIndex)
|
|
|
|
FilterMonitorFunc = func(string, string, time.Duration) bool { return true }
|
2013-11-13 13:11:03 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
func AdminIndex(rw http.ResponseWriter, r *http.Request) {
|
|
|
|
rw.Write([]byte("Welcome to Admin Dashboard"))
|
|
|
|
}
|
|
|
|
|
2013-11-15 10:08:53 +00:00
|
|
|
func QpsIndex(rw http.ResponseWriter, r *http.Request) {
|
|
|
|
info := admin.UrlMap.GetMap(rw)
|
|
|
|
}
|
|
|
|
func ProfIndex(rw http.ResponseWriter, r *http.Request) {
|
|
|
|
r.ParseForm()
|
|
|
|
command := r.Form.Get("command")
|
|
|
|
if command != "" {
|
|
|
|
admin.ProcessInput(command)
|
|
|
|
} else {
|
|
|
|
rw.Write([]byte("request url like '/prof?command=lookup goroutine'"))
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2013-11-13 13:11:03 +00:00
|
|
|
type AdminApp struct {
|
|
|
|
routers map[string]http.HandlerFunc
|
|
|
|
}
|
|
|
|
|
|
|
|
func (admin *AdminApp) Route(pattern string, f http.HandlerFunc) {
|
|
|
|
admin.routers[pattern] = f
|
|
|
|
}
|
|
|
|
|
|
|
|
func (admin *AdminApp) Run() {
|
|
|
|
addr := AdminHttpAddr
|
|
|
|
|
|
|
|
if AdminHttpPort != 0 {
|
|
|
|
addr = fmt.Sprintf("%s:%d", AdminHttpAddr, AdminHttpPort)
|
|
|
|
}
|
|
|
|
for p, f := range admin.routers {
|
|
|
|
http.Handle(p, f)
|
|
|
|
}
|
|
|
|
err := http.ListenAndServe(addr, nil)
|
|
|
|
if err != nil {
|
|
|
|
BeeLogger.Critical("Admin ListenAndServe: ", err)
|
|
|
|
}
|
|
|
|
}
|