ddmt/gitdts
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Make modules/context.Context a context.Context (#16031)

Browse Source 
* Make modules/context.Context a context.Context

Signed-off-by: Andrew Thornton <art27@cantab.net>

* Simplify context calls

Signed-off-by: Andrew Thornton <art27@cantab.net>

* Set the base context for requests to the HammerContext

Signed-off-by: Andrew Thornton <art27@cantab.net>

Co-authored-by: Lunny Xiao <xiaolunwen@gmail.com>
This commit is contained in:
zeripath
2021-05-31 07:18:11 +01:00
committed by GitHub
parent 518ed504ef
commit 3183a465d7
14 changed files with 46 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
routers/install.go
View File

@@ -400,7 +400,7 @@ func InstallPost(ctx *context.Context) {
}
// Re-read settings
PostInstallInit(ctx.Req.Context())
PostInstallInit(ctx)
// Create admin account
if len(form.AdminName) > 0 {
@@ -454,7 +454,7 @@ func InstallPost(ctx *context.Context) {
// Now get the http.Server from this request and shut it down
// NB: This is not our hammerable graceful shutdown this is http.Server.Shutdown
srv := ctx.Req.Context().Value(http.ServerContextKey).(*http.Server)
srv := ctx.Value(http.ServerContextKey).(*http.Server)
go func() {
if err := srv.Shutdown(graceful.GetManager().HammerContext()); err != nil {
log.Error("Unable to shutdown the install server! Error: %v", err)