Golang Http Middleware 判断 增加cookie

本文介绍了一个使用Go语言编写的简单Web应用,该应用包含一个认证中间件,用于处理用户的身份验证过程。通过检查URL参数或Cookies中的用户名和密码来决定是否允许访问资源。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >


源码:

package main

import (
	"fmt"
	"net/http"

	. "github.com/soekchl/myUtils"
)

func main() {
	finalHandler := http.HandlerFunc(final) // 设定最后访问

	http.Handle("/", middleware(finalHandler)) // 设置中间件

	if err := http.ListenAndServe(":8088", nil); err != nil {
		Error(err)
		return
	}
}

func final(w http.ResponseWriter, r *http.Request) {
	Info("final handler")
	fmt.Fprintln(w, "final")
}

func middleware(next http.Handler) http.Handler {
	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		Info("middleware handler")

		id := r.FormValue("id") // 判断是否有 id 和 passwd 变量
		passwd := r.FormValue("passwd")

		if id == "" { // 没有先从cookie 获取
			c, err := r.Cookie("id")
			if err == nil {
				id = c.Value
			}
		}

		if passwd == "" {
			c, err := r.Cookie("passwd")
			if err == nil {
				passwd = c.Value
			}
		}

		if id == "admin" && passwd == "123456" { // 账号密码判断
			cookie := http.Cookie{Name: "id", Value: id, Path: "/", MaxAge: 86400}
			http.SetCookie(w, &cookie)
			cookie = http.Cookie{Name: "passwd", Value: passwd, Path: "/", MaxAge: 86400}
			http.SetCookie(w, &cookie)

			next.ServeHTTP(w, r) // 跳转
		} else {
			fmt.Fprintln(w, "Id or Passwd is Error!") // 不满足条件
		}

		Info("middleware over") // 中间件结束
	})
}


访问地址:http://localhost:8088/?id=admin&passwd=123456

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值