Is it possible to wrap http.ServeHttp in go to add session using alexedwards/scs/v2

php hello editor Zimo! Regarding your question about wrapping http.ServeHTTP in Go to add sessions using alexedwards/scs/v2, the answer is yes. You can use the alexedwards/scs/v2 library to manage sessions and wrap http.ServeHTTP in your Go code. This way you can add session management capabilities to use sessions in your application when handling HTTP requests. This way you can easily add authentication, authorization, and other session-related functionality to your application. Hope this answer is helpful to you!

Question content

I'm trying to add a session to an existing http server written in go. I have the following code

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) {
     ...
}

I want to add session using https://pkg.go.dev/github.com/alexedwards/scs/v2#sessionmanager.loadandsave

The sample code in the link is

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

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

The example code passes mux to loadandsave and then passes the new handler to http.listenandserve(port, newhandler). In my case, the handler came from the servehttp method that I added to the *httpserver structure. The handler in the example comes from a multiplexer.

I am new here. Is it possible to pass my servehttp method to loadandsave and use the handler returned from loadandsave? If not, is there a way to pass the http.server struct fields used in my example into http.listenandserve(port, handler) ?

Solution

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

This means that the sessionmanager is created and configured outside this method (using scs.new()) and passed in when calling run.
There you can set the sessionmanager to a field in the httpserver structure.
This will allow your (s *httpserver) servehttp(writer http.responsewriter, r *http.request) to retrieve it.

For example:

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)
}

The above is the detailed content of Is it possible to wrap http.ServeHttp in go to add session using alexedwards/scs/v2. For more information, please follow other related articles on the PHP Chinese website!