1
0
mirror of https://github.com/fumiama/paper-manager.git synced 2026-06-08 17:40:23 +08:00

finish register

This commit is contained in:
源文雨
2023-03-19 21:07:22 +08:00
parent bee5caaadc
commit f3757deecf
6 changed files with 92 additions and 26 deletions

View File

@@ -6,19 +6,10 @@ import (
"github.com/sirupsen/logrus"
)
// IP gets ip from r.Header's X-FORWARDED-FOR or r.RemoteAddr
func IP(r *http.Request) string {
forwarded := r.Header.Get("X-FORWARDED-FOR")
if forwarded != "" {
return forwarded
}
return r.RemoteAddr
}
// IsMethod check if the method meets the requirement
// and response 405 Method Not Allowed if not matched
func IsMethod(m string, w http.ResponseWriter, r *http.Request) bool {
logrus.Infoln("[utils.IsMethod] accept", IP(r), r.Method, r.URL)
logrus.Infoln("[utils.IsMethod] accept", r.RemoteAddr, r.Method, r.URL)
if r.Method != m {
http.Error(w, "405 Method Not Allowed", http.StatusMethodNotAllowed)
return false