24 lines
920 B
Go
24 lines
920 B
Go
|
package router
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/fonchain_enterprise/fonchain-main/pkg/middleware"
|
||
|
|
"github.com/fonchain_enterprise/fonchain-main/pkg/service"
|
||
|
|
websocketController "github.com/fonchain_enterprise/fonchain-main/pkg/service/websocket/controller"
|
||
|
|
websocketServer "github.com/fonchain_enterprise/fonchain-main/pkg/service/websocket/server"
|
||
|
|
"github.com/gin-gonic/gin"
|
||
|
|
)
|
||
|
|
|
||
|
|
func websocketRoute(r *gin.RouterGroup) {
|
||
|
|
noAuth := r.Group("")
|
||
|
|
auth := r.Group("")
|
||
|
|
noAuth.GET("/ws/info", websocketServer.HandleWebSocket)
|
||
|
|
websocketServer.Register("similar-result", websocketController.SimilarResult)
|
||
|
|
|
||
|
|
auth.Use(middleware.CheckLogin(service.AccountProvider), middleware.CheckAuth(service.AccountProvider, service.RuleProvider), middleware.AutoLog(service.AccountProvider, service.GrpcLogImpl))
|
||
|
|
auth = auth.Group("socket")
|
||
|
|
{
|
||
|
|
auth.POST("login", websocketController.Login)
|
||
|
|
auth.POST("info", websocketController.Info)
|
||
|
|
}
|
||
|
|
}
|