add
This commit is contained in:
parent
90014cec0c
commit
8107b7d567
|
@ -1,2 +1,10 @@
|
||||||
# gohttp
|
# gohttp
|
||||||
|
|
||||||
|
## 目标
|
||||||
|
|
||||||
|
实现一个类似nginx功能的http服务器
|
||||||
|
## 功能
|
||||||
|
|
||||||
|
- 支持静态文件
|
||||||
|
- Proxy
|
||||||
|
- 支持rewrite
|
||||||
|
|
|
@ -0,0 +1,57 @@
|
||||||
|
package admin
|
||||||
|
|
||||||
|
import (
|
||||||
|
"net/http"
|
||||||
|
"strings"
|
||||||
|
)
|
||||||
|
|
||||||
|
type Route struct {
|
||||||
|
Method string
|
||||||
|
Path string
|
||||||
|
Handle http.HandlerFunc
|
||||||
|
}
|
||||||
|
|
||||||
|
func about(w http.ResponseWriter, r *http.Request) {
|
||||||
|
w.WriteHeader(http.StatusOK)
|
||||||
|
w.Write([]byte("About Page"))
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
var AdminRoutes = []Route{
|
||||||
|
// Admin Routes
|
||||||
|
{"GET", "/about", about},
|
||||||
|
}
|
||||||
|
|
||||||
|
type PServeMux struct {
|
||||||
|
// routes map[string]map[string]http.HandlerFunc
|
||||||
|
*http.ServeMux
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *PServeMux) Use(route Route) {
|
||||||
|
method := strings.ToLower(route.Method)
|
||||||
|
p.ServeMux.HandleFunc("/"+method+route.Path, route.Handle)
|
||||||
|
// _, exist := p.routes[route.Path]
|
||||||
|
// if !exist {
|
||||||
|
// mroutes := make(map[string]http.HandlerFunc)
|
||||||
|
// p.routes[route.Path] = mroutes
|
||||||
|
// p.ServeMux.HandleFunc(route.Path, func(w http.ResponseWriter, r *http.Request) {
|
||||||
|
// p.routes[route.Path][r.Method](w, r)
|
||||||
|
// })
|
||||||
|
// }
|
||||||
|
// p.routes[route.Path][route.Method] = route.Handle
|
||||||
|
}
|
||||||
|
func (p *PServeMux) ServeHTTP(w http.ResponseWriter, r *http.Request) {
|
||||||
|
r.URL.Path = "/" + strings.ToLower(r.Method) + r.URL.Path
|
||||||
|
r.RequestURI = "/" + strings.ToLower(r.Method) + r.RequestURI
|
||||||
|
p.ServeMux.ServeHTTP(w, r)
|
||||||
|
}
|
||||||
|
|
||||||
|
var AdminServerMux *PServeMux
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
AdminServerMux = &PServeMux{ServeMux: http.NewServeMux()}
|
||||||
|
// AdminServerMux.routes = make(map[string]map[string]http.HandlerFunc)
|
||||||
|
for _, route := range AdminRoutes {
|
||||||
|
AdminServerMux.Use(route)
|
||||||
|
}
|
||||||
|
}
|
|
@ -0,0 +1,5 @@
|
||||||
|
module git.pyer.club/kingecg/gohttpd
|
||||||
|
|
||||||
|
go 1.19
|
||||||
|
|
||||||
|
require git.pyer.club/kingecg/gologger v1.0.0 // indirect
|
|
@ -0,0 +1,2 @@
|
||||||
|
git.pyer.club/kingecg/gologger v1.0.0 h1:H3oFIJ1p7mlAgJgJ/wiM+hxn34x7IxY4YiafY8iMfAk=
|
||||||
|
git.pyer.club/kingecg/gologger v1.0.0/go.mod h1:SNSl2jRHPzIpHSzdKOoVG798rtYMjPDPFyxUrEgivkY=
|
|
@ -0,0 +1,42 @@
|
||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"net/http"
|
||||||
|
|
||||||
|
admin "git.pyer.club/kingecg/gohttpd/admin"
|
||||||
|
logger "git.pyer.club/kingecg/gologger"
|
||||||
|
)
|
||||||
|
|
||||||
|
func main() {
|
||||||
|
logger.Configure(logger.LoggersConfig{
|
||||||
|
Categories: map[string]logger.LogConfig{
|
||||||
|
"default": {
|
||||||
|
Level: "debug",
|
||||||
|
Appenders: []string{
|
||||||
|
"console",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
"fat": {
|
||||||
|
Level: "debug",
|
||||||
|
Appenders: []string{
|
||||||
|
"console",
|
||||||
|
"file",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
Appenders: map[string]logger.LogAppenderConfig{
|
||||||
|
"console": {
|
||||||
|
Type: "console",
|
||||||
|
},
|
||||||
|
"file": {
|
||||||
|
Type: "file",
|
||||||
|
Options: map[string]interface{}{
|
||||||
|
"file": "./log/fat.log",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
})
|
||||||
|
defaultLogger := logger.GetLogger("default")
|
||||||
|
defaultLogger.Info("Listening...")
|
||||||
|
http.ListenAndServe(":8080", admin.AdminServerMux)
|
||||||
|
}
|
|
@ -0,0 +1,25 @@
|
||||||
|
package model
|
||||||
|
|
||||||
|
type HttpPath struct {
|
||||||
|
Path string `json:"path"`
|
||||||
|
Root string `json:"root"`
|
||||||
|
Proxyto string `json:"proxyto"`
|
||||||
|
ProxyHeaders string `json:"proxyheaders"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type HttpServerConfig struct {
|
||||||
|
ServerName string `json:"server"`
|
||||||
|
Port int `json:"port"`
|
||||||
|
Host string `json:"host"`
|
||||||
|
Paths []HttpPath
|
||||||
|
}
|
||||||
|
|
||||||
|
type GoHttpdConfig struct {
|
||||||
|
Admin HttpServerConfig `json:"admin"`
|
||||||
|
Servers []HttpServerConfig `json:"servers"`
|
||||||
|
}
|
||||||
|
|
||||||
|
var DefaultAdminConfig HttpServerConfig = HttpServerConfig{
|
||||||
|
ServerName: "admin",
|
||||||
|
Port: 8080,
|
||||||
|
}
|
|
@ -0,0 +1,17 @@
|
||||||
|
# ---> Go
|
||||||
|
# Binaries for programs and plugins
|
||||||
|
*.exe
|
||||||
|
*.exe~
|
||||||
|
*.dll
|
||||||
|
*.so
|
||||||
|
*.dylib
|
||||||
|
|
||||||
|
# Test binary, built with `go test -c`
|
||||||
|
*.test
|
||||||
|
|
||||||
|
# Output of the go coverage tool, specifically when used with LiteIDE
|
||||||
|
*.out
|
||||||
|
|
||||||
|
# Dependency directories (remove the comment below to include it)
|
||||||
|
# vendor/
|
||||||
|
|
|
@ -0,0 +1,3 @@
|
||||||
|
# gologger
|
||||||
|
|
||||||
|
a logger used in go
|
|
@ -0,0 +1,44 @@
|
||||||
|
package gologger
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
)
|
||||||
|
|
||||||
|
const (
|
||||||
|
ErrorTemplate = "\033[1;31m%s\033[0m"
|
||||||
|
WarnTemplate = "\033[1;33m%s\033[0m"
|
||||||
|
InfoTemplate = "\033[1;32m%s\033[0m"
|
||||||
|
DebugTemplate = "\033[1;34m%s\033[0m"
|
||||||
|
TraceTemplate = "\033[1;35m%s\033[0m"
|
||||||
|
)
|
||||||
|
|
||||||
|
type ConsoleAppender struct {
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *ConsoleAppender) GetName() string {
|
||||||
|
return "console"
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *ConsoleAppender) Append(logEvent LogEvent) {
|
||||||
|
|
||||||
|
logMsg := format(logEvent)
|
||||||
|
switch logEvent.Level {
|
||||||
|
case Error:
|
||||||
|
fmt.Printf(ErrorTemplate, logMsg)
|
||||||
|
case Warn:
|
||||||
|
fmt.Printf(WarnTemplate, logMsg)
|
||||||
|
case Info:
|
||||||
|
fmt.Printf(InfoTemplate, logMsg)
|
||||||
|
case Debug:
|
||||||
|
fmt.Printf(DebugTemplate, logMsg)
|
||||||
|
case Trace:
|
||||||
|
fmt.Printf(TraceTemplate, logMsg)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
func makeConsoleAppender(appenderConfig LogAppenderConfig) LoggerAppender {
|
||||||
|
var appender LoggerAppender = &ConsoleAppender{}
|
||||||
|
return appender
|
||||||
|
}
|
||||||
|
func init() {
|
||||||
|
RegistAppender("console", makeConsoleAppender)
|
||||||
|
}
|
|
@ -0,0 +1,47 @@
|
||||||
|
package gologger
|
||||||
|
|
||||||
|
import (
|
||||||
|
"os"
|
||||||
|
"path/filepath"
|
||||||
|
)
|
||||||
|
|
||||||
|
type FileAppender struct {
|
||||||
|
filePath string
|
||||||
|
file *os.File
|
||||||
|
}
|
||||||
|
|
||||||
|
func (f *FileAppender) GetName() string {
|
||||||
|
return "FileAppender:" + f.filePath
|
||||||
|
}
|
||||||
|
|
||||||
|
func (f *FileAppender) Append(logEvent LogEvent) {
|
||||||
|
if f.file == nil || int(f.file.Fd()) == -1 {
|
||||||
|
dirName := filepath.Dir(f.filePath)
|
||||||
|
_, err := os.Stat(dirName)
|
||||||
|
if err != nil && os.IsNotExist(err) {
|
||||||
|
os.MkdirAll(dirName, 0755)
|
||||||
|
}
|
||||||
|
f.file, _ = os.OpenFile(f.filePath, os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0644)
|
||||||
|
}
|
||||||
|
|
||||||
|
logMsg := format(logEvent)
|
||||||
|
f.file.WriteString(logMsg)
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
func makeFileAppender(appenderConfig LogAppenderConfig) LoggerAppender {
|
||||||
|
var logfile interface{}
|
||||||
|
var ok bool
|
||||||
|
logfile, ok = appenderConfig.Options["file"]
|
||||||
|
if !ok {
|
||||||
|
logfile = "default.log"
|
||||||
|
}
|
||||||
|
var ret LoggerAppender = &FileAppender{
|
||||||
|
filePath: logfile.(string),
|
||||||
|
}
|
||||||
|
return ret
|
||||||
|
}
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
RegistAppender("file", makeFileAppender)
|
||||||
|
}
|
|
@ -0,0 +1,23 @@
|
||||||
|
package gologger
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
)
|
||||||
|
|
||||||
|
const logTemplate = "[%s] %s - %s\n"
|
||||||
|
|
||||||
|
func format(logEvent LogEvent) string {
|
||||||
|
data := logEvent.Ts.Format("2006-01-02 15:04:05")
|
||||||
|
msg := fmt.Sprint(logEvent.Data...)
|
||||||
|
ret := fmt.Sprintf(logTemplate, data, getLogLevelStr(logEvent.Level), msg)
|
||||||
|
return ret
|
||||||
|
}
|
||||||
|
|
||||||
|
func getLogLevelStr(level int) string {
|
||||||
|
for name, slevel := range logLevelMap {
|
||||||
|
if slevel == level {
|
||||||
|
return name
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return "Unknown"
|
||||||
|
}
|
|
@ -0,0 +1,151 @@
|
||||||
|
package gologger
|
||||||
|
|
||||||
|
import (
|
||||||
|
"strings"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
const (
|
||||||
|
NoLog = iota
|
||||||
|
Error
|
||||||
|
Warn
|
||||||
|
Info
|
||||||
|
Debug
|
||||||
|
Trace
|
||||||
|
)
|
||||||
|
|
||||||
|
var logLevelMap map[string]int = map[string]int{
|
||||||
|
"off": NoLog,
|
||||||
|
"error": Error,
|
||||||
|
"warn": Warn,
|
||||||
|
"info": Info,
|
||||||
|
"debug": Debug,
|
||||||
|
"trace": Trace,
|
||||||
|
}
|
||||||
|
var loggerMap map[string]Logger = map[string]Logger{}
|
||||||
|
var appenderFactoryMap map[string]func(LogAppenderConfig) LoggerAppender = map[string]func(LogAppenderConfig) LoggerAppender{}
|
||||||
|
var appenders map[string]LoggerAppender = map[string]LoggerAppender{}
|
||||||
|
var loggerConfig LoggersConfig
|
||||||
|
|
||||||
|
type LogAppenderConfig struct {
|
||||||
|
Type string `json:"type"`
|
||||||
|
Options map[string]interface{} `json:"options"`
|
||||||
|
}
|
||||||
|
type LogConfig struct {
|
||||||
|
Level string `json:"level"`
|
||||||
|
Appenders []string `json:"appenders"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type LoggersConfig struct {
|
||||||
|
Appenders map[string]LogAppenderConfig `json:"appenders"`
|
||||||
|
Categories map[string]LogConfig `json:"categories"`
|
||||||
|
}
|
||||||
|
type Logger struct {
|
||||||
|
category string
|
||||||
|
level int
|
||||||
|
appenders []LoggerAppender
|
||||||
|
}
|
||||||
|
|
||||||
|
type LogEvent struct {
|
||||||
|
Category string
|
||||||
|
Ts time.Time
|
||||||
|
Level int
|
||||||
|
Data []interface{}
|
||||||
|
}
|
||||||
|
|
||||||
|
type LoggerAppender interface {
|
||||||
|
GetName() string
|
||||||
|
Append(logEvent LogEvent)
|
||||||
|
}
|
||||||
|
|
||||||
|
var consoleAppender LoggerAppender = &ConsoleAppender{}
|
||||||
|
var defaultLogger = &Logger{
|
||||||
|
|
||||||
|
level: Error,
|
||||||
|
appenders: []LoggerAppender{consoleAppender},
|
||||||
|
}
|
||||||
|
|
||||||
|
func (l *Logger) log(Level int, msg []interface{}) {
|
||||||
|
|
||||||
|
if Level <= l.level {
|
||||||
|
now := time.Now()
|
||||||
|
logEvent := LogEvent{l.category, now, Level, msg}
|
||||||
|
for _, appender := range l.appenders {
|
||||||
|
appender.Append(logEvent)
|
||||||
|
}
|
||||||
|
// l.Appender.Append(logEvent)
|
||||||
|
// fmt.Println(now.Format("2006-01-02 15:04:05"), " ", l.Name, ": ", msg)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (l *Logger) Error(msg ...interface{}) {
|
||||||
|
l.log(Error, msg)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (l *Logger) Warn(msg ...interface{}) {
|
||||||
|
l.log(Warn, msg)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (l *Logger) Info(msg ...interface{}) {
|
||||||
|
l.log(Info, msg)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (l *Logger) Debug(msg ...interface{}) {
|
||||||
|
l.log(Debug, msg)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (l *Logger) Trace(msg ...interface{}) {
|
||||||
|
l.log(Trace, msg)
|
||||||
|
}
|
||||||
|
func GetLogger(name string) Logger {
|
||||||
|
if logger, ok := loggerMap[name]; ok {
|
||||||
|
return logger
|
||||||
|
} else {
|
||||||
|
logConfig, ok := loggerConfig.Categories[name]
|
||||||
|
if ok {
|
||||||
|
return makeLogger(name, logConfig)
|
||||||
|
}
|
||||||
|
if name == "default" {
|
||||||
|
return *defaultLogger
|
||||||
|
}
|
||||||
|
ret := GetLogger("default")
|
||||||
|
ret.category = name
|
||||||
|
return ret
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func makeLogger(name string, config LogConfig) Logger {
|
||||||
|
logger := &Logger{category: name}
|
||||||
|
levelstr := strings.ToLower(config.Level)
|
||||||
|
logger.level = logLevelMap[levelstr]
|
||||||
|
|
||||||
|
if config.Appenders == nil || len(config.Appenders) == 0 {
|
||||||
|
logger.appenders = []LoggerAppender{consoleAppender}
|
||||||
|
} else {
|
||||||
|
logger.appenders = make([]LoggerAppender, len(config.Appenders))
|
||||||
|
for i, appenderName := range config.Appenders {
|
||||||
|
logger.appenders[i] = appenders[appenderName]
|
||||||
|
}
|
||||||
|
}
|
||||||
|
loggerMap[name] = *logger
|
||||||
|
return *logger
|
||||||
|
}
|
||||||
|
func Configure(config LoggersConfig) {
|
||||||
|
loggerConfig = config
|
||||||
|
for name, appenderConfig := range loggerConfig.Appenders {
|
||||||
|
appenderFactory, ok := appenderFactoryMap[appenderConfig.Type]
|
||||||
|
if ok {
|
||||||
|
appenders[name] = appenderFactory(appenderConfig)
|
||||||
|
} else {
|
||||||
|
appenders[name] = &ConsoleAppender{}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
for name, _ := range loggerConfig.Categories {
|
||||||
|
GetLogger(name)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func RegistAppender(typeName string, appenderCreatCb func(LogAppenderConfig) LoggerAppender) {
|
||||||
|
|
||||||
|
appenderFactoryMap[typeName] = appenderCreatCb
|
||||||
|
}
|
|
@ -0,0 +1,3 @@
|
||||||
|
# git.pyer.club/kingecg/gologger v1.0.0
|
||||||
|
## explicit; go 1.19
|
||||||
|
git.pyer.club/kingecg/gologger
|
Loading…
Reference in New Issue