evalphobia / aws-sdk-go-v2-wrapper

package athena

import (

	SDK "github.com/aws/aws-sdk-go-v2/service/athena"

	"github.com/evalphobia/aws-sdk-go-v2-wrapper/config"

	"github.com/evalphobia/aws-sdk-go-v2-wrapper/errors"

	"github.com/evalphobia/aws-sdk-go-v2-wrapper/log"

)

const (

	serviceName = "athena"

)

// Athena has Athena client.

type Athena struct {

	client *SDK.Client

	logger  log.Logger

	errWrap func(errors.ErrorData) error

}

// New returns initialized *Athena.

func New(conf config.Config) (*Athena, error) {

	c, err := conf.AWSConfig()

	if err != nil {

		return nil, err

	}

	if conf.Endpoints.HasAthena() {

		c.EndpointResolver = conf.Endpoints.GetAthena()

	}

	svc := &Athena{

		client:  SDK.New(c),

		logger:  conf.GetLogger(),

		errWrap: conf.GetErrWrap(),

	}

	return svc, nil

}

// GetClient gets original SDK client.

func (svc *Athena) GetClient() *SDK.Client {

	return svc.client

}

// SetLogger sets logger.

func (svc *Athena) SetLogger(logger log.Logger) {

	svc.logger = logger

}

// Infof logging information.

func (svc *Athena) Infof(format string, v ...interface{}) {

	svc.logger.Infof(serviceName, format, v...)

}

// Errorf logging error information.

func (svc *Athena) Errorf(format string, v ...interface{}) {

}