add go micro_service

add k8s
This commit is contained in:
yanghao05
2023-05-15 19:58:05 +08:00
parent 0fc585e077
commit 35e8fd6ac5
6 changed files with 320 additions and 10 deletions

View File

@@ -0,0 +1,39 @@
package k8s
import (
"os"
"path/filepath"
"github.com/pkg/errors"
"github.com/rogeecn/atom/container"
"github.com/rogeecn/atom/providers/log"
"github.com/rogeecn/atom/utils/opt"
"k8s.io/client-go/kubernetes"
"k8s.io/client-go/tools/clientcmd"
)
func DefaultProvider() container.ProviderContainer {
return container.ProviderContainer{
Provider: Provide,
Options: []opt.Option{},
}
}
func Provide(opts ...opt.Option) error {
o := opt.New(opts...)
return container.Container.Provide(func() (*kubernetes.Clientset, error) {
config := filepath.Join(os.Getenv("HOME"), ".kube", "config")
if _, err := os.Stat(config); err != nil {
config = ""
} else {
log.Debugf("using kube config: %s", config)
}
clientConfig, err := clientcmd.BuildConfigFromFlags("", config)
if err != nil {
return nil, errors.Wrap(err, "failed to build config")
}
return kubernetes.NewForConfig(clientConfig)
}, o.DiOptions()...)
}

View File

@@ -0,0 +1,15 @@
package micro_service
const DefaultPrefix = "Micro"
type Config struct {
Name string
Version string
Port uint
Tls *Tls
}
type Tls struct {
Cert string
Key string
}

View File

@@ -0,0 +1,6 @@
package micro_service
type Service interface {
Serve() error
GetEngine() any
}

View File

@@ -0,0 +1,55 @@
package gomicro
import (
"github.com/rogeecn/atom/container"
"github.com/rogeecn/atom/providers/http"
"github.com/rogeecn/atom/providers/log"
"github.com/rogeecn/atom/providers/micro_service"
"github.com/rogeecn/atom/utils/opt"
"go-micro.dev/v4"
goMicro "go-micro.dev/v4"
)
func DefaultProvider() container.ProviderContainer {
return container.ProviderContainer{
Provider: Provide,
Options: []opt.Option{
opt.Prefix(http.DefaultPrefix),
},
}
}
func Provide(opts ...opt.Option) error {
o := opt.New(opts...)
var config micro_service.Config
if err := o.UnmarshalConfig(&config); err != nil {
log.Fatal(err)
}
return container.Container.Provide(func() (micro_service.Service, error) {
service := &Service{
conf: &config,
Engine: goMicro.NewService(),
}
return service, nil
}, o.DiOptions()...)
}
type Service struct {
conf *micro_service.Config
Engine goMicro.Service
}
func (s *Service) Serve() error {
return s.Engine.Run()
}
func (s *Service) GetEngine() any {
return s.Engine
}
func (s *Service) Init(f func()) {
s.Engine.Init(
micro.Name("abc"),
)
}