mirror of
https://github.com/zhufuyi/sponge.git
synced 2025-09-27 04:55:57 +08:00
101 lines
2.6 KiB
Go
101 lines
2.6 KiB
Go
package initial
|
|
|
|
import (
|
|
"fmt"
|
|
"strconv"
|
|
"time"
|
|
|
|
"github.com/zhufuyi/sponge/internal/config"
|
|
"github.com/zhufuyi/sponge/internal/server"
|
|
|
|
"github.com/zhufuyi/sponge/pkg/app"
|
|
"github.com/zhufuyi/sponge/pkg/logger"
|
|
"github.com/zhufuyi/sponge/pkg/servicerd/registry"
|
|
"github.com/zhufuyi/sponge/pkg/servicerd/registry/consul"
|
|
"github.com/zhufuyi/sponge/pkg/servicerd/registry/etcd"
|
|
"github.com/zhufuyi/sponge/pkg/servicerd/registry/nacos"
|
|
)
|
|
|
|
// CreateServices create grpc or http service
|
|
func CreateServices() []app.IServer {
|
|
var cfg = config.Get()
|
|
var servers []app.IServer
|
|
|
|
// creating grpc service
|
|
grpcAddr := ":" + strconv.Itoa(cfg.Grpc.Port)
|
|
grpcRegistry, grpcInstance := registerService("grpc", cfg.App.Host, cfg.Grpc.Port)
|
|
grpcServer := server.NewGRPCServer(grpcAddr,
|
|
server.WithGrpcReadTimeout(time.Duration(cfg.Grpc.ReadTimeout)*time.Second),
|
|
server.WithGrpcWriteTimeout(time.Duration(cfg.Grpc.WriteTimeout)*time.Second),
|
|
server.WithGrpcRegistry(grpcRegistry, grpcInstance),
|
|
)
|
|
servers = append(servers, grpcServer)
|
|
|
|
return servers
|
|
}
|
|
|
|
func registerService(scheme string, host string, port int) (registry.Registry, *registry.ServiceInstance) {
|
|
var (
|
|
instanceEndpoint = fmt.Sprintf("%s://%s:%d", scheme, host, port)
|
|
cfg = config.Get()
|
|
|
|
iRegistry registry.Registry
|
|
instance *registry.ServiceInstance
|
|
err error
|
|
|
|
id = cfg.App.Name + "_" + scheme + "_" + host
|
|
logField logger.Field
|
|
)
|
|
|
|
switch cfg.App.RegistryDiscoveryType {
|
|
// registering service with consul
|
|
case "consul":
|
|
iRegistry, instance, err = consul.NewRegistry(
|
|
cfg.Consul.Addr,
|
|
id,
|
|
cfg.App.Name,
|
|
[]string{instanceEndpoint},
|
|
)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
logField = logger.Any("consulAddress", cfg.Consul.Addr)
|
|
|
|
// registering service with etcd
|
|
case "etcd":
|
|
iRegistry, instance, err = etcd.NewRegistry(
|
|
cfg.Etcd.Addrs,
|
|
id,
|
|
cfg.App.Name,
|
|
[]string{instanceEndpoint},
|
|
)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
logField = logger.Any("etcdAddress", cfg.Etcd.Addrs)
|
|
|
|
// registering service with nacos
|
|
case "nacos":
|
|
iRegistry, instance, err = nacos.NewRegistry(
|
|
cfg.NacosRd.IPAddr,
|
|
cfg.NacosRd.Port,
|
|
cfg.NacosRd.NamespaceID,
|
|
id,
|
|
cfg.App.Name,
|
|
[]string{instanceEndpoint},
|
|
)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
logField = logger.String("nacosAddress", fmt.Sprintf("%v:%d", cfg.NacosRd.IPAddr, cfg.NacosRd.Port))
|
|
}
|
|
|
|
if instance != nil {
|
|
msg := fmt.Sprintf("register service address to %s", cfg.App.RegistryDiscoveryType)
|
|
logger.Info(msg, logField, logger.String("id", id), logger.String("name", cfg.App.Name), logger.String("endpoint", instanceEndpoint))
|
|
return iRegistry, instance
|
|
}
|
|
|
|
return nil, nil
|
|
}
|