Function New

plugins/destination/test/client/client.go:37–83

(_ context.Context, logger zerolog.Logger, specBytes []byte, _ plugin.NewClientOptions)

)

func New(_ context.Context, logger zerolog.Logger, specBytes []byte, _ plugin.NewClientOptions) (plugin.Client, error) {
	var spec Spec
	if err := json.Unmarshal(specBytes, &spec); err != nil {
		return nil, fmt.Errorf("failed to unmarshal spec: %w", err)
	}
	spec.SetDefaults()
	if err := spec.Validate(); err != nil {
		return nil, fmt.Errorf("invalid spec: %w", err)
	}

	c := &Client{
		logger: logger.With().Str("module", "test").Logger(),
		spec:   spec,
	}

	var err error
	switch {
	case spec.BatchWriter:
		c.writer, err = batchwriter.New(c,
			batchwriter.WithBatchSize(*c.spec.BatchSize),
			batchwriter.WithBatchSizeBytes(*c.spec.BatchSizeBytes),
			batchwriter.WithBatchTimeout(c.spec.BatchTimeout.Duration()),
			batchwriter.WithLogger(c.logger),
		)
	case spec.MixedBatchWriter:
		var mbw *mixedbatchwriter.MixedBatchWriter
		mbw, err = mixedbatchwriter.New(c,
			mixedbatchwriter.WithBatchSize(*c.spec.BatchSize),
			mixedbatchwriter.WithBatchSizeBytes(*c.spec.BatchSizeBytes),
			mixedbatchwriter.WithBatchTimeout(c.spec.BatchTimeout.Duration()),
			mixedbatchwriter.WithLogger(c.logger),
		)
		c.writer = adaptMixedBatchWriter(mbw)
	case spec.StreamBatchWriter:
		c.writer, err = streamingbatchwriter.New(newClientForStreamingBatchWriter(c),
			streamingbatchwriter.WithBatchSizeRows(*c.spec.BatchSize),
			streamingbatchwriter.WithBatchSizeBytes(*c.spec.BatchSizeBytes),
			streamingbatchwriter.WithBatchTimeout(c.spec.BatchTimeout.Duration()),
			streamingbatchwriter.WithLogger(c.logger),
		)
	}
	if err != nil {
		return nil, err
	}

	return c, nil
}

func (*Client) Read(context.Context, *schema.Table, chan<- arrow.RecordBatch) error {
	return nil