Function NewSQLRepository

asset/sql_repository.go:38–79

NewSQLRepository takes a database driver, URL, and dialect for the asset repository and connects to it.

(dbDriver, dbURL string, dialect SQLRepositoryDialect)

// NewSQLRepository takes a database driver, URL, and dialect for the asset repository and connects to it.
func NewSQLRepository(dbDriver, dbURL string, dialect SQLRepositoryDialect) (*SQLRepository, error) {
	db, err := sql.Open(dbDriver, dbURL)
	if err != nil {
		return nil, fmt.Errorf("unable to connect database: %w", err)
	}

	_, err = db.Exec(dialect.CreateTable())
	if err != nil {
		return nil, helper.CloseDatabaseWithError(db, fmt.Errorf("unable to create table: %w", err))
	}

	assetQuery, err := db.Prepare(dialect.Assets())
	if err != nil {
		return nil, helper.CloseDatabaseWithError(db, fmt.Errorf("unable to prepare assets: %w", err))
	}

	getSinceQuery, err := db.Prepare(dialect.GetSince())
	if err != nil {
		return nil, helper.CloseDatabaseWithError(db, fmt.Errorf("unable to prepare get since query: %w", err))
	}

	lastDateQuery, err := db.Prepare(dialect.LastDate())
	if err != nil {
		return nil, helper.CloseDatabaseWithError(db, fmt.Errorf("unable to prepare last date query: %w", err))
	}

	appendQuery, err := db.Prepare(dialect.Append())
	if err != nil {
		return nil, helper.CloseDatabaseWithError(db, fmt.Errorf("unable to prepare append: %w", err))
	}

	repository := &SQLRepository{
		db,
		dialect,
		assetQuery,
		getSinceQuery,
		lastDateQuery,
		appendQuery,
	}

	return repository, nil
}

// Close closes the database connection.
func (s *SQLRepository) Close() error {