shard

type FakeProviderServerSender struct {
	ProviderServerSender
}

type GRPCClient struct {
	// contains filtered or unexported fields
}

type GRPCServer struct {
	// This is the real implementation
	Impl ProviderServer
	internal.UnimplementedProviderServer
}

type GetProviderConfigRequest struct{}

type GetProviderConfigResponse struct {
	Name        string              `json:"name"`
	Version     string              `json:"version"`
	Config      string              `json:"config"`
	Diagnostics *schema.Diagnostics `json:"diagnostics"`
}

type GetProviderInformationRequest struct{}

type GetProviderInformationResponse struct {
	Name    string                   `json:"name"`
	Version string                   `json:"version"`
	Tables  map[string]*schema.Table `json:"tables"`

	DefaultConfigTemplate string `json:"default_config_template"`

	Diagnostics *schema.Diagnostics
}

type Plugin struct {
	// GRPCPlugin must still implement the Stub interface
	plugin.Plugin
	// Concrete implementation, written in Go. This is only used for plugins
	// that are written in Go.
	Impl ProviderServer
}

type Provider interface {
	Init(ctx context.Context, in *ProviderInitRequest) (*ProviderInitResponse, error)
	GetProviderInformation(ctx context.Context, in *GetProviderInformationRequest) (*GetProviderInformationResponse, error)
	GetProviderConfig(ctx context.Context, in *GetProviderConfigRequest) (*GetProviderConfigResponse, error)
	SetProviderConfig(ctx context.Context, in *SetProviderConfigRequest) (*SetProviderConfigResponse, error)
	DropTableAll(ctx context.Context, in *ProviderDropTableAllRequest) (*ProviderDropTableAllResponse, error)
	CreateAllTables(ctx context.Context, in *ProviderCreateAllTablesRequest) (*ProviderCreateAllTablesResponse, error)
}

type ProviderClient interface {
	Provider
	PullTables(ctx context.Context, in *PullTablesRequest) (ProviderServerStream, error)
}

type ProviderConfiguration struct {
	Storage Storage `json:"storage"`

	NamespaceSuffix string `json:"namespace_suffix"`
}

type ProviderCreateAllTablesRequest struct{}

type ProviderCreateAllTablesResponse struct {
	Diagnostics *schema.Diagnostics `json:"diagnostics"`
}

type ProviderDropTableAllRequest struct{}

type ProviderDropTableAllResponse struct {
	Diagnostics *schema.Diagnostics `json:"diagnostics"`
}

type ProviderInitRequest struct {
	Storage        *Storage
	Workspace      *string
	ProviderConfig *string
	IsInstallInit  *bool
}

type ProviderInitResponse struct {
	Diagnostics *schema.Diagnostics `json:"diagnostics"`
}

type ProviderServer interface {
	Provider
	PullTables(context.Context, *PullTablesRequest, ProviderServerSender) error
}

type ProviderServerSender interface {
	Send(*PullTablesResponse) error
}

type ProviderServerStream interface {
	Recv() (*PullTablesResponse, error)
}

type PullTablesRequest struct {

	// The table to be pulled, this later expands to support parameters
	Tables []string `json:"tables"`

	// Maximum number of threads used
	MaxGoroutines uint64 `json:"max_goroutines"`

	// Pull timeout period
	Timeout int64 `json:"timeout"`
}

type PullTablesResponse struct {
	FinishedTables map[string]bool     `json:"finished_tables"`
	TableCount     uint64              `json:"table_count"`
	Table          string              `json:"table"`
	Diagnostics    *schema.Diagnostics `json:"diagnostic"`
}

type Recv struct {
	// contains filtered or unexported fields
}

type Send struct {
	// contains filtered or unexported fields
}

type SetProviderConfigRequest struct {
	Storage        *Storage `json:"storage"`
	ProviderConfig *string  `json:"provider_config"`
}

type SetProviderConfigResponse struct {
	Diagnostics *schema.Diagnostics `json:"diagnostics"`
}

type Storage struct {
	Type           StorageType `json:"type"`
	StorageOptions []byte      `json:"options"`
}

type StorageType int