Function New

provisionerd/provisionerd.go:72–118

New creates and starts a provisioner daemon.

(clientDialer Dialer, opts *Options)

// New creates and starts a provisioner daemon.
func New(clientDialer Dialer, opts *Options) *Server {
	if opts == nil {
		opts = &Options{}
	}
	if opts.UpdateInterval == 0 {
		opts.UpdateInterval = 5 * time.Second
	}
	if opts.ForceCancelInterval == 0 {
		opts.ForceCancelInterval = 10 * time.Minute
	}
	if opts.LogBufferInterval == 0 {
		opts.LogBufferInterval = 250 * time.Millisecond
	}
	if opts.TracerProvider == nil {
		opts.TracerProvider = trace.NewNoopTracerProvider()
	}
	if opts.Metrics == nil {
		reg := prometheus.NewRegistry()
		mets := NewMetrics(reg)
		opts.Metrics = &mets
	}
	if opts.InitConnectionCh == nil {
		opts.InitConnectionCh = make(chan struct{})
	}

	ctx, ctxCancel := context.WithCancel(context.Background())
	daemon := &Server{
		opts:   opts,
		tracer: opts.TracerProvider.Tracer(tracing.TracerName),

		clientDialer: clientDialer,
		clientCh:     make(chan proto.DRPCProvisionerDaemonClient),

		closeContext:        ctx,
		closeCancel:         ctxCancel,
		closedCh:            make(chan struct{}),
		shuttingDownCh:      make(chan struct{}),
		acquireDoneCh:       make(chan struct{}),
		initConnectionCh:    opts.InitConnectionCh,
		externalProvisioner: opts.ExternalProvisioner,
	}

	daemon.wg.Add(2)
	go daemon.connect()
	go daemon.acquireLoop()
	return daemon
}

type Server struct {
	opts   *Options