iridium/neko
2
2
Fork 0
mirror of https://github.com/m1k1o/neko.git synced 2024-07-24 14:40:50 +12:00
Code Issues Packages Projects Releases Wiki Activity

add pprof.

Browse Source
This commit is contained in:
Miroslav Šedivý
2022-02-12 20:22:50 +01:00
parent d068698836
commit 89ba775a71
3 changed files with 47 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
internal/config/server.go
View File

@ -12,6 +12,7 @@ type Server struct {
Key string
Bind string
Static string
PProf bool
CORS []string
}
@ -36,6 +37,11 @@ func (Server) Init(cmd *cobra.Command) error {
return err
}
cmd.PersistentFlags().Bool("server.pprof", false, "enable pprof endpoint available at /debug/pprof")
if err := viper.BindPFlag("server.pprof", cmd.PersistentFlags().Lookup("server.pprof")); err != nil {
return err
}
cmd.PersistentFlags().StringSlice("server.cors", []string{"*"}, "list of allowed origins for CORS")
if err := viper.BindPFlag("server.cors", cmd.PersistentFlags().Lookup("server.cors")); err != nil {
return err
@ -49,6 +55,7 @@ func (s *Server) Set() {
s.Key = viper.GetString("server.key")
s.Bind = viper.GetString("server.bind")
s.Static = viper.GetString("server.static")
s.PProf = viper.GetBool("server.pprof")
s.CORS = viper.GetStringSlice("server.cors")
in, _ := utils.ArrayIn("*", s.CORS)