drone/internal/store/database/util.go

// Copyright 2022 Harness Inc. All rights reserved.
// Use of this source code is governed by the Polyform Free Trial License
// that can be found in the LICENSE.md file for this repository.

package database

import (
	"database/sql"
	"fmt"

	"github.com/harness/gitness/internal/store"
	"github.com/mattn/go-sqlite3"
	"github.com/pkg/errors"
	"github.com/rs/zerolog/log"
)

// default query range limit.
const defaultLimit = 100

// limit returns the page size to a sql limit.
func limit(size int) int {
	if size == 0 {
		size = defaultLimit
	}
	return size
}

// offset converts the page to a sql offset.
func offset(page, size int) int {
	if page == 0 {
		page = 1
	}
	if size == 0 {
		size = defaultLimit
	}
	page--
	return page * size
}

// Logs the error and message, returns either the original error or a store equivalent if possible.
func processSQLErrorf(err error, format string, args ...interface{}) error {
	// always log DB error (print formated message)
	log.Debug().Msgf("%s %s", fmt.Sprintf(format, args...), err)

	// If it's a known error, return converted error instead.
	if errors.Is(err, sql.ErrNoRows) {
		return store.ErrResourceNotFound
	} else if isSQLUniqueConstraintError(err) {
		return store.ErrDuplicate
	}

	return err
}

func isSQLUniqueConstraintError(original error) bool {
	err := sqlite3.Error{}
	ok := errors.As(original, &err)
	return ok && errors.Is(err.ExtendedCode, sqlite3.ErrConstraintUnique)
}