drone/internal/store/database/util.go
Enver Bisevac f03528e862 [MAINT] initial config for ci linter (#17)
* initial config for ci linter

* more linter work

* linter errors fix

* linter errors fix

* linter conf minor changes
2022-09-19 18:13:18 +02:00

60 lines
1.4 KiB
Go

// Copyright 2021 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.Warn().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)
}