1 Star 0 Fork 0

qhitc/gweb

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
克隆/下载
serve.go 2.75 KB
一键复制 编辑 原始数据 按行查看 历史
qhitc 提交于 2023-07-14 15:12 +08:00 . 添加跨域访问
/*
Copyright © 2023 NAME HERE <EMAIL ADDRESS>
*/
package cmd
import (
"crypto/md5"
"fmt"
"net/http"
"gitee.com/qhitc_admin/gweb/internal/example/config"
"gitee.com/qhitc_admin/gweb/internal/example/grpcservice"
"gitee.com/qhitc_admin/gweb/internal/pkg/db/handler"
"gitee.com/qhitc_admin/gweb/internal/pkg/db/model"
"gitee.com/qhitc_admin/gweb/pkg/db"
"gitee.com/qhitc_admin/gweb/pkg/grpcwebserver"
"gitee.com/qhitc_admin/gweb/web/static"
"github.com/gin-gonic/gin"
log "github.com/sirupsen/logrus"
"github.com/spf13/cobra"
)
// serveCmd represents the serve command
var serveCmd = &cobra.Command{
Use: "serve",
Short: "Start server...",
Long: `Run Server`,
Run: func(cmd *cobra.Command, args []string) {
log.SetLevel(5)
log.Println("server start")
db, err := db.NewDataBase(db.SQLITE, "data.db")
if err != nil {
log.Fatalln("open database error")
}
h := handler.NewHandler(db)
h.AutoMigrate()
user, err := h.InserUser(model.User{
UserName: "admin",
PassWd: fmt.Sprintf("%x", md5.Sum([]byte("pass490312"))),
Role: "admin"})
if err != nil {
log.Println(err)
} else {
log.Println(user)
}
grpcService := grpcservice.NewService(&config.Config{})
grpcWebServer := grpcwebserver.GrpcWebServer{
StreamInterceptor: grpcService.Stream(),
UnaryInterceptor: grpcService.Unary(),
AllowAllOrigins: true,
}
grpcWebServer.Serve()
//httpBind := ":80"
httpBind := ""
httpsConf := &grpcwebserver.HttpsConf{BindAddr: ":443", CertFile: "cert.pem", KeyFile: "key.pem"}
grpcService.RegisterServers(grpcWebServer.GrpcServer)
mux := grpcService.GetRESTfulHttpMux(httpBind, httpsConf)
// 初始化一个空Gin路由
webServer := grpcwebserver.NewServer(&grpcWebServer, true)
webServer.SetRESTfulMux(mux)
go func() {
if err := webServer.Serve(httpBind, httpsConf); err != nil {
log.Errorln(err)
}
}()
log.Println("web started..")
webServer.Router.GET("/hello", func(ctx *gin.Context) {
fmt.Println("已进入到HTTP实现")
ctx.JSON(200, map[string]interface{}{
"code": 200,
"url": ctx.Request.URL,
"msg:": "success",
})
})
webServer.Router.StaticFS("/static/", http.FS(static.FS))
webServer.Router.GET("/", func(context *gin.Context) {
context.Redirect(http.StatusMovedPermanently, "/static/")
})
select {}
},
}
func init() {
rootCmd.AddCommand(serveCmd)
// Here you will define your flags and configuration settings.
// Cobra supports Persistent Flags which will work for this command
// and all subcommands, e.g.:
// serveCmd.PersistentFlags().String("foo", "", "A help for foo")
// Cobra supports local flags which will only run when this command
// is called directly, e.g.:
// serveCmd.Flags().BoolP("toggle", "t", false, "Help message for toggle")
}
Loading...
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
Go
1
https://gitee.com/qhitc_admin/gweb.git
git@gitee.com:qhitc_admin/gweb.git
qhitc_admin
gweb
gweb
v1.0.0

搜索帮助