utils

Documentation

Overview

Package utils [general utility functions for the db1000n app]

Index

• func CheckCountry(logger *zap.Logger, countriesToAvoid []string, strictCountryCheck bool, ...) (bool, string)
• func CheckCountryOrFail(logger *zap.Logger, cfg *CountryCheckerConfig, proxyURLs string) string
• func Decode(input any, output any) error
• func Decrypt(cfg []byte) ([]byte, error)
• func GetEncryptionKeys() ([]string, error)
• func GetEnvBoolDefault(key string, defaultValue bool) bool
• func GetEnvDurationDefault(key string, defaultValue time.Duration) time.Duration
• func GetEnvIntDefault(key string, defaultValue int) int
• func GetEnvStringDefault(key, defaultValue string) string
• func InfiniteRange[T any](ctx context.Context, input []T) chan T
• func IsEncrypted(cfg []byte) bool
• func NonNilOrDefault[T any](v *T, dflt T) T
• func PanicHandler(logger *zap.Logger)
• func Sleep(ctx context.Context, t time.Duration) bool
• func Unmarshal(input []byte, output any, format string) error
• func UpdateRLimit(logger *zap.Logger) error
• type BackoffConfig
  • func DefaultBackoffConfig() BackoffConfig
• type BackoffController
  • func (c BackoffController) GetTimeout() time.Duration
  • func (c *BackoffController) Increment() *BackoffController
  • func (c *BackoffController) Reset()
• type Counter
  • func (c *Counter) Next() bool
• type CountryCheckerConfig
  • func NewCountryCheckerConfigWithFlags() *CountryCheckerConfig
• type ProxyFunc
  • func GetProxyFunc(proxyURLs string, timeout time.Duration, httpEnabled bool) ProxyFunc

Functions

func CheckCountry

func CheckCountry(logger *zap.Logger, countriesToAvoid []string, strictCountryCheck bool, proxyURLs string, maxFetchRetries int) (bool, string)

CheckCountry checks which country the app is running from and whether it is in the blacklist.

func CheckCountryOrFail

func CheckCountryOrFail(logger *zap.Logger, cfg *CountryCheckerConfig, proxyURLs string) string

CheckCountryOrFail checks the country of client origin by IP and exits the program if it is in the blacklist.

func Decode

func Decode(input any, output any) error

Decode is an alias to a mapstructure.NewDecoder({Squash: true}).Decode() with WeaklyTypedInput set to true and MatchFunc that only compares aplhanumeric sequence in field names

func Decrypt

func Decrypt(cfg []byte) ([]byte, error)

Decrypt decrypts config using EncryptionKeys

func GetEncryptionKeys

func GetEncryptionKeys() ([]string, error)

GetEncryptionKeys returns list of encryption keys from ENCRYPTION_KEYS env variable name or default value

func GetEnvBoolDefault

func GetEnvBoolDefault(key string, defaultValue bool) bool

GetEnvBoolDefault returns environment variable or default value if no env varible is present

func GetEnvDurationDefault

func GetEnvDurationDefault(key string, defaultValue time.Duration) time.Duration

GetEnvDurationDefault returns environment variable or default value if no env varible is present

func GetEnvIntDefault

func GetEnvIntDefault(key string, defaultValue int) int

GetEnvIntDefault returns environment variable or default value if no env varible is present

func GetEnvStringDefault

func GetEnvStringDefault(key, defaultValue string) string

GetEnvStringDefault returns environment variable or default value if no env varible is present

func InfiniteRange

func InfiniteRange[T any](ctx context.Context, input []T) chan T

It could've been more effective if go had yield but this should be close enough

func IsEncrypted

func IsEncrypted(cfg []byte) bool

IsEncrypted returns true if cfg encrypted with age tool (https://github.com/FiloSottile/age)

func NonNilOrDefault

func NonNilOrDefault[T any](v *T, dflt T) T

func PanicHandler

func PanicHandler(logger *zap.Logger)

PanicHandler just stub it in the beginning of every major module invocation to prevent single module failure from crashing the whole app

func Sleep

func Sleep(ctx context.Context, t time.Duration) bool

func Unmarshal

func Unmarshal(input []byte, output any, format string) error

func UpdateRLimit

func UpdateRLimit(logger *zap.Logger) error

Types

type BackoffConfig

type BackoffConfig struct {
	Multiplier int
	Limit      int
	Timeout    time.Duration
}

func DefaultBackoffConfig

func DefaultBackoffConfig() BackoffConfig

type BackoffController

type BackoffController struct {
	BackoffConfig
	// contains filtered or unexported fields
}

func (BackoffController) GetTimeout

func (c BackoffController) GetTimeout() time.Duration

func (*BackoffController) Increment

func (c *BackoffController) Increment() *BackoffController

func (*BackoffController) Reset

func (c *BackoffController) Reset()

type Counter

type Counter struct {
	Count int
	// contains filtered or unexported fields
}

func (*Counter) Next

func (c *Counter) Next() bool

type CountryCheckerConfig

type CountryCheckerConfig struct {
	// contains filtered or unexported fields
}

func NewCountryCheckerConfigWithFlags

func NewCountryCheckerConfigWithFlags() *CountryCheckerConfig

NewGlobalConfigWithFlags returns a GlobalConfig initialized with command line flags.

type ProxyFunc

type ProxyFunc func(network, addr string) (net.Conn, error)

func GetProxyFunc

func GetProxyFunc(proxyURLs string, timeout time.Duration, httpEnabled bool) ProxyFunc