使用 alexedwards/scs/v2 为 go 中的 http.ServeHttp 添加会话包装

学习Golang要努力，但是不要急！今天的这篇文章《使用 alexedwards/scs/v2 为 go 中的 http.ServeHttp 添加会话包装》将会介绍到等等知识点，如果你想深入学习Golang，可以关注我！我会持续更新相关文章的，希望对大家都能有所帮助！

我正在尝试将会话添加到用 go 编写的现有 http 服务器。我有如下代码

type httpserver struct {
    getroutes  map[string]http.handlerfunc // pattern => handler
    postroutes map[string]http.handlerfunc
    server     http.server
}
func (s *httpserver) run() {
    address := "127.0.0.1:8080"
    s.server = http.server{
        addr:           address,
        handler:        s,
        readtimeout:    10 * time.second,
        writetimeout:   10 * time.second,
        maxheaderbytes: 1 << 20,
    }
    log.fatal(s.server.listenandserve())
}
func (s *httpserver) servehttp(writer http.responsewriter, r *http.request) {
     ...
}

我想使用添加会话 https://pkg.go.dev/github.com/alexedwards/scs/v2#sessionmanager.loadandsave

链接中的示例代码是

mux := http.NewServeMux()
    mux.HandleFunc("/put", putHandler)
    mux.HandleFunc("/get", getHandler)

    // Wrap your handlers with the LoadAndSave() middleware.
    http.ListenAndServe(":4000", sessionManager.LoadAndSave(mux))

示例代码将 mux 传递到 loadandsave，然后将新处理程序传递到 http.listenandserve(port, newhandler)。就我而言，处理程序来自我添加到 *httpserver 结构中的 servehttp 方法。示例中的处理程序来自多路复用器。

我是新来的。是否可以将我的 servehttp 方法传递到 loadandsave 并使用从 loadandsave 返回的处理程序？如果没有，有没有办法将我的示例中使用的 http.server 结构字段传递到 http.listenandserve(port, handler) 中？

正确答案

正如 peter 的 func (s *httpserver) run(sessionmanager *scs.sessionmanager) { address := "127.0.0.1:8080" // wrap your httpserver with the loadandsave middleware. handlerwithsessions := sessionmanager.loadandsave(s) ... }

这意味着在此方法之外创建和配置 sessionmanager （使用 scs.new())，并在调用 run 时传入。
在那里，您可以将 sessionmanager 设置为 httpserver 结构中的字段。
这将允许您的 (s *httpserver) servehttp(writer http.responsewriter, r *http.request) 检索它。

例如：

package main

import (
    "github.com/alexedwards/scs/v2"
    "log"
    "net/http"
    "time"
)

type HttpServer struct {
    getRoutes     map[string]http.HandlerFunc // pattern => handler
    postRoutes    map[string]http.HandlerFunc
    server        http.Server
    sessionManager *scs.SessionManager
}

func (s *HttpServer) Run(sessionManager *scs.SessionManager) {
    address := "127.0.0.1:8080"
    
    // Set the sessionManager field
    s.sessionManager = sessionManager
    
    // Wrap your HttpServer with the LoadAndSave middleware.
    handlerWithSessions := sessionManager.LoadAndSave(s)
    
    s.server = http.Server{
        Addr:           address,
        Handler:        handlerWithSessions, // Use the wrapped handler
        ReadTimeout:    10 * time.Second,
        WriteTimeout:   10 * time.Second,
        MaxHeaderBytes: 1 << 20,
    }
    
    log.Fatal(s.server.ListenAndServe())
}

func (s *HttpServer) ServeHTTP(writer http.ResponseWriter, r *http.Request) {
    // Access the session via the sessionManager field
    session := s.sessionManager.Load(r)
    
    // Use the session, e.g. session.Put, session.Get, etc.
    // ...
}

func main() {
    // Create and configure the session manager
    sessionManager := scs.New()
    sessionManager.Lifetime = 24 * time.Hour
    
    // Create your custom HttpServer
    httpServer := &HttpServer{
        getRoutes:  make(map[string]http.HandlerFunc),
        postRoutes: make(map[string]http.HandlerFunc),
    }
    
    // Start the server with the session manager
    httpServer.Run(sessionManager)
}

好了，本文到此结束，带大家了解了《使用 alexedwards/scs/v2 为 go 中的 http.ServeHttp 添加会话包装》，希望本文对你有所帮助！关注golang学习网公众号，给大家分享更多Golang知识！