fix: change server init
This commit is contained in:
parent
85ccd47bb6
commit
bf7c262d78
|
@ -12,7 +12,8 @@ app:
|
|||
upload_image_max_size: 5 # MB
|
||||
upload_image_allow_exts: ['.jpg','.jpeg','.png']
|
||||
server:
|
||||
port: 8000
|
||||
http_port: 8001
|
||||
grpc_port: 8002
|
||||
read_timeout: 60
|
||||
write_timeout: 60
|
||||
tracer_host_port: '127.0.0.1:6831'
|
||||
|
|
|
@ -1,13 +1,14 @@
|
|||
/*
|
||||
* @Date: 2021-03-22 09:54:07
|
||||
* @LastEditors: viletyy
|
||||
* @LastEditTime: 2021-06-20 19:45:31
|
||||
* @LastEditTime: 2021-07-09 14:22:47
|
||||
* @FilePath: /potato/config/server.go
|
||||
*/
|
||||
package config
|
||||
|
||||
type Server struct {
|
||||
Port int64 `mapstructure:"port" json:"port" yaml:"port"`
|
||||
HttpPort int64 `mapstructure:"http_port" json:"http_port" yaml:"http_port"`
|
||||
GrpcPort int64 `mapstructure:"grpc_port" json:"grpc_port" yaml:"grpc_port"`
|
||||
ReadTimeout int64 `mapstructure:"read_timeout" json:"read_timeout" yaml:"read_timeout"`
|
||||
WriteTimeout int64 `mapstructure:"write_timeout" json:"write_timeout" yaml:"write_timeout"`
|
||||
TracerHostPort string `mapstructure:"tracer_host_port" json:"tracer_host_port" yaml:"tracer_host_port"`
|
||||
|
|
|
@ -1,22 +1,28 @@
|
|||
/*
|
||||
* @Date: 2021-06-17 00:19:32
|
||||
* @LastEditors: viletyy
|
||||
* @LastEditTime: 2021-06-20 19:46:17
|
||||
* @LastEditTime: 2021-07-09 14:24:14
|
||||
* @FilePath: /potato/initialize/grpc_server.go
|
||||
*/
|
||||
package initialize
|
||||
|
||||
import (
|
||||
"net"
|
||||
|
||||
pb "github.com/viletyy/potato/proto/basic"
|
||||
"github.com/viletyy/potato/server/basic"
|
||||
"google.golang.org/grpc"
|
||||
"google.golang.org/grpc/reflection"
|
||||
)
|
||||
|
||||
func RunGrpcServer() *grpc.Server {
|
||||
func RunGrpcServer(port string) error {
|
||||
server := grpc.NewServer()
|
||||
pb.RegisterVendorServiceServer(server, basic.NewVendorServer())
|
||||
reflection.Register(server)
|
||||
|
||||
return server
|
||||
lis, err := net.Listen("tcp", ":"+port)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
return server.Serve(lis)
|
||||
}
|
||||
|
|
|
@ -1,12 +1,13 @@
|
|||
/*
|
||||
* @Date: 2021-03-22 17:03:27
|
||||
* @LastEditors: viletyy
|
||||
* @LastEditTime: 2021-06-20 22:22:10
|
||||
* @LastEditTime: 2021-07-09 14:24:36
|
||||
* @FilePath: /potato/initialize/http_server.go
|
||||
*/
|
||||
package initialize
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"net/http"
|
||||
"time"
|
||||
|
||||
|
@ -14,15 +15,15 @@ import (
|
|||
"github.com/viletyy/potato/internal/routers"
|
||||
)
|
||||
|
||||
func RunHttpServer(port string) *http.Server {
|
||||
func RunHttpServer(port string) error {
|
||||
router := routers.InitRouter()
|
||||
server := &http.Server{
|
||||
Addr: port,
|
||||
Addr: fmt.Sprintf(":%s", port),
|
||||
Handler: router,
|
||||
ReadTimeout: time.Duration(global.GO_CONFIG.Server.ReadTimeout) * time.Second,
|
||||
WriteTimeout: time.Duration(global.GO_CONFIG.Server.ReadTimeout) * time.Second,
|
||||
MaxHeaderBytes: 1 << 20,
|
||||
}
|
||||
|
||||
return server
|
||||
return server.ListenAndServe()
|
||||
}
|
||||
|
|
Loading…
Reference in New Issue