mirror of
https://github.com/go-micro/go-micro.git
synced 2024-11-30 08:06:40 +02:00
198 lines
4.1 KiB
Go
198 lines
4.1 KiB
Go
package service
|
|
|
|
import (
|
|
"context"
|
|
"sync"
|
|
|
|
"github.com/micro/go-micro/v2/client"
|
|
"github.com/micro/go-micro/v2/runtime"
|
|
pb "github.com/micro/go-micro/v2/runtime/service/proto"
|
|
)
|
|
|
|
type svc struct {
|
|
sync.RWMutex
|
|
options runtime.Options
|
|
runtime pb.RuntimeService
|
|
}
|
|
|
|
// Init initializes runtime with given options
|
|
func (s *svc) Init(opts ...runtime.Option) error {
|
|
s.Lock()
|
|
defer s.Unlock()
|
|
|
|
for _, o := range opts {
|
|
o(&s.options)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// Create registers a service in the runtime
|
|
func (s *svc) Create(svc *runtime.Service, opts ...runtime.CreateOption) error {
|
|
options := runtime.CreateOptions{}
|
|
// apply requested options
|
|
for _, o := range opts {
|
|
o(&options)
|
|
}
|
|
|
|
// set the default source from MICRO_RUNTIME_SOURCE
|
|
if len(svc.Source) == 0 {
|
|
svc.Source = s.options.Source
|
|
}
|
|
|
|
// runtime service create request
|
|
req := &pb.CreateRequest{
|
|
Service: &pb.Service{
|
|
Name: svc.Name,
|
|
Version: svc.Version,
|
|
Source: svc.Source,
|
|
Metadata: svc.Metadata,
|
|
},
|
|
Options: &pb.CreateOptions{
|
|
Command: options.Command,
|
|
Args: options.Args,
|
|
Env: options.Env,
|
|
Type: options.Type,
|
|
Image: options.Image,
|
|
},
|
|
}
|
|
|
|
if _, err := s.runtime.Create(context.Background(), req); err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// Read returns the service with the given name from the runtime
|
|
func (s *svc) Read(opts ...runtime.ReadOption) ([]*runtime.Service, error) {
|
|
options := runtime.ReadOptions{}
|
|
// apply requested options
|
|
for _, o := range opts {
|
|
o(&options)
|
|
}
|
|
|
|
// runtime service create request
|
|
req := &pb.ReadRequest{
|
|
Options: &pb.ReadOptions{
|
|
Service: options.Service,
|
|
Version: options.Version,
|
|
Type: options.Type,
|
|
},
|
|
}
|
|
|
|
resp, err := s.runtime.Read(context.Background(), req)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
services := make([]*runtime.Service, 0, len(resp.Services))
|
|
for _, service := range resp.Services {
|
|
svc := &runtime.Service{
|
|
Name: service.Name,
|
|
Version: service.Version,
|
|
Source: service.Source,
|
|
Metadata: service.Metadata,
|
|
}
|
|
services = append(services, svc)
|
|
}
|
|
|
|
return services, nil
|
|
}
|
|
|
|
// Update updates the running service
|
|
func (s *svc) Update(svc *runtime.Service) error {
|
|
// runtime service create request
|
|
req := &pb.UpdateRequest{
|
|
Service: &pb.Service{
|
|
Name: svc.Name,
|
|
Version: svc.Version,
|
|
Source: svc.Source,
|
|
Metadata: svc.Metadata,
|
|
},
|
|
}
|
|
|
|
if _, err := s.runtime.Update(context.Background(), req); err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// Delete stops and removes the service from the runtime
|
|
func (s *svc) Delete(svc *runtime.Service) error {
|
|
// runtime service create request
|
|
req := &pb.DeleteRequest{
|
|
Service: &pb.Service{
|
|
Name: svc.Name,
|
|
Version: svc.Version,
|
|
Source: svc.Source,
|
|
Metadata: svc.Metadata,
|
|
},
|
|
}
|
|
|
|
if _, err := s.runtime.Delete(context.Background(), req); err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// List lists all services managed by the runtime
|
|
func (s *svc) List() ([]*runtime.Service, error) {
|
|
// list all services managed by the runtime
|
|
resp, err := s.runtime.List(context.Background(), &pb.ListRequest{})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
services := make([]*runtime.Service, 0, len(resp.Services))
|
|
for _, service := range resp.Services {
|
|
svc := &runtime.Service{
|
|
Name: service.Name,
|
|
Version: service.Version,
|
|
Source: service.Source,
|
|
Metadata: service.Metadata,
|
|
}
|
|
services = append(services, svc)
|
|
}
|
|
|
|
return services, nil
|
|
}
|
|
|
|
// Start starts the runtime
|
|
func (s *svc) Start() error {
|
|
// NOTE: nothing to be done here
|
|
return nil
|
|
}
|
|
|
|
// Stop stops the runtime
|
|
func (s *svc) Stop() error {
|
|
// NOTE: nothing to be done here
|
|
return nil
|
|
}
|
|
|
|
// Returns the runtime service implementation
|
|
func (s *svc) String() string {
|
|
return "service"
|
|
}
|
|
|
|
// NewRuntime creates new service runtime and returns it
|
|
func NewRuntime(opts ...runtime.Option) runtime.Runtime {
|
|
// get default options
|
|
options := runtime.Options{}
|
|
|
|
// apply requested options
|
|
for _, o := range opts {
|
|
o(&options)
|
|
}
|
|
|
|
// create default client
|
|
cli := client.DefaultClient
|
|
|
|
return &svc{
|
|
options: options,
|
|
runtime: pb.NewRuntimeService(runtime.DefaultName, cli),
|
|
}
|
|
}
|