mirror of
https://github.com/tuxdotrs/trok.git
synced 2025-07-06 03:56:33 +05:30
feat: add zerolog logger
This commit is contained in:
@ -6,6 +6,7 @@ package cmd
|
||||
import (
|
||||
"os"
|
||||
|
||||
"github.com/0xtux/trok/internal/config"
|
||||
"github.com/spf13/cobra"
|
||||
)
|
||||
|
||||
@ -15,6 +16,10 @@ var rootCmd = &cobra.Command{
|
||||
Long: "Simple TCP tunnel in Go that exposes local ports to internet, bypassing NAT firewalls.",
|
||||
}
|
||||
|
||||
func init() {
|
||||
config.InitLogger()
|
||||
}
|
||||
|
||||
func Execute() {
|
||||
rootCmd.CompletionOptions.HiddenDefaultCmd = true
|
||||
err := rootCmd.Execute()
|
||||
|
@ -5,9 +5,9 @@ Copyright © 2024 tux <0xtux@pm.me>
|
||||
package client
|
||||
|
||||
import (
|
||||
"log"
|
||||
"github.com/rs/zerolog/log"
|
||||
)
|
||||
|
||||
func Start(port uint16) {
|
||||
log.Printf("Hello from client: %d", port)
|
||||
log.Info().Msgf("Hello from client: %d", port)
|
||||
}
|
||||
|
30
internal/config/logger.go
Normal file
30
internal/config/logger.go
Normal file
@ -0,0 +1,30 @@
|
||||
/*
|
||||
Copyright © 2024 tux <0xtux@pm.me>
|
||||
*/
|
||||
|
||||
package config
|
||||
|
||||
import (
|
||||
"os"
|
||||
"time"
|
||||
|
||||
"github.com/rs/zerolog"
|
||||
"github.com/rs/zerolog/log"
|
||||
)
|
||||
|
||||
func InitLogger() {
|
||||
output := zerolog.ConsoleWriter{
|
||||
Out: os.Stdout,
|
||||
TimeFormat: time.RFC3339,
|
||||
NoColor: false,
|
||||
}
|
||||
|
||||
log.Logger = zerolog.New(output).
|
||||
With().
|
||||
Timestamp().
|
||||
Caller().
|
||||
Stack().
|
||||
Logger()
|
||||
|
||||
zerolog.SetGlobalLevel(zerolog.DebugLevel)
|
||||
}
|
@ -5,9 +5,9 @@ Copyright © 2024 tux <0xtux@pm.me>
|
||||
package server
|
||||
|
||||
import (
|
||||
"log"
|
||||
"github.com/rs/zerolog/log"
|
||||
)
|
||||
|
||||
func Start(port uint16) {
|
||||
log.Printf("Hello from server: %d", port)
|
||||
log.Info().Msgf("Hello from server: %d", port)
|
||||
}
|
||||
|
Reference in New Issue
Block a user