db-migrator.go

1
package args
2

3
import (
4
	"fmt"
5
	"strconv"
6

7
	"github.com/pkg/errors"
8
)
9

10
const (
11
	empty = ""
12
	all   = "all"
13
)
14

15
var ErrArgumentMustBeGreaterThanZero = errors.New("the step argument must be greater than 0")
16

17
func ParseStepStringOrDefault(value string, defaults int) (int, error) {
18
	switch value {
19
	case empty:
20
		return defaults, nil
21
	case all:
22
		return 0, nil
23
	default:
24
		i, err := strconv.Atoi(value)
25
		if err != nil {
26
			return -1, fmt.Errorf("the step argument %s is not valid", value)
27
		}
28

29
		if i < 1 {
30
			return -1, ErrArgumentMustBeGreaterThanZero
31
		}
32

33
		return i, nil
34
	}
35
}
36