signoz/pkg/errors/code.go

package errors

import (
	"fmt"
	"regexp"
)

var (
	CodeInvalidInput     Code = Code{"invalid_input"}
	CodeInternal              = Code{"internal"}
	CodeUnsupported           = Code{"unsupported"}
	CodeNotFound              = Code{"not_found"}
	CodeMethodNotAllowed      = Code{"method_not_allowed"}
	CodeAlreadyExists         = Code{"already_exists"}
	CodeUnauthenticated       = Code{"unauthenticated"}
	CodeForbidden             = Code{"forbidden"}
	CodeCanceled              = Code{"canceled"}
	CodeTimeout               = Code{"timeout"}
)

var (
	codeRegex = regexp.MustCompile(`^[a-z_]+$`)
)

type Code struct{ s string }

func NewCode(s string) (Code, error) {
	if !codeRegex.MatchString(s) {
		return Code{}, fmt.Errorf("invalid code: %v", s)
	}

	return Code{s: s}, nil
}

func MustNewCode(s string) Code {
	code, err := NewCode(s)
	if err != nil {
		panic(err)
	}

	return code
}

func (c Code) String() string {
	return c.s
}