mirror of
https://github.com/httprunner/httprunner.git
synced 2026-05-16 12:17:38 +08:00
feat: server支持自定义公共方法
This commit is contained in:
@@ -11,7 +11,7 @@ import (
|
||||
"github.com/httprunner/httprunner/v5/server"
|
||||
)
|
||||
|
||||
func (r *RouterExt) GetDriver(c *gin.Context) (driverExt uixt.IXTDriver, err error) {
|
||||
func (p RouterBaseMethodExt) GetDriver(c *gin.Context) (driverExt uixt.IXTDriver, err error) {
|
||||
platform := c.Param("platform")
|
||||
deviceObj, exists := c.Get("device")
|
||||
var device uixt.IDevice
|
||||
|
||||
@@ -1,6 +1,7 @@
|
||||
package server_ext
|
||||
|
||||
import (
|
||||
"github.com/gin-gonic/gin"
|
||||
"github.com/httprunner/httprunner/v5/server"
|
||||
)
|
||||
|
||||
@@ -8,15 +9,22 @@ type RouterExt struct {
|
||||
*server.Router
|
||||
}
|
||||
|
||||
type RouterBaseMethodExt struct {
|
||||
server.RouterBaseMethod
|
||||
}
|
||||
|
||||
func NewExtRouter() *RouterExt {
|
||||
router := &RouterExt{
|
||||
Router: server.NewRouter(),
|
||||
Router: &server.Router{
|
||||
Engine: gin.Default(),
|
||||
IRouterBaseMethod: &RouterBaseMethodExt{},
|
||||
},
|
||||
}
|
||||
router.Setup()
|
||||
router.Init()
|
||||
return router
|
||||
}
|
||||
|
||||
func (r *RouterExt) Setup() {
|
||||
func (r *RouterExt) Init() {
|
||||
r.Router.Init()
|
||||
apiV1PlatformSerial := r.Group("/api/v1").Group("/:platform").Group("/:serial")
|
||||
|
||||
|
||||
Reference in New Issue
Block a user