Run code before and after request?

[crazyprimes]

Problem statement

In frameworks like Flask, you can e.g. start a database transaction before the request starts, and commit/rollback when the request ends. Is there a way to do the same here? I can sort of get the start logic working by making a middleware, but I am not sure how to intercept as a final thing before the response is sent to the user?

Thanks.

[casualjim]

You can use a middleware to this end.

func beforeAndAfter(db some.DBConn, next http.Handler) http.Handler {
  return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
    // Do something before the request
   next.ServeHTTP(w, r)
    // Do something after the request
  })
}

In your case you may want to use something like httpsnoop to wrap the response writer and add some logic in there that allows you to intercept the content.