dbcommon

Documentation

Overview

Package dbcommon provides common database functionality

Index

• Constants
• Variables
• func EnumScan(src interface{}) (uint8, error)
• func EnumValue(enum EnumInter) (driver.Value, error)
• func GetGormLogger(zapLogger *log.ZapEventLogger) gormLogger.Interface
• func GetModelName(db *gorm.DB, model interface{}) (string, error)
• func GetTestConnString() string
• type DBType
  • func DBTypeFromString(str string) (DBType, error)
  • func (i DBType) String() string
• type Enum
• type EnumInter
• type Namer
  • func NewNamer(models []interface{}) Namer
  • func (n Namer) GetConsistentName(fieldName string) string

Examples

• Enum
• Namer.GetConsistentName

Constants

const (
	// EnableMysqlTestVar is the environment variable to enable mysql tests.
	EnableMysqlTestVar = "ENABLE_MYSQL_TEST"
	// MysqlDatabaseVar is the environment variable for the mysql database dsn.
	MysqlDatabaseVar = "MYSQL_DATABASE"
	// MysqlUserVar is the environment variable for the mysql user.
	MysqlUserVar = "MYSQL_USER"
	// MysqlPasswordVar is the environment variable for the mysql password.
	MysqlPasswordVar = "MYSQL_PASSWORD"
	// MysqlHostVar is the environment variable for the mysql host.
	MysqlHostVar = "MYSQL_HOST"
	// MysqlPortVar is the environment variable for the mysql port.
	MysqlPortVar = "MYSQL_PORT"
)

const EnumDataType = "integer"

EnumDataType is exported here to be passed as GormDataType.

Variables

var AllDBTypes []DBType

AllDBTypes is a list of all contract types. Since we use stringer and this is a testing library, instead of manually copying all these out we pull the names out of stringer. In order to make sure stringer is updated, we panic on any method called where the index is higher than the stringer array length.

Functions

func EnumScan

func EnumScan(src interface{}) (uint8, error)

EnumScan converts the enum to a value. the returned uint 8 should be used. Note this should only be used for a non-nullable enum. TODO: consider using generics to make this more type safe.

func EnumValue

func EnumValue(enum EnumInter) (driver.Value, error)

EnumValue converts the enum to a value.

func GetGormLogger

func GetGormLogger(zapLogger *log.ZapEventLogger) gormLogger.Interface

GetGormLogger gets a gorm logger at the correct level TODO investigate https://github.com/moul/zapgorm, we want to use the same write group. TODO: otel logging.

func GetModelName

func GetModelName(db *gorm.DB, model interface{}) (string, error)

GetModelName returns the name of the model.

func GetTestConnString

func GetTestConnString() string

GetTestConnString returns the connection string for the mysql test database. this is derived from environment variables. TODO: test this in ci.

Types

type DBType

type DBType int

DBType is the database driver to use.

const (
	// Mysql is a mysql base db.
	Mysql DBType = 0 // mysql
	// Sqlite file based db.
	Sqlite DBType = iota // sqlite
	// Clickhouse performant db by yandex.
	Clickhouse DBType = iota // clickhouse
)

func DBTypeFromString

func DBTypeFromString(str string) (DBType, error)

DBTypeFromString parses a database type from a string.

func (DBType) String

func (i DBType) String() string

type Enum

type Enum interface {
	EnumInter
	fmt.Stringer
	GormDataType() string
	sql.Scanner
	driver.Valuer
}

Enum is a type used to define interfaces db enums must conform to. it is kept here so it can be used throughout the app.

Example

ExampleEnum demonstrates example use of the enum interface. this implementation can be confusing, so there's an example below.

res, err := RunEnumExample(context.Background(), fmt.Sprintf("%s/sql.db", os.TempDir()))
if err != nil {
	panic(err)
}

for _, res := range res {
	fmt.Printf("got result %s \n", res.Fruit.String())
}

type EnumInter

type EnumInter interface {
	Int() uint8
}

EnumInter ensures an enum has a type to access an integer.

type Namer

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

Namer is used to pull consistent names for fields from models. This prevents inconsistency by panicing on breaking changes. It also allows us to avoid using xModelFieldName, yModelFieldName by taking advantage of introspection.

func NewNamer

func NewNamer(models []interface{}) Namer

NewNamer creates a new namer.

func (Namer) GetConsistentName

func (n Namer) GetConsistentName(fieldName string) string

GetConsistentName makes sure a `ColumnName` has a common column tag against all models we use this so we don't have to define bridgeRedeemModelFieldName, originChainFieldName, etc. panic's on an inconsistency. Note: this should only be called from init.

after this is called, you can safely call getGormFieldName(AnyModelWithField, fieldName) if all fields cannot make this guarantee, they should be separated out to avoid developer confusion.

this returns the result of getGormFieldName for a valid model. It also panics if no model is valid todo consider generating getters for all field names.

Example

namer := dbcommon.NewNamer(getTestModels())
fmt.Println(namer.GetConsistentName("CreatedAt"))

Output:

created_at