You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
82 lines
2.7 KiB
82 lines
2.7 KiB
package main
|
|
|
|
import (
|
|
"Open_IM/internal/demo/register"
|
|
"Open_IM/pkg/utils"
|
|
"flag"
|
|
"fmt"
|
|
"io"
|
|
"os"
|
|
"strconv"
|
|
|
|
"Open_IM/pkg/common/config"
|
|
"Open_IM/pkg/common/constant"
|
|
"Open_IM/pkg/common/log"
|
|
|
|
promePkg "Open_IM/pkg/common/prometheus"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
func main() {
|
|
log.NewPrivateLog(constant.LogFileName)
|
|
gin.SetMode(gin.ReleaseMode)
|
|
f, _ := os.Create("../logs/api.log")
|
|
gin.DefaultWriter = io.MultiWriter(f)
|
|
r := gin.Default()
|
|
r.Use(utils.CorsHandler())
|
|
if config.Config.Prometheus.Enable {
|
|
r.GET("/metrics", promePkg.PrometheusHandler())
|
|
}
|
|
authRouterGroup := r.Group("/demo")
|
|
{
|
|
authRouterGroup.POST("/code", register.SendVerificationCode)
|
|
authRouterGroup.POST("/verify", register.Verify)
|
|
authRouterGroup.POST("/password", register.SetPassword)
|
|
authRouterGroup.POST("/login", register.Login)
|
|
authRouterGroup.POST("/reset_password", register.ResetPassword)
|
|
authRouterGroup.POST("/check_login", register.CheckLoginLimit)
|
|
}
|
|
demoRouterGroup := r.Group("/auth")
|
|
{
|
|
demoRouterGroup.POST("/code", register.SendVerificationCode)
|
|
demoRouterGroup.POST("/verify", register.Verify)
|
|
demoRouterGroup.POST("/password", register.SetPassword)
|
|
demoRouterGroup.POST("/login", register.Login)
|
|
demoRouterGroup.POST("/reset_password", register.ResetPassword)
|
|
demoRouterGroup.POST("/check_login", register.CheckLoginLimit)
|
|
}
|
|
|
|
//deprecated
|
|
cmsRouterGroup := r.Group("/cms_admin")
|
|
{
|
|
cmsRouterGroup.POST("/generate_invitation_code", register.GenerateInvitationCode)
|
|
cmsRouterGroup.POST("/query_invitation_code", register.QueryInvitationCode)
|
|
cmsRouterGroup.POST("/get_invitation_codes", register.GetInvitationCodes)
|
|
|
|
cmsRouterGroup.POST("/query_user_ip_limit_login", register.QueryUserIDLimitLogin)
|
|
cmsRouterGroup.POST("/add_user_ip_limit_login", register.AddUserIPLimitLogin)
|
|
cmsRouterGroup.POST("/remove_user_ip_limit_login", register.RemoveUserIPLimitLogin)
|
|
|
|
cmsRouterGroup.POST("/query_ip_register", register.QueryIPRegister)
|
|
cmsRouterGroup.POST("/add_ip_limit", register.AddIPLimit)
|
|
cmsRouterGroup.POST("/remove_ip_Limit", register.RemoveIPLimit)
|
|
}
|
|
defaultPorts := config.Config.Demo.Port
|
|
ginPort := flag.Int("port", defaultPorts[0], "get ginServerPort from cmd,default 10004 as port")
|
|
flag.Parse()
|
|
fmt.Println("start demo api server, port: ", *ginPort)
|
|
address := "0.0.0.0:" + strconv.Itoa(*ginPort)
|
|
if config.Config.Api.ListenIP != "" {
|
|
address = config.Config.Api.ListenIP + ":" + strconv.Itoa(*ginPort)
|
|
}
|
|
address = config.Config.CmsApi.ListenIP + ":" + strconv.Itoa(*ginPort)
|
|
fmt.Println("start demo api server address: ", address)
|
|
go register.OnboardingProcessRoutine()
|
|
go register.ImportFriendRoutine()
|
|
err := r.Run(address)
|
|
if err != nil {
|
|
log.Error("", "run failed ", *ginPort, err.Error())
|
|
}
|
|
}
|