29 lines
391 B
Go
29 lines
391 B
Go
package api
|
|
|
|
import (
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
type APIHandler struct {
|
|
router *gin.Engine
|
|
addr []string
|
|
}
|
|
|
|
func NewAPIHandler(addr ...string) *APIHandler {
|
|
return &APIHandler{
|
|
addr: addr,
|
|
}
|
|
}
|
|
|
|
func (h *APIHandler) Init(r *gin.Engine) {
|
|
h.router = r
|
|
}
|
|
|
|
func (h *APIHandler) Run() {
|
|
h.router.Run(h.addr...)
|
|
}
|
|
|
|
func (h *APIHandler) GetRouter() *gin.Engine {
|
|
return h.router
|
|
}
|