Skip to content
Snippets Groups Projects
Select Git revision
  • 7f25aee6703c77caf9b44d1652c7282de47dd5e9
  • master default protected
  • feature/PRXS-3383-CollectionsRankSortAPI
  • fix/PRXS-3401-ValidateValidationOpts
  • feature/3149-LocaleCodeAsID-Feature
  • feature/PRXS-3383-CollectionsSort
  • feature/3109-SerializeFeature
  • release/0.33
  • feature/3109-RecoverySchema
  • feature/3109-feature
  • fix/PRXS-3369-ValidateFields
  • refactor/PRXS-3306-MovePkgGroup1
  • refactor/6-pkg-refactor-expr
  • fix/PRXS-3360-TemplateBuilderPatch
  • feature/3293-MongoV2
  • feature/3272-GoVersionUp
  • feature/PRXS-3218-HideTemplateActions
  • feature/PRXS-3234-PruneIdents
  • feature/3146-UpdateItemStorageInterface
  • feature/3274-ObjectIndexesFixes
  • feature/PRXS-3143-3235-ReferenceOptions
  • v0.33.1
  • v0.32.0
  • v0.31.1
  • v0.31.0
  • v0.30.0
  • v0.29.0
  • v0.28.0
  • v0.27.0-alpha.1+16
  • v0.27.0-alpha.1+15
  • v0.27.0-alpha.1+14
  • v0.27.0-alpha.1+13
  • v0.27.0-alpha.1+12
  • v0.27.0-alpha.1+11
  • v0.27.0-alpha.1+10
  • v0.27.0-alpha.1+9
  • v0.27.0-alpha.1+8
  • v0.27.0-alpha.1+7
  • v0.27.0-alpha.1+6
  • v0.27.0-alpha.1+5
  • v0.27.0-alpha.1+4
41 results

registrar.go

Blame
  • registrar.go 2.49 KiB
    package service
    
    import (
    	"context"
    	"time"
    
    	"git.perx.ru/perxis/perxis-go/pkg/auth"
    	"git.perx.ru/perxis/perxis-go/pkg/extension"
    	retry "github.com/avast/retry-go/v4"
    	"go.uber.org/zap"
    	"google.golang.org/grpc"
    )
    
    const RegistrationDelay = time.Minute
    
    // Registrar выполняет действия по регистрации и обновления регистрации расширений в менеджере расширений. Одновременно
    // выполняется регистрация одного или нескольких расширений
    type Registrar struct {
    	addr        string
    	managerConn *grpc.ClientConn
    	manager     extension.Manager
    	exts        []extension.Extension
    	logger      *zap.Logger
    	stopFn      func() error
    }
    
    func NewRegistrar(addr string, man extension.Manager, exts []extension.Extension, logger *zap.Logger) *Registrar {
    	return &Registrar{
    		addr:    addr,
    		manager: man,
    		exts:    exts,
    		logger:  logger,
    	}
    }
    
    func (reg *Registrar) register(ctx context.Context, manager extension.Manager, desc []*extension.ExtensionConnector) error {
    	err := retry.Do(
    		func() error {
    			ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
    			defer cancel()
    			err := manager.RegisterExtensions(auth.WithSystem(ctx), desc...)
    			return err
    		},
    		retry.RetryIf(func(err error) bool { return err != nil }),
    		retry.OnRetry(func(n uint, err error) {
    			reg.logger.Warn("Fail to register extension", zap.Uint("Retry", n), zap.Error(err))
    		}),
    		retry.DelayType(retry.BackOffDelay),
    		retry.MaxDelay(2*time.Minute),
    		retry.Attempts(1000),
    		retry.Context(ctx),
    	)
    
    	if err == nil {
    		reg.logger.Debug("Extensions successful registered")
    	}
    
    	return err
    }
    
    func (reg *Registrar) Start() error {
    	registrationDelay := time.Duration(0)
    
    	regCtx, regStop := context.WithCancel(context.Background())
    
    	reg.stopFn = func() error {
    		regStop()
    		return nil
    	}
    
    	extList := make([]*extension.ExtensionConnector, 0, len(reg.exts))
    	for _, v := range reg.exts {
    		desc := *v.GetDescriptor()
    		desc.Url = reg.addr
    		extList = append(extList, &extension.ExtensionConnector{Descriptor: &desc})
    	}
    
    	reg.logger.Info("Start registration process")
    
    	go func() {
    		for {
    			select {
    			case <-time.After(registrationDelay):
    				reg.register(regCtx, reg.manager, extList)
    				registrationDelay = RegistrationDelay
    			case <-regCtx.Done():
    				reg.logger.Info("Stop registration process")
    				return
    			}
    		}
    	}()
    
    	return nil
    }
    
    func (reg *Registrar) Stop() error {
    	return reg.stopFn()
    }