代码拉取完成,页面将自动刷新
/*******************************************************************************
* Copyright 2020 huanggefan.cn
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
******************************************************************************/
package main
import (
"context"
"flag"
"log"
"net"
"net/http"
"runtime"
"sync"
"time"
"gitee.com/WisdomClassroom/core/models"
"gorm.io/gorm"
"gorm.io/gorm/logger"
"gorm.io/gorm/schema"
grpcRuntime "github.com/grpc-ecosystem/grpc-gateway/runtime"
"google.golang.org/grpc"
"gorm.io/driver/postgres"
"gorm.io/driver/sqlite"
"gitee.com/WisdomClassroom/core"
"gitee.com/WisdomClassroom/core/protobufs/v1/pb"
"gitee.com/WisdomClassroom/userInfo/glb"
"gitee.com/WisdomClassroom/userInfo/service"
)
var (
BuildTag string = ""
BuildGo string = ""
BuildTime string = ""
)
func runGateway(wg *sync.WaitGroup) {
defer wg.Done()
headerMatcher := func(key string) (string, bool) {
return key, true
}
gateway := grpcRuntime.NewServeMux(
grpcRuntime.WithIncomingHeaderMatcher(headerMatcher),
grpcRuntime.WithMarshalerOption(grpcRuntime.MIMEWildcard, &grpcRuntime.JSONPb{OrigName: true, EmitDefaults: true}),
)
opts := []grpc.DialOption{grpc.WithInsecure()}
err := pb.RegisterUserServiceHandlerFromEndpoint(context.Background(), gateway, core.FlagGRPCListen, opts)
if err != nil {
glb.Logger.Error(err.Error())
return
}
mux := http.NewServeMux()
mux.Handle("/api/", gateway)
err = http.ListenAndServe(core.FlagHttpListen, mux)
if err != nil {
glb.Logger.Error(err.Error())
return
}
}
func runServer(wg *sync.WaitGroup) {
defer wg.Done()
listen, err := net.Listen("tcp", core.FlagGRPCListen)
if err != nil {
glb.Logger.Error(err.Error())
return
}
grpcServer := grpc.NewServer()
pb.RegisterUserServiceServer(grpcServer, &service.Service{})
if err = grpcServer.Serve(listen); err != nil {
glb.Logger.Error(err.Error())
return
}
}
func main() {
core.PublicFlag(BuildTag, BuildGo, BuildTime)
core.PublicServerFlag()
flag.Parse()
glb.TokenCert = core.NewTokenCert(core.FlagTokenCert)
if core.FlagHelp {
flag.Usage()
return
}
if core.FlagVersion {
core.PublicVersion()
return
}
if !core.FlagUseSqlite && !core.FlagUsePG {
flag.Usage()
return
}
if core.FlagLogWithUDP {
addr, err := net.ResolveUDPAddr("udp", core.FlagLogAddr)
if err != nil {
log.Fatalln(err)
}
glb.Logger, err = core.NewUDPLogger(core.FlagLogLevel, addr, 10240)
if err != nil {
log.Fatalln(err)
}
} else {
glb.Logger = core.NewLogger(core.FlagLogLevel)
}
if core.FlagPPROFEnable {
go core.StartPPOFDebug(http.Server{Addr: core.FlagPPROFListen}, true)
time.Sleep(1 * time.Second)
}
config := &gorm.Config{
NamingStrategy: schema.NamingStrategy{
SingularTable: true,
},
Logger: logger.New(
glb.Logger,
logger.Config{
LogLevel: core.GetGormLogLevel(),
Colorful: false,
},
),
}
var err error
if core.FlagUseSqlite {
glb.DB, err = gorm.Open(sqlite.Open(core.FlagSqliteDB), config)
} else {
glb.DB, err = gorm.Open(postgres.Open(models.GetPgSQLConf()), config)
}
if err != nil {
log.Fatalln(err)
}
glb.Logger.Info("http listen at: http://" + core.FlagHttpListen)
glb.Logger.Info("grpc listen at: grpc://" + core.FlagGRPCListen)
if core.FlagLogWithUDP {
glb.Logger.Info("log to: udp://" + core.FlagLogAddr)
}
wg := new(sync.WaitGroup)
wg.Add(2)
go runGateway(wg)
go runServer(wg)
wg.Wait()
}
func init() {
runtime.GOMAXPROCS(1)
}
此处可能存在不合适展示的内容,页面不予展示。您可通过相关编辑功能自查并修改。
如您确认内容无涉及 不当用语 / 纯广告导流 / 暴力 / 低俗色情 / 侵权 / 盗版 / 虚假 / 无价值内容或违法国家有关法律法规的内容,可点击提交进行申诉,我们将尽快为您处理。