go/kratos
1
0
Fork 0
mirror of https://github.com/go-kratos/kratos.git synced 2025-01-28 03:57:02 +02:00
Code Issues Releases Activity

fix: add context for kubernetes registry, implement the registrar interface (#1436)

Browse Source
This commit is contained in:
cachalots 2021-09-07 10:46:07 +08:00 committed by GitHub
parent 842be155a3
commit cd35233d1d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
contrib/registry/kubernetes/registry.go
View File

@ -108,7 +108,7 @@ func NewRegistry(clientSet *kubernetes.Clientset) *Registry {
// Register is used to register services // Register is used to register services
// Note that on Kubernetes, it can only be used to update the id/name/version/metadata/protocols of the current service, // Note that on Kubernetes, it can only be used to update the id/name/version/metadata/protocols of the current service,
// but it cannot be used to update node. // but it cannot be used to update node.
func (s *Registry) Register(service *registry.ServiceInstance) error { func (s *Registry) Register(ctx context.Context, service *registry.ServiceInstance) error {
// GetMetadata // GetMetadata
metadataVal, err := marshal(service.Metadata) metadataVal, err := marshal(service.Metadata)
if err != nil { if err != nil {
@ -145,15 +145,15 @@ func (s *Registry) Register(service *registry.ServiceInstance) error {
if _, err = s.clientSet. if _, err = s.clientSet.
CoreV1(). CoreV1().
Pods(GetNamespace()). Pods(GetNamespace()).
Patch(context.TODO(), GetPodName(), types.StrategicMergePatchType, patchBytes, metav1.PatchOptions{}); err != nil { Patch(ctx, GetPodName(), types.StrategicMergePatchType, patchBytes, metav1.PatchOptions{}); err != nil {
return err return err
} }
return nil return nil
} }
// Deregister the registration. // Deregister the registration.
func (s *Registry) Deregister(service *registry.ServiceInstance) error { func (s *Registry) Deregister(ctx context.Context, service *registry.ServiceInstance) error {
return s.Register(&registry.ServiceInstance{ return s.Register(ctx, &registry.ServiceInstance{
Metadata: map[string]string{}, Metadata: map[string]string{},
}) })
} }