GithubBackup/httprunner
1
0
Fork 0
mirror of https://github.com/httprunner/httprunner.git synced 2026-09-07 08:27:35 +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
server/app.go
+1 -1
View File
@@ -59,7 +59,7 @@ func (r *Router) clearAppHandler(c *gin.Context) {
if err != nil { if err != nil {
return return
} }
err = driver.GetIDriver().(*uixt.ADBDriver).AppClear(appClearReq.PackageName) err = driver.GetIDriver().AppClear(appClearReq.PackageName)
if err != nil { if err != nil {
RenderError(c, err) RenderError(c, err)
return return
server/ext/context.go
+1 -1
View File
@@ -37,8 +37,8 @@ func (p RouterBaseMethodExt) GetDriver(c *gin.Context) (driverExt uixt.IXTDriver
server.RenderErrorInitDriver(c, err) server.RenderErrorInitDriver(c, err)
return return
} }
c.Set("driver", driver)
driverExt = driver_ext.NewXTDriver(driver, driverExt = driver_ext.NewXTDriver(driver,
ai.WithCVService(ai.CVServiceTypeVEDEM)) ai.WithCVService(ai.CVServiceTypeVEDEM))
c.Set("driver", driverExt)
return driverExt, nil return driverExt, nil
} }
server/main.go
+3 -3
View File
@@ -32,7 +32,7 @@ type IRouterBaseMethod interface {
} }
func (r *Router) Init() { func (r *Router) Init() {
r.Engine.Use(teardown()) r.Engine.Use(r.teardown())
r.Engine.GET("/ping", r.pingHandler) r.Engine.GET("/ping", r.pingHandler)
r.Engine.GET("/", r.pingHandler) r.Engine.GET("/", r.pingHandler)
r.Engine.POST("/", r.pingHandler) r.Engine.POST("/", r.pingHandler)
@@ -91,7 +91,7 @@ func (r *Router) pingHandler(c *gin.Context) {
RenderSuccess(c, true) RenderSuccess(c, true)
} }
func teardown() gin.HandlerFunc { func (r *Router) teardown() gin.HandlerFunc {
return func(c *gin.Context) { return func(c *gin.Context) {
logID := c.Request.Header.Get("x-tt-logid") logID := c.Request.Header.Get("x-tt-logid")
startTime := time.Now() startTime := time.Now()
@@ -107,7 +107,7 @@ func teardown() gin.HandlerFunc {
driverObj, exists := c.Get("driver") driverObj, exists := c.Get("driver")
if exists { if exists {
if driver, ok := driverObj.(*uixt.XTDriver); ok { if driver, ok := driverObj.(uixt.IXTDriver); ok {
_ = driver.TearDown() _ = driver.TearDown()
} }
} }