GithubBackup/httprunner
1
0
Fork 0
mirror of https://github.com/httprunner/httprunner.git synced 2026-05-12 02:21:29 +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

2
server/app.go
View File

@@ -59,7 +59,7 @@ func (r *Router) clearAppHandler(c *gin.Context) {
if err != nil {
return
}
err = driver.GetIDriver().(*uixt.ADBDriver).AppClear(appClearReq.PackageName)
err = driver.GetIDriver().AppClear(appClearReq.PackageName)
if err != nil {
RenderError(c, err)
return

2
server/ext/context.go
View File

@@ -37,8 +37,8 @@ func (p RouterBaseMethodExt) GetDriver(c *gin.Context) (driverExt uixt.IXTDriver
server.RenderErrorInitDriver(c, err)
return
}
c.Set("driver", driver)
driverExt = driver_ext.NewXTDriver(driver,
ai.WithCVService(ai.CVServiceTypeVEDEM))
c.Set("driver", driverExt)
return driverExt, nil
}

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()
}
}