httprunner/server/ext/main.go

package server_ext

import (
	"github.com/gin-gonic/gin"
	"github.com/httprunner/httprunner/v5/server"
)

type RouterExt struct {
	*server.Router
}

type RouterBaseMethodExt struct {
	server.RouterBaseMethod
}

func NewExtRouter() *RouterExt {
	router := &RouterExt{
		Router: &server.Router{
			Engine:            gin.Default(),
			IRouterBaseMethod: &RouterBaseMethodExt{},
		},
	}
	router.Init()
	return router
}

func (r *RouterExt) Init() {
	r.Router.Init()
	apiV1PlatformSerial := r.Group("/api/v1").Group("/:platform").Group("/:serial")

	apiV1PlatformSerial.GET("/stub/source", r.sourceHandler)
	apiV1PlatformSerial.POST("/stub/login", r.loginHandler)
	apiV1PlatformSerial.POST("/stub/logout", r.logoutHandler)
	apiV1PlatformSerial.POST("/app/install", r.installAppHandler)
}