db2

type Adapter struct {
	DB2Client Client
}

type Client interface {
	GetPage(ctx context.Context, request *Request) (*Response, *framework.Error)
}

type Config struct {
	// Common configuration
	*config.CommonConfig

	// API Version for configuration compatibility
	APIVersion string `json:"apiVersion,omitempty"`

	// DB2 database to connect to.
	Database string `json:"database,omitempty"`

	// Schema name to use for table queries (optional, defaults to username if not specified)
	Schema string `json:"schema,omitempty"`

	// CertificateChain is a base64-encoded PEM certificate chain to use for SSL connections.
	// When provided, SSL will be enabled and this certificate chain will be used
	// to validate the DB2 server's certificate.
	CertificateChain string `json:"certificateChain,omitempty"`

	Filters map[string]condexpr.Condition `json:"filters,omitempty"`
}

type Datasource struct {
	Client SQLClient
}

type MockRows struct {
	Data []map[string]interface{}
	// contains filtered or unexported fields
}

type MockSQLClient struct {
	ConnectFunc func(dataSourceName string) (*sql.DB, error)
	QueryFunc   func(ctx context.Context, query string, args ...interface{}) (Rows, error)
}

type Request struct {
	// BaseURL is the Base URL of the datasource to query.
	BaseURL string `json:"baseURL"`

	// Username is the user name used to authenticate with the DB2 instance.
	Username string `json:"username"`

	// Password is the password used to authenticate with the DB2 instance.
	Password string `json:"password"`

	// PageSize is the maximum number of objects to return from the entity.
	PageSize int64 `json:"pageSize,omitempty"`

	// EntityConfig contains the entity configuration for the current request.
	EntityConfig framework.EntityConfig

	// A filter to apply to the DB2 request when pulling data for the current entity.
	Filter *condexpr.Condition

	// Cursor identifies the first object of the page to return, as returned by
	// the last request for the entity.
	// nil in the request for the first page.
	Cursor *string `json:"cursor,omitempty"`

	// DB2 database to connect to.
	Database string `json:"database"`

	// Schema name for table queries (optional)
	Schema string `json:"schema,omitempty"`

	// UniqueAttributeExternalID is used to specify the unique ID that should be used when ordering results from
	// the specified table.
	UniqueAttributeExternalID string `json:"uniqueAttributeExternalID"`

	// UniqueKeyColumns contains the columns that comprise the unique key for composite ID generation
	UniqueKeyColumns []string `json:"uniqueKeyColumns,omitempty"`

	// ConfigStruct contains the parsed configuration for SSL and other advanced options
	ConfigStruct interface{} `json:"configStruct,omitempty"`
}

type Response struct {
	// Objects is a list of objects returned from the query.
	Objects []map[string]interface{} `json:"objects,omitempty"`

	// NextCursor is the cursor to use for the next page of results.
	NextCursor *string `json:"nextCursor,omitempty"`

	// TotalCount is the total number of objects matching the query.
	TotalCount int64 `json:"totalCount,omitempty"`

	// StatusCode is the HTTP status code returned from the DB2 query.
	StatusCode int `json:"statusCode,omitempty"`
}

type Rows interface {
	Next() bool
	Scan(dest ...interface{}) error
	Close() error
	Err() error
	Columns() ([]string, error)
}

type SQLClient interface {
	Connect(dataSourceName string) (*sql.DB, error)
	Query(ctx context.Context, query string, args ...interface{}) (Rows, error)
}

type SQLColumnTypes map[string]string

type SQLRow map[string]string

type SQLRows []SQLRow

type UniqueConstraintColumn struct {
	ColumnName string
	Position   int
}