add log

sters · Oct 26, 2021 · aad82ce · aad82ce
1 parent dd051be
commit aad82ce
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 4 deletions.
diff --git a/conf/conf.go b/conf/conf.go
@@ -27,6 +27,7 @@ type c struct {
 	RepositoriesDirectory             string
 	KeyDirectoryRelatedFromRepository string
 	HTTPHeaderKey                     string
+	AccessLog                         bool
 	logger                            *zap.Logger
 }
 
@@ -44,6 +45,7 @@ func Init() {
 		RepositoriesDirectory:             getenv("REPOSITORIES_DIRECTORY", "repositories/"),
 		KeyDirectoryRelatedFromRepository: getenv("KEY_DIRECTORY_RELATED_FROM_REPOSITORY", "."),
 		HTTPHeaderKey:                     getenv("HTTP_HEADER_KEY", "onstaticonstaticonstatic"),
+		AccessLog:                         getenvBool("ACCESS_LOG", true),
 		logger: logger(
 			zapcore.InfoLevel,
 			getenv("STDLOG_OUTPUT_PATH", "stdout"), // "/var/log/onstatic/stdout.log"),

diff --git a/onstatic/handler.go b/onstatic/handler.go
@@ -14,10 +14,25 @@ var fileserver http.Handler
 func RegisterHandler(s *http.ServeMux) {
 	fileserver = http.FileServer(http.Dir(getRepositoriesDir()))
 
-	s.HandleFunc("/register", handleRegister)
-	s.HandleFunc("/pull", handlePull)
-	s.HandleFunc("/unregister", handleUnregister)
-	s.HandleFunc("/", handleAll)
+	for path, handler := range map[string]http.HandlerFunc{
+		"/register":   http.HandlerFunc(handleRegister),
+		"/pull":       http.HandlerFunc(handlePull),
+		"/unregister": http.HandlerFunc(handleUnregister),
+		"/":           http.HandlerFunc(handleAll),
+	} {
+		handle := handler
+		if conf.Variables.AccessLog {
+			handle = func(rw http.ResponseWriter, r *http.Request) {
+				zap.L().Info(
+					"access log",
+					zap.String("path", r.URL.Path),
+				)
+				handler(rw, r)
+			}
+		}
+
+		s.HandleFunc(path, handle)
+	}
 }
 
 func handleRegister(res http.ResponseWriter, req *http.Request) {