148 lines
3 KiB
Go
148 lines
3 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"fmt"
|
|
"log"
|
|
"net/http"
|
|
"os"
|
|
"os/signal"
|
|
"strconv"
|
|
"time"
|
|
|
|
"github.com/gorilla/mux"
|
|
"go.uber.org/zap"
|
|
|
|
// _ "github.com/motemen/go-loghttp/global"
|
|
"github.com/omeid/uconfig"
|
|
"gopkg.in/yaml.v3"
|
|
)
|
|
|
|
func init() {
|
|
log.SetFlags(log.LstdFlags | log.Lshortfile)
|
|
|
|
// loghttp.DefaultLogRequest = func(req *http.Request) {
|
|
// if !strings.HasPrefix(req.URL.RequestURI(), "/api/v2/sync/maindata") {
|
|
// if false ||
|
|
// strings.HasPrefix(req.URL.RequestURI(), "/api/v2/torrents/delete") ||
|
|
// strings.HasPrefix(req.URL.RequestURI(), "/api/v2/torrents/add") {
|
|
// log.Printf("--> %s %s", req.Method, req.URL)
|
|
// for name, values := range req.Header {
|
|
// // Loop over all values for the name.
|
|
// for _, value := range values {
|
|
// log.Println(name, value)
|
|
// }
|
|
// }
|
|
// fmt.Println("")
|
|
// for k, v := range req.Form {
|
|
// log.Println(k, v)
|
|
// }
|
|
// fmt.Println("")
|
|
// }
|
|
// }
|
|
// }
|
|
|
|
// loghttp.DefaultLogResponse = func(resp *http.Response) {
|
|
// if !strings.HasPrefix(resp.Request.URL.RequestURI(), "/api/v2/sync/maindata") {
|
|
// ctx := resp.Request.Context()
|
|
// if start, ok := ctx.Value(loghttp.ContextKeyRequestStart).(time.Time); ok {
|
|
// log.Printf("<-- %d %s (%s)", resp.StatusCode, resp.Request.URL, roundtime.Duration(time.Now().Sub(start), 2))
|
|
// } else {
|
|
// log.Printf("<-- %d %s", resp.StatusCode, resp.Request.URL)
|
|
// }
|
|
// for name, values := range resp.Header {
|
|
// // Loop over all values for the name.
|
|
// for _, value := range values {
|
|
// log.Println(name, value)
|
|
// }
|
|
// }
|
|
// fmt.Println("")
|
|
// }
|
|
// }
|
|
|
|
}
|
|
|
|
func main() {
|
|
|
|
log.Println("starting up")
|
|
|
|
logger, _ := zap.NewProduction()
|
|
defer logger.Sync() // Flush buffer
|
|
|
|
conf := &Config{}
|
|
|
|
files := uconfig.Files{
|
|
{
|
|
"config.json",
|
|
json.Unmarshal,
|
|
true,
|
|
},
|
|
{
|
|
"config.yaml",
|
|
yaml.Unmarshal,
|
|
true,
|
|
},
|
|
}
|
|
|
|
_, err := uconfig.Classic(&conf, files)
|
|
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
// Router
|
|
|
|
r := mux.NewRouter()
|
|
|
|
r.PathPrefix("/").HandlerFunc(conf.HandleAll)
|
|
|
|
// r.Use(zapchi.Logger(logger, "router"))
|
|
|
|
errs := conf.Validate()
|
|
if errs != nil {
|
|
fmt.Println("Errors in config:")
|
|
for _, err := range errs {
|
|
fmt.Println(" -", err)
|
|
}
|
|
os.Exit(1)
|
|
}
|
|
|
|
errs = conf.Prime()
|
|
if errs != nil {
|
|
log.Println(errs)
|
|
}
|
|
|
|
srv := &http.Server{
|
|
Addr: conf.Multiplexer.Address + ":" + strconv.FormatUint(uint64(conf.Multiplexer.Port), 10),
|
|
WriteTimeout: time.Second * 15,
|
|
ReadTimeout: time.Second * 15,
|
|
IdleTimeout: time.Second * 60,
|
|
Handler: r,
|
|
}
|
|
|
|
go func() {
|
|
if err := srv.ListenAndServe(); err != nil {
|
|
log.Println(err)
|
|
}
|
|
}()
|
|
|
|
c := make(chan os.Signal, 1)
|
|
signal.Notify(c, os.Interrupt)
|
|
|
|
<-c
|
|
|
|
ctx, cancel := context.WithTimeout(context.Background(), conf.Multiplexer.ShutdownTimeout)
|
|
defer cancel()
|
|
srv.Shutdown(ctx)
|
|
log.Println("shutting down")
|
|
|
|
// err = ShutdownDB()
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
os.Exit(0)
|
|
|
|
}
|