GithubBackup/httprunner
1
0
Fork 0
mirror of https://github.com/httprunner/httprunner.git synced 2026-08-08 07:03:40 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: 优化driver teardown

Browse Source
This commit is contained in:
余泓铮
2025-02-20 21:39:15 +08:00
parent c6b9f17db2
commit 3b509ba6c6
3 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
server/main.go
View File

@@ -32,7 +32,7 @@ type IRouterBaseMethod interface {
}
func (r *Router) Init() {
r.Engine.Use(teardown())
r.Engine.Use(r.teardown())
r.Engine.GET("/ping", r.pingHandler)
r.Engine.GET("/", r.pingHandler)
r.Engine.POST("/", r.pingHandler)
@@ -91,7 +91,7 @@ func (r *Router) pingHandler(c *gin.Context) {
RenderSuccess(c, true)
}
func teardown() gin.HandlerFunc {
func (r *Router) teardown() gin.HandlerFunc {
return func(c *gin.Context) {
logID := c.Request.Header.Get("x-tt-logid")
startTime := time.Now()
@@ -107,7 +107,7 @@ func teardown() gin.HandlerFunc {
driverObj, exists := c.Get("driver")
if exists {
if driver, ok := driverObj.(*uixt.XTDriver); ok {
if driver, ok := driverObj.(uixt.IXTDriver); ok {
_ = driver.TearDown()
}
}