feat: use context in Go example middleware (#865)

Author: bweston92

code-examples/protect-page-login/go/handler.go

@@ -13,7 +13,7 @@ func (app *App) dashboardHandler() http.HandlerFunc {
 			http.Error(writer, err.Error(), http.StatusInternalServerError)
 			return
 		}
-		session, err := json.Marshal(app.session)
+		session, err := json.Marshal(getSession(request.Context()))
 		if err != nil {
 			http.Error(writer, err.Error(), http.StatusInternalServerError)
 			return

code-examples/protect-page-login/go/main.go

@@ -1,45 +1,42 @@
 package main
 
 import (
-  "fmt"
-  ory "github.com/ory/client-go"
-  "net/http"
-  "os"
+	"fmt"
+	"net/http"
+	"os"
+
+	ory "github.com/ory/client-go"
 )
 
 type App struct {
-  ory *ory.APIClient
-  // save the cookies for any upstream calls to the Ory apis
-  cookies string
-  // save the session to display it on the dashboard
-  session *ory.Session
+	ory *ory.APIClient
 }
 
 func main() {
-  proxyPort := os.Getenv("PROXY_PORT")
-  if proxyPort == "" {
-    proxyPort = "4000"
-  }
-
-  // register a new Ory client with the URL set to the Ory CLI Proxy
-  // we can also read the URL from the env or a config file
-  c := ory.NewConfiguration()
-  c.Servers = ory.ServerConfigurations{{URL: fmt.Sprintf("http://localhost:%s/.ory", proxyPort)}}
-
-  app := &App{
-    ory: ory.NewAPIClient(c),
-  }
-  mux := http.NewServeMux()
-
-  // dashboard
-  mux.Handle("/", app.sessionMiddleware(app.dashboardHandler()))
-
-  port := os.Getenv("PORT")
-  if port == "" {
-    port = "3000"
-  }
-
-  fmt.Printf("Application launched and running on http://127.0.0.1:%s\n", port)
-  // start the server
-  http.ListenAndServe(":"+port, mux)
+	proxyPort := os.Getenv("PROXY_PORT")
+	if proxyPort == "" {
+		proxyPort = "4000"
+	}
+
+	// register a new Ory client with the URL set to the Ory CLI Proxy
+	// we can also read the URL from the env or a config file
+	c := ory.NewConfiguration()
+	c.Servers = ory.ServerConfigurations{{URL: fmt.Sprintf("http://localhost:%s/.ory", proxyPort)}}
+
+	app := &App{
+		ory: ory.NewAPIClient(c),
+	}
+	mux := http.NewServeMux()
+
+	// dashboard
+	mux.Handle("/", app.sessionMiddleware(app.dashboardHandler()))
+
+	port := os.Getenv("PORT")
+	if port == "" {
+		port = "3000"
+	}
+
+	fmt.Printf("Application launched and running on http://127.0.0.1:%s\n", port)
+	// start the server
+	http.ListenAndServe(":"+port, mux)
 }

code-examples/protect-page-login/go/middleware.go

@@ -1,10 +1,31 @@
 package main
 
 import (
+	"context"
 	"log"
 	"net/http"
+
+	ory "github.com/ory/client-go"
 )
 
+// save the cookies for any upstream calls to the Ory apis
+func withCookies(ctx context.Context, v string) context.Context {
+	return context.WithValue(ctx, "req.cookies", v)
+}
+
+func getCookies(ctx context.Context) string {
+	return ctx.Value("req.cookies").(string)
+}
+
+// save the session to display it on the dashboard
+func withSession(ctx context.Context, v *ory.Session) context.Context {
+	return context.WithValue(ctx, "req.session", v)
+}
+
+func getSession(ctx context.Context) *ory.Session {
+	return ctx.Value("req.session").(*ory.Session)
+}
+
 func (app *App) sessionMiddleware(next http.HandlerFunc) http.HandlerFunc {
 	return func(writer http.ResponseWriter, request *http.Request) {
 		log.Printf("handling middleware request\n")
@@ -26,10 +47,12 @@ func (app *App) sessionMiddleware(next http.HandlerFunc) http.HandlerFunc {
 			http.Redirect(writer, request, "/.ory/self-service/login/browser", http.StatusSeeOther)
 			return
 		}
-		app.cookies = cookies
-		app.session = session
+
+		ctx := withCookies(request.Context(), cookies)
+		ctx = withSession(ctx, session)
+
 		// continue to the requested page (in our case the Dashboard)
-		next.ServeHTTP(writer, request)
+		next.ServeHTTP(writer, request.WithContext(ctx))
 		return
 	}
 }