Break utils package into subpackages.
This commit is contained in:
parent
06ecd73f67
commit
d7a5adb239
25 changed files with 276 additions and 318 deletions
|
|
@ -6,6 +6,7 @@ import (
|
||||||
"flag"
|
"flag"
|
||||||
"fmt"
|
"fmt"
|
||||||
"github.com/go-jet/jet/v2/internal/utils/errfmt"
|
"github.com/go-jet/jet/v2/internal/utils/errfmt"
|
||||||
|
"github.com/go-jet/jet/v2/internal/utils/strslice"
|
||||||
"os"
|
"os"
|
||||||
"strings"
|
"strings"
|
||||||
|
|
||||||
|
|
@ -13,7 +14,6 @@ import (
|
||||||
sqlitegen "github.com/go-jet/jet/v2/generator/sqlite"
|
sqlitegen "github.com/go-jet/jet/v2/generator/sqlite"
|
||||||
"github.com/go-jet/jet/v2/generator/template"
|
"github.com/go-jet/jet/v2/generator/template"
|
||||||
"github.com/go-jet/jet/v2/internal/jet"
|
"github.com/go-jet/jet/v2/internal/jet"
|
||||||
"github.com/go-jet/jet/v2/internal/utils"
|
|
||||||
"github.com/go-jet/jet/v2/mysql"
|
"github.com/go-jet/jet/v2/mysql"
|
||||||
postgres2 "github.com/go-jet/jet/v2/postgres"
|
postgres2 "github.com/go-jet/jet/v2/postgres"
|
||||||
"github.com/go-jet/jet/v2/sqlite"
|
"github.com/go-jet/jet/v2/sqlite"
|
||||||
|
|
@ -232,15 +232,15 @@ func parseList(list string) []string {
|
||||||
func genTemplate(dialect jet.Dialect, ignoreTables []string, ignoreViews []string, ignoreEnums []string) template.Template {
|
func genTemplate(dialect jet.Dialect, ignoreTables []string, ignoreViews []string, ignoreEnums []string) template.Template {
|
||||||
|
|
||||||
shouldSkipTable := func(table metadata.Table) bool {
|
shouldSkipTable := func(table metadata.Table) bool {
|
||||||
return utils.StringSliceContains(ignoreTables, strings.ToLower(table.Name))
|
return strslice.Contains(ignoreTables, strings.ToLower(table.Name))
|
||||||
}
|
}
|
||||||
|
|
||||||
shouldSkipView := func(view metadata.Table) bool {
|
shouldSkipView := func(view metadata.Table) bool {
|
||||||
return utils.StringSliceContains(ignoreViews, strings.ToLower(view.Name))
|
return strslice.Contains(ignoreViews, strings.ToLower(view.Name))
|
||||||
}
|
}
|
||||||
|
|
||||||
shouldSkipEnum := func(enum metadata.Enum) bool {
|
shouldSkipEnum := func(enum metadata.Enum) bool {
|
||||||
return utils.StringSliceContains(ignoreEnums, strings.ToLower(enum.Name))
|
return strslice.Contains(ignoreEnums, strings.ToLower(enum.Name))
|
||||||
}
|
}
|
||||||
|
|
||||||
return template.Default(dialect).
|
return template.Default(dialect).
|
||||||
|
|
|
||||||
|
|
@ -8,7 +8,6 @@ import (
|
||||||
|
|
||||||
"github.com/go-jet/jet/v2/generator/metadata"
|
"github.com/go-jet/jet/v2/generator/metadata"
|
||||||
"github.com/go-jet/jet/v2/generator/template"
|
"github.com/go-jet/jet/v2/generator/template"
|
||||||
"github.com/go-jet/jet/v2/internal/utils"
|
|
||||||
"github.com/go-jet/jet/v2/mysql"
|
"github.com/go-jet/jet/v2/mysql"
|
||||||
mysqldr "github.com/go-sql-driver/mysql"
|
mysqldr "github.com/go-sql-driver/mysql"
|
||||||
)
|
)
|
||||||
|
|
@ -35,7 +34,7 @@ func Generate(destDir string, dbConn DBConnection, generatorTemplate ...template
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return fmt.Errorf("failed to open db connection: %w", err)
|
return fmt.Errorf("failed to open db connection: %w", err)
|
||||||
}
|
}
|
||||||
defer utils.DBClose(db)
|
defer db.Close()
|
||||||
|
|
||||||
err = generate(db, dbConn.DBName, destDir, generatorTemplate...)
|
err = generate(db, dbConn.DBName, destDir, generatorTemplate...)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
|
|
@ -67,7 +66,7 @@ func GenerateDSN(dsn, destDir string, templates ...template.Template) error {
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return fmt.Errorf("failed to open db connection: %w", err)
|
return fmt.Errorf("failed to open db connection: %w", err)
|
||||||
}
|
}
|
||||||
defer utils.DBClose(db)
|
defer db.Close()
|
||||||
|
|
||||||
err = generate(db, cfg.DBName, destDir, templates...)
|
err = generate(db, cfg.DBName, destDir, templates...)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
|
|
|
||||||
|
|
@ -9,7 +9,6 @@ import (
|
||||||
|
|
||||||
"github.com/go-jet/jet/v2/generator/metadata"
|
"github.com/go-jet/jet/v2/generator/metadata"
|
||||||
"github.com/go-jet/jet/v2/generator/template"
|
"github.com/go-jet/jet/v2/generator/template"
|
||||||
"github.com/go-jet/jet/v2/internal/utils"
|
|
||||||
"github.com/go-jet/jet/v2/postgres"
|
"github.com/go-jet/jet/v2/postgres"
|
||||||
"github.com/jackc/pgconn"
|
"github.com/jackc/pgconn"
|
||||||
)
|
)
|
||||||
|
|
@ -54,7 +53,7 @@ func GenerateDSN(dsn, schema, destDir string, templates ...template.Template) er
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return fmt.Errorf("failed to open db connection: %w", err)
|
return fmt.Errorf("failed to open db connection: %w", err)
|
||||||
}
|
}
|
||||||
defer utils.DBClose(db)
|
defer db.Close()
|
||||||
|
|
||||||
fmt.Println("Retrieving schema information...")
|
fmt.Println("Retrieving schema information...")
|
||||||
generatorTemplate := template.Default(postgres.Dialect)
|
generatorTemplate := template.Default(postgres.Dialect)
|
||||||
|
|
|
||||||
|
|
@ -5,7 +5,6 @@ import (
|
||||||
"fmt"
|
"fmt"
|
||||||
"github.com/go-jet/jet/v2/generator/metadata"
|
"github.com/go-jet/jet/v2/generator/metadata"
|
||||||
"github.com/go-jet/jet/v2/generator/template"
|
"github.com/go-jet/jet/v2/generator/template"
|
||||||
"github.com/go-jet/jet/v2/internal/utils"
|
|
||||||
"github.com/go-jet/jet/v2/sqlite"
|
"github.com/go-jet/jet/v2/sqlite"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
@ -15,7 +14,7 @@ func GenerateDSN(dsn, destDir string, templates ...template.Template) error {
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return fmt.Errorf("failed to open sqlite connection: %w", err)
|
return fmt.Errorf("failed to open sqlite connection: %w", err)
|
||||||
}
|
}
|
||||||
defer utils.DBClose(db)
|
defer db.Close()
|
||||||
|
|
||||||
fmt.Println("Retrieving schema information...")
|
fmt.Println("Retrieving schema information...")
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -3,7 +3,7 @@ package template
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
"github.com/go-jet/jet/v2/generator/metadata"
|
"github.com/go-jet/jet/v2/generator/metadata"
|
||||||
"github.com/go-jet/jet/v2/internal/utils"
|
"github.com/go-jet/jet/v2/internal/utils/dbidentifier"
|
||||||
"github.com/google/uuid"
|
"github.com/google/uuid"
|
||||||
"path"
|
"path"
|
||||||
"reflect"
|
"reflect"
|
||||||
|
|
@ -77,8 +77,8 @@ var DefaultViewModel = DefaultTableModel
|
||||||
// DefaultTableModel is default table template implementation
|
// DefaultTableModel is default table template implementation
|
||||||
func DefaultTableModel(tableMetaData metadata.Table) TableModel {
|
func DefaultTableModel(tableMetaData metadata.Table) TableModel {
|
||||||
return TableModel{
|
return TableModel{
|
||||||
FileName: utils.ToGoFileName(tableMetaData.Name),
|
FileName: dbidentifier.ToGoFileName(tableMetaData.Name),
|
||||||
TypeName: utils.ToGoIdentifier(tableMetaData.Name),
|
TypeName: dbidentifier.ToGoIdentifier(tableMetaData.Name),
|
||||||
Field: DefaultTableModelField,
|
Field: DefaultTableModelField,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
@ -142,13 +142,13 @@ func (em EnumModel) UseTypeName(typeName string) EnumModel {
|
||||||
|
|
||||||
// DefaultEnumModel returns default implementation for EnumModel
|
// DefaultEnumModel returns default implementation for EnumModel
|
||||||
func DefaultEnumModel(enumMetaData metadata.Enum) EnumModel {
|
func DefaultEnumModel(enumMetaData metadata.Enum) EnumModel {
|
||||||
typeName := utils.ToGoIdentifier(enumMetaData.Name)
|
typeName := dbidentifier.ToGoIdentifier(enumMetaData.Name)
|
||||||
|
|
||||||
return EnumModel{
|
return EnumModel{
|
||||||
FileName: utils.ToGoFileName(enumMetaData.Name),
|
FileName: dbidentifier.ToGoFileName(enumMetaData.Name),
|
||||||
TypeName: typeName,
|
TypeName: typeName,
|
||||||
ValueName: func(value string) string {
|
ValueName: func(value string) string {
|
||||||
return typeName + "_" + utils.ToGoIdentifier(value)
|
return typeName + "_" + dbidentifier.ToGoIdentifier(value)
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
@ -169,7 +169,7 @@ func DefaultTableModelField(columnMetaData metadata.Column) TableModelField {
|
||||||
}
|
}
|
||||||
|
|
||||||
return TableModelField{
|
return TableModelField{
|
||||||
Name: utils.ToGoIdentifier(columnMetaData.Name),
|
Name: dbidentifier.ToGoIdentifier(columnMetaData.Name),
|
||||||
Type: getType(columnMetaData),
|
Type: getType(columnMetaData),
|
||||||
Tags: tags,
|
Tags: tags,
|
||||||
}
|
}
|
||||||
|
|
@ -247,7 +247,7 @@ func getType(columnMetadata metadata.Column) Type {
|
||||||
func getUserDefinedType(column metadata.Column) string {
|
func getUserDefinedType(column metadata.Column) string {
|
||||||
switch column.DataType.Kind {
|
switch column.DataType.Kind {
|
||||||
case metadata.EnumType:
|
case metadata.EnumType:
|
||||||
return utils.ToGoIdentifier(column.DataType.Name)
|
return dbidentifier.ToGoIdentifier(column.DataType.Name)
|
||||||
case metadata.UserDefinedType, metadata.ArrayType:
|
case metadata.UserDefinedType, metadata.ArrayType:
|
||||||
return "string"
|
return "string"
|
||||||
}
|
}
|
||||||
|
|
|
||||||
|
|
@ -4,13 +4,13 @@ import (
|
||||||
"bytes"
|
"bytes"
|
||||||
"errors"
|
"errors"
|
||||||
"fmt"
|
"fmt"
|
||||||
|
"github.com/go-jet/jet/v2/internal/utils/filesys"
|
||||||
"path"
|
"path"
|
||||||
"strings"
|
"strings"
|
||||||
"text/template"
|
"text/template"
|
||||||
|
|
||||||
"github.com/go-jet/jet/v2/generator/metadata"
|
"github.com/go-jet/jet/v2/generator/metadata"
|
||||||
"github.com/go-jet/jet/v2/internal/jet"
|
"github.com/go-jet/jet/v2/internal/jet"
|
||||||
"github.com/go-jet/jet/v2/internal/utils"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
// ProcessSchema will process schema metadata and constructs go files using generator Template
|
// ProcessSchema will process schema metadata and constructs go files using generator Template
|
||||||
|
|
@ -24,7 +24,7 @@ func ProcessSchema(dirPath string, schemaMetaData metadata.Schema, generatorTemp
|
||||||
|
|
||||||
fmt.Println("Destination directory:", schemaPath)
|
fmt.Println("Destination directory:", schemaPath)
|
||||||
fmt.Println("Cleaning up destination directory...")
|
fmt.Println("Cleaning up destination directory...")
|
||||||
err := utils.CleanUpGeneratedFiles(schemaPath)
|
err := filesys.RemoveDir(schemaPath)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return errors.New("failed to cleanup generated files")
|
return errors.New("failed to cleanup generated files")
|
||||||
}
|
}
|
||||||
|
|
@ -52,7 +52,7 @@ func processModel(dirPath string, schemaMetaData metadata.Schema, schemaTemplate
|
||||||
|
|
||||||
modelDirPath := path.Join(dirPath, modelTemplate.Path)
|
modelDirPath := path.Join(dirPath, modelTemplate.Path)
|
||||||
|
|
||||||
err := utils.EnsureDirPath(modelDirPath)
|
err := filesys.EnsureDirPathExist(modelDirPath)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return fmt.Errorf("destination dir path does not exist: %w", err)
|
return fmt.Errorf("destination dir path does not exist: %w", err)
|
||||||
}
|
}
|
||||||
|
|
@ -119,7 +119,7 @@ func processEnumSQLBuilder(dirPath string, dialect jet.Dialect, enumsMetaData []
|
||||||
|
|
||||||
enumSQLBuilderPath := path.Join(dirPath, enumTemplate.Path)
|
enumSQLBuilderPath := path.Join(dirPath, enumTemplate.Path)
|
||||||
|
|
||||||
err := utils.EnsureDirPath(enumSQLBuilderPath)
|
err := filesys.EnsureDirPathExist(enumSQLBuilderPath)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return fmt.Errorf("failed to create enum sql builder directory - %s: %w", enumSQLBuilderPath, err)
|
return fmt.Errorf("failed to create enum sql builder directory - %s: %w", enumSQLBuilderPath, err)
|
||||||
}
|
}
|
||||||
|
|
@ -145,7 +145,7 @@ func processEnumSQLBuilder(dirPath string, dialect jet.Dialect, enumsMetaData []
|
||||||
return fmt.Errorf("failed to generete enum type %s: %w", enumTemplate.FileName, err)
|
return fmt.Errorf("failed to generete enum type %s: %w", enumTemplate.FileName, err)
|
||||||
}
|
}
|
||||||
|
|
||||||
err = utils.SaveGoFile(enumSQLBuilderPath, enumTemplate.FileName, text)
|
err = filesys.FormatAndSaveGoFile(enumSQLBuilderPath, enumTemplate.FileName, text)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return fmt.Errorf("failed to format and save '%s' enum type : %w", enumTemplate.FileName, err)
|
return fmt.Errorf("failed to format and save '%s' enum type : %w", enumTemplate.FileName, err)
|
||||||
}
|
}
|
||||||
|
|
@ -183,7 +183,7 @@ func processTableSQLBuilder(fileTypes, dirPath string,
|
||||||
|
|
||||||
tableSQLBuilderPath := path.Join(dirPath, tableSQLBuilder.Path)
|
tableSQLBuilderPath := path.Join(dirPath, tableSQLBuilder.Path)
|
||||||
|
|
||||||
err := utils.EnsureDirPath(tableSQLBuilderPath)
|
err := filesys.EnsureDirPathExist(tableSQLBuilderPath)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return fmt.Errorf("failed to create table sql builder directory - %s: %w", tableSQLBuilderPath, err)
|
return fmt.Errorf("failed to create table sql builder directory - %s: %w", tableSQLBuilderPath, err)
|
||||||
}
|
}
|
||||||
|
|
@ -220,7 +220,7 @@ func processTableSQLBuilder(fileTypes, dirPath string,
|
||||||
return fmt.Errorf("failed to generate table sql builder type %s: %w", tableSQLBuilder.TypeName, err)
|
return fmt.Errorf("failed to generate table sql builder type %s: %w", tableSQLBuilder.TypeName, err)
|
||||||
}
|
}
|
||||||
|
|
||||||
err = utils.SaveGoFile(tableSQLBuilderPath, tableSQLBuilder.FileName, text)
|
err = filesys.FormatAndSaveGoFile(tableSQLBuilderPath, tableSQLBuilder.FileName, text)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return fmt.Errorf("failed to format and save generated sql builder type '%s': %w", tableSQLBuilder.FileName, err)
|
return fmt.Errorf("failed to format and save generated sql builder type '%s': %w", tableSQLBuilder.FileName, err)
|
||||||
}
|
}
|
||||||
|
|
@ -256,7 +256,7 @@ func generateUseSchemaFunc(dirPath, fileTypes string, builders []TableSQLBuilder
|
||||||
basePath := path.Join(dirPath, builders[0].Path)
|
basePath := path.Join(dirPath, builders[0].Path)
|
||||||
fileName := fileTypes + "_use_schema"
|
fileName := fileTypes + "_use_schema"
|
||||||
|
|
||||||
err = utils.SaveGoFile(basePath, fileName, text)
|
err = filesys.FormatAndSaveGoFile(basePath, fileName, text)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return fmt.Errorf("failed to save %s file: %w", fileName, err)
|
return fmt.Errorf("failed to save %s file: %w", fileName, err)
|
||||||
}
|
}
|
||||||
|
|
@ -312,7 +312,7 @@ func processTableModels(fileTypes, modelDirPath string, tablesMetaData []metadat
|
||||||
return fmt.Errorf("failed to generate model type '%s': %w", tableMetaData.Name, err)
|
return fmt.Errorf("failed to generate model type '%s': %w", tableMetaData.Name, err)
|
||||||
}
|
}
|
||||||
|
|
||||||
err = utils.SaveGoFile(modelDirPath, tableTemplate.FileName, text)
|
err = filesys.FormatAndSaveGoFile(modelDirPath, tableTemplate.FileName, text)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return fmt.Errorf("failed to save '%s' model type: %w", tableTemplate.FileName, err)
|
return fmt.Errorf("failed to save '%s' model type: %w", tableTemplate.FileName, err)
|
||||||
}
|
}
|
||||||
|
|
@ -353,7 +353,7 @@ func processEnumModels(modelDir string, enumsMetaData []metadata.Enum, modelTemp
|
||||||
return fmt.Errorf("failed to generate enum type '%s': %w", enumMetaData.Name, err)
|
return fmt.Errorf("failed to generate enum type '%s': %w", enumMetaData.Name, err)
|
||||||
}
|
}
|
||||||
|
|
||||||
err = utils.SaveGoFile(modelDir, enumTemplate.FileName, text)
|
err = filesys.FormatAndSaveGoFile(modelDir, enumTemplate.FileName, text)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return fmt.Errorf("failed to save '%s' enum type: %w", enumTemplate.FileName, err)
|
return fmt.Errorf("failed to save '%s' enum type: %w", enumTemplate.FileName, err)
|
||||||
}
|
}
|
||||||
|
|
|
||||||
|
|
@ -3,7 +3,7 @@ package template
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
"github.com/go-jet/jet/v2/generator/metadata"
|
"github.com/go-jet/jet/v2/generator/metadata"
|
||||||
"github.com/go-jet/jet/v2/internal/utils"
|
"github.com/go-jet/jet/v2/internal/utils/dbidentifier"
|
||||||
"path"
|
"path"
|
||||||
"strings"
|
"strings"
|
||||||
"unicode"
|
"unicode"
|
||||||
|
|
@ -69,9 +69,9 @@ type ViewSQLBuilder = TableSQLBuilder
|
||||||
func DefaultTableSQLBuilder(tableMetaData metadata.Table) TableSQLBuilder {
|
func DefaultTableSQLBuilder(tableMetaData metadata.Table) TableSQLBuilder {
|
||||||
return TableSQLBuilder{
|
return TableSQLBuilder{
|
||||||
Path: "/table",
|
Path: "/table",
|
||||||
FileName: utils.ToGoFileName(tableMetaData.Name),
|
FileName: dbidentifier.ToGoFileName(tableMetaData.Name),
|
||||||
InstanceName: utils.ToGoIdentifier(tableMetaData.Name),
|
InstanceName: dbidentifier.ToGoIdentifier(tableMetaData.Name),
|
||||||
TypeName: utils.ToGoIdentifier(tableMetaData.Name) + "Table",
|
TypeName: dbidentifier.ToGoIdentifier(tableMetaData.Name) + "Table",
|
||||||
Column: DefaultTableSQLBuilderColumn,
|
Column: DefaultTableSQLBuilderColumn,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
@ -127,7 +127,7 @@ type TableSQLBuilderColumn struct {
|
||||||
// DefaultTableSQLBuilderColumn returns default implementation of TableSQLBuilderColumn
|
// DefaultTableSQLBuilderColumn returns default implementation of TableSQLBuilderColumn
|
||||||
func DefaultTableSQLBuilderColumn(columnMetaData metadata.Column) TableSQLBuilderColumn {
|
func DefaultTableSQLBuilderColumn(columnMetaData metadata.Column) TableSQLBuilderColumn {
|
||||||
return TableSQLBuilderColumn{
|
return TableSQLBuilderColumn{
|
||||||
Name: utils.ToGoIdentifier(columnMetaData.Name),
|
Name: dbidentifier.ToGoIdentifier(columnMetaData.Name),
|
||||||
Type: getSqlBuilderColumnType(columnMetaData),
|
Type: getSqlBuilderColumnType(columnMetaData),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
@ -185,8 +185,8 @@ type EnumSQLBuilder struct {
|
||||||
func DefaultEnumSQLBuilder(enumMetaData metadata.Enum) EnumSQLBuilder {
|
func DefaultEnumSQLBuilder(enumMetaData metadata.Enum) EnumSQLBuilder {
|
||||||
return EnumSQLBuilder{
|
return EnumSQLBuilder{
|
||||||
Path: "/enum",
|
Path: "/enum",
|
||||||
FileName: utils.ToGoFileName(enumMetaData.Name),
|
FileName: dbidentifier.ToGoFileName(enumMetaData.Name),
|
||||||
InstanceName: utils.ToGoIdentifier(enumMetaData.Name),
|
InstanceName: dbidentifier.ToGoIdentifier(enumMetaData.Name),
|
||||||
ValueName: func(enumValue string) string {
|
ValueName: func(enumValue string) string {
|
||||||
return defaultEnumValueName(enumMetaData.Name, enumValue)
|
return defaultEnumValueName(enumMetaData.Name, enumValue)
|
||||||
},
|
},
|
||||||
|
|
@ -217,9 +217,9 @@ func (e EnumSQLBuilder) UseInstanceName(name string) EnumSQLBuilder {
|
||||||
}
|
}
|
||||||
|
|
||||||
func defaultEnumValueName(enumName, enumValue string) string {
|
func defaultEnumValueName(enumName, enumValue string) string {
|
||||||
enumValueName := utils.ToGoIdentifier(enumValue)
|
enumValueName := dbidentifier.ToGoIdentifier(enumValue)
|
||||||
if !unicode.IsLetter([]rune(enumValueName)[0]) {
|
if !unicode.IsLetter([]rune(enumValueName)[0]) {
|
||||||
return utils.ToGoIdentifier(enumName) + enumValueName
|
return dbidentifier.ToGoIdentifier(enumName) + enumValueName
|
||||||
}
|
}
|
||||||
|
|
||||||
return enumValueName
|
return enumValueName
|
||||||
|
|
|
||||||
|
|
@ -1,8 +1,6 @@
|
||||||
package jet
|
package jet
|
||||||
|
|
||||||
import (
|
import "github.com/go-jet/jet/v2/internal/utils/is"
|
||||||
"github.com/go-jet/jet/v2/internal/utils"
|
|
||||||
)
|
|
||||||
|
|
||||||
// Clause interface
|
// Clause interface
|
||||||
type Clause interface {
|
type Clause interface {
|
||||||
|
|
@ -322,7 +320,7 @@ func (u *ClauseUpdate) Serialize(statementType StatementType, out *SQLBuilder, o
|
||||||
out.WriteString("UPDATE")
|
out.WriteString("UPDATE")
|
||||||
u.OptimizerHints.Serialize(statementType, out, options...)
|
u.OptimizerHints.Serialize(statementType, out, options...)
|
||||||
|
|
||||||
if utils.IsNil(u.Table) {
|
if is.Nil(u.Table) {
|
||||||
panic("jet: table to update is nil")
|
panic("jet: table to update is nil")
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
@ -387,7 +385,7 @@ func (i *ClauseInsert) GetColumns() []Column {
|
||||||
|
|
||||||
// Serialize serializes clause into SQLBuilder
|
// Serialize serializes clause into SQLBuilder
|
||||||
func (i *ClauseInsert) Serialize(statementType StatementType, out *SQLBuilder, options ...SerializeOption) {
|
func (i *ClauseInsert) Serialize(statementType StatementType, out *SQLBuilder, options ...SerializeOption) {
|
||||||
if utils.IsNil(i.Table) {
|
if is.Nil(i.Table) {
|
||||||
panic("jet: table is nil for INSERT clause")
|
panic("jet: table is nil for INSERT clause")
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -5,7 +5,7 @@ import (
|
||||||
"database/sql/driver"
|
"database/sql/driver"
|
||||||
"fmt"
|
"fmt"
|
||||||
"github.com/go-jet/jet/v2/internal/3rdparty/pq"
|
"github.com/go-jet/jet/v2/internal/3rdparty/pq"
|
||||||
"github.com/go-jet/jet/v2/internal/utils"
|
"github.com/go-jet/jet/v2/internal/utils/is"
|
||||||
"github.com/google/uuid"
|
"github.com/google/uuid"
|
||||||
"reflect"
|
"reflect"
|
||||||
"sort"
|
"sort"
|
||||||
|
|
@ -206,7 +206,7 @@ func (s *SQLBuilder) insertRawQuery(raw string, namedArg map[string]interface{})
|
||||||
}
|
}
|
||||||
|
|
||||||
func argToString(value interface{}) string {
|
func argToString(value interface{}) string {
|
||||||
if utils.IsNil(value) {
|
if is.Nil(value) {
|
||||||
return "NULL"
|
return "NULL"
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -1,8 +1,6 @@
|
||||||
package jet
|
package jet
|
||||||
|
|
||||||
import (
|
import "github.com/go-jet/jet/v2/internal/utils/is"
|
||||||
"github.com/go-jet/jet/v2/internal/utils"
|
|
||||||
)
|
|
||||||
|
|
||||||
// SerializerTable interface
|
// SerializerTable interface
|
||||||
type SerializerTable interface {
|
type SerializerTable interface {
|
||||||
|
|
@ -158,7 +156,7 @@ func (t *joinTableImpl) serialize(statement StatementType, out *SQLBuilder, opti
|
||||||
panic("jet: Join table is nil. ")
|
panic("jet: Join table is nil. ")
|
||||||
}
|
}
|
||||||
|
|
||||||
if utils.IsNil(t.lhs) {
|
if is.Nil(t.lhs) {
|
||||||
panic("jet: left hand side of join operation is nil table")
|
panic("jet: left hand side of join operation is nil table")
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
@ -179,7 +177,7 @@ func (t *joinTableImpl) serialize(statement StatementType, out *SQLBuilder, opti
|
||||||
out.WriteString("CROSS JOIN")
|
out.WriteString("CROSS JOIN")
|
||||||
}
|
}
|
||||||
|
|
||||||
if utils.IsNil(t.rhs) {
|
if is.Nil(t.rhs) {
|
||||||
panic("jet: right hand side of join operation is nil table")
|
panic("jet: right hand side of join operation is nil table")
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -1,7 +1,8 @@
|
||||||
package jet
|
package jet
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"github.com/go-jet/jet/v2/internal/utils"
|
"github.com/go-jet/jet/v2/internal/utils/dbidentifier"
|
||||||
|
"github.com/go-jet/jet/v2/internal/utils/must"
|
||||||
"reflect"
|
"reflect"
|
||||||
"strings"
|
"strings"
|
||||||
)
|
)
|
||||||
|
|
@ -150,11 +151,11 @@ func UnwindRowFromModel(columns []Column, data interface{}) []Serializer {
|
||||||
|
|
||||||
row := []Serializer{}
|
row := []Serializer{}
|
||||||
|
|
||||||
utils.ValueMustBe(structValue, reflect.Struct, "jet: data has to be a struct")
|
must.ValueBeOfTypeKind(structValue, reflect.Struct, "jet: data has to be a struct")
|
||||||
|
|
||||||
for _, column := range columns {
|
for _, column := range columns {
|
||||||
columnName := column.Name()
|
columnName := column.Name()
|
||||||
structFieldName := utils.ToGoIdentifier(columnName)
|
structFieldName := dbidentifier.ToGoIdentifier(columnName)
|
||||||
|
|
||||||
structField := structValue.FieldByName(structFieldName)
|
structField := structValue.FieldByName(structFieldName)
|
||||||
|
|
||||||
|
|
@ -179,7 +180,7 @@ func UnwindRowFromModel(columns []Column, data interface{}) []Serializer {
|
||||||
// UnwindRowsFromModels func
|
// UnwindRowsFromModels func
|
||||||
func UnwindRowsFromModels(columns []Column, data interface{}) [][]Serializer {
|
func UnwindRowsFromModels(columns []Column, data interface{}) [][]Serializer {
|
||||||
sliceValue := reflect.Indirect(reflect.ValueOf(data))
|
sliceValue := reflect.Indirect(reflect.ValueOf(data))
|
||||||
utils.ValueMustBe(sliceValue, reflect.Slice, "jet: data has to be a slice.")
|
must.ValueBeOfTypeKind(sliceValue, reflect.Slice, "jet: data has to be a slice.")
|
||||||
|
|
||||||
rows := [][]Serializer{}
|
rows := [][]Serializer{}
|
||||||
|
|
||||||
|
|
|
||||||
22
internal/utils/datetime/duration.go
Normal file
22
internal/utils/datetime/duration.go
Normal file
|
|
@ -0,0 +1,22 @@
|
||||||
|
package datetime
|
||||||
|
|
||||||
|
import "time"
|
||||||
|
|
||||||
|
// ExtractTimeComponents extracts number of days, hours, minutes, seconds, microseconds from duration
|
||||||
|
func ExtractTimeComponents(duration time.Duration) (days, hours, minutes, seconds, microseconds int64) {
|
||||||
|
days = int64(duration / (24 * time.Hour))
|
||||||
|
reminder := duration % (24 * time.Hour)
|
||||||
|
|
||||||
|
hours = int64(reminder / time.Hour)
|
||||||
|
reminder = reminder % time.Hour
|
||||||
|
|
||||||
|
minutes = int64(reminder / time.Minute)
|
||||||
|
reminder = reminder % time.Minute
|
||||||
|
|
||||||
|
seconds = int64(reminder / time.Second)
|
||||||
|
reminder = reminder % time.Second
|
||||||
|
|
||||||
|
microseconds = int64(reminder / time.Microsecond)
|
||||||
|
|
||||||
|
return
|
||||||
|
}
|
||||||
24
internal/utils/dbidentifier/dbidentifier.go
Normal file
24
internal/utils/dbidentifier/dbidentifier.go
Normal file
|
|
@ -0,0 +1,24 @@
|
||||||
|
package dbidentifier
|
||||||
|
|
||||||
|
import (
|
||||||
|
"github.com/go-jet/jet/v2/internal/3rdparty/snaker"
|
||||||
|
"strings"
|
||||||
|
)
|
||||||
|
|
||||||
|
// ToGoIdentifier converts database identifier to Go identifier.
|
||||||
|
func ToGoIdentifier(databaseIdentifier string) string {
|
||||||
|
return snaker.SnakeToCamel(replaceInvalidChars(databaseIdentifier))
|
||||||
|
}
|
||||||
|
|
||||||
|
// ToGoFileName converts database identifier to Go file name.
|
||||||
|
func ToGoFileName(databaseIdentifier string) string {
|
||||||
|
return strings.ToLower(replaceInvalidChars(databaseIdentifier))
|
||||||
|
}
|
||||||
|
|
||||||
|
func replaceInvalidChars(str string) string {
|
||||||
|
str = strings.Replace(str, " ", "_", -1)
|
||||||
|
str = strings.Replace(str, "-", "_", -1)
|
||||||
|
str = strings.Replace(str, ".", "_", -1)
|
||||||
|
|
||||||
|
return str
|
||||||
|
}
|
||||||
|
|
@ -1,7 +1,6 @@
|
||||||
package utils
|
package dbidentifier
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
|
||||||
"github.com/stretchr/testify/require"
|
"github.com/stretchr/testify/require"
|
||||||
"testing"
|
"testing"
|
||||||
)
|
)
|
||||||
|
|
@ -24,27 +23,3 @@ func TestToGoIdentifier(t *testing.T) {
|
||||||
require.Equal(t, ToGoIdentifier("My Table"), "MyTable")
|
require.Equal(t, ToGoIdentifier("My Table"), "MyTable")
|
||||||
require.Equal(t, ToGoIdentifier("My-Table"), "MyTable")
|
require.Equal(t, ToGoIdentifier("My-Table"), "MyTable")
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestErrorCatchErr(t *testing.T) {
|
|
||||||
var err error
|
|
||||||
|
|
||||||
func() {
|
|
||||||
defer ErrorCatch(&err)
|
|
||||||
|
|
||||||
panic(fmt.Errorf("newError"))
|
|
||||||
}()
|
|
||||||
|
|
||||||
require.Error(t, err, "newError")
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestErrorCatchNonErr(t *testing.T) {
|
|
||||||
var err error
|
|
||||||
|
|
||||||
func() {
|
|
||||||
defer ErrorCatch(&err)
|
|
||||||
|
|
||||||
panic(11)
|
|
||||||
}()
|
|
||||||
|
|
||||||
require.Error(t, err, "11")
|
|
||||||
}
|
|
||||||
85
internal/utils/filesys/filesys.go
Normal file
85
internal/utils/filesys/filesys.go
Normal file
|
|
@ -0,0 +1,85 @@
|
||||||
|
package filesys
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"go/format"
|
||||||
|
"os"
|
||||||
|
"path/filepath"
|
||||||
|
"strings"
|
||||||
|
)
|
||||||
|
|
||||||
|
// FormatAndSaveGoFile saves go file at folder dir, with name fileName and contents text.
|
||||||
|
func FormatAndSaveGoFile(dirPath, fileName string, text []byte) error {
|
||||||
|
newGoFilePath := filepath.Join(dirPath, fileName)
|
||||||
|
|
||||||
|
if !strings.HasSuffix(newGoFilePath, ".go") {
|
||||||
|
newGoFilePath += ".go"
|
||||||
|
}
|
||||||
|
|
||||||
|
file, err := os.Create(newGoFilePath)
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
defer file.Close()
|
||||||
|
|
||||||
|
p, err := format.Source(text)
|
||||||
|
|
||||||
|
// if there is a format error we will write unformulated text for debug purposes
|
||||||
|
if err != nil {
|
||||||
|
file.Write(text)
|
||||||
|
return fmt.Errorf("failed to format '%s', check '%s' for syntax errors: %w", fileName, newGoFilePath, err)
|
||||||
|
}
|
||||||
|
|
||||||
|
_, err = file.Write(p)
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("failed to save '%s' file: %w", newGoFilePath, err)
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// EnsureDirPathExist ensures dir path exists. If path does not exist, creates new path.
|
||||||
|
func EnsureDirPathExist(dirPath string) error {
|
||||||
|
if _, err := os.Stat(dirPath); os.IsNotExist(err) {
|
||||||
|
err := os.MkdirAll(dirPath, os.ModePerm)
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("can't create directory - %s: %w", dirPath, err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// RemoveDir deletes everything at folder dir.
|
||||||
|
func RemoveDir(dir string) error {
|
||||||
|
exist, err := DirExists(dir)
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
if exist {
|
||||||
|
err := os.RemoveAll(dir)
|
||||||
|
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// DirExists checks if folder at path exist.
|
||||||
|
func DirExists(path string) (bool, error) {
|
||||||
|
_, err := os.Stat(path)
|
||||||
|
if err == nil {
|
||||||
|
return true, nil
|
||||||
|
}
|
||||||
|
if os.IsNotExist(err) {
|
||||||
|
return false, nil
|
||||||
|
}
|
||||||
|
return true, err
|
||||||
|
}
|
||||||
8
internal/utils/is/is.go
Normal file
8
internal/utils/is/is.go
Normal file
|
|
@ -0,0 +1,8 @@
|
||||||
|
package is
|
||||||
|
|
||||||
|
import "reflect"
|
||||||
|
|
||||||
|
// Nil check if v is nil
|
||||||
|
func Nil(v interface{}) bool {
|
||||||
|
return v == nil || (reflect.ValueOf(v).Kind() == reflect.Ptr && reflect.ValueOf(v).IsNil())
|
||||||
|
}
|
||||||
41
internal/utils/must/must.go
Normal file
41
internal/utils/must/must.go
Normal file
|
|
@ -0,0 +1,41 @@
|
||||||
|
package must
|
||||||
|
|
||||||
|
import (
|
||||||
|
"github.com/go-jet/jet/v2/internal/utils/is"
|
||||||
|
"reflect"
|
||||||
|
)
|
||||||
|
|
||||||
|
// BeTrue panics when condition is false
|
||||||
|
func BeTrue(condition bool, errorStr string) {
|
||||||
|
if !condition {
|
||||||
|
panic(errorStr)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// BeTypeKind panics with errorStr error, if v interface is not of reflect kind
|
||||||
|
func BeTypeKind(v interface{}, kind reflect.Kind, errorStr string) {
|
||||||
|
if reflect.TypeOf(v).Kind() != kind {
|
||||||
|
panic(errorStr)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// ValueBeOfTypeKind panics with errorStr error, if v value is not of reflect kind
|
||||||
|
func ValueBeOfTypeKind(v reflect.Value, kind reflect.Kind, errorStr string) {
|
||||||
|
if v.Kind() != kind {
|
||||||
|
panic(errorStr)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// TypeBeOfKind panics with errorStr error, if v type is not of reflect kind
|
||||||
|
func TypeBeOfKind(v reflect.Type, kind reflect.Kind, errorStr string) {
|
||||||
|
if v.Kind() != kind {
|
||||||
|
panic(errorStr)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// BeInitializedPtr panics with errorStr if val interface is nil
|
||||||
|
func BeInitializedPtr(val interface{}, errorStr string) {
|
||||||
|
if is.Nil(val) {
|
||||||
|
panic(errorStr)
|
||||||
|
}
|
||||||
|
}
|
||||||
12
internal/utils/strslice/strslice.go
Normal file
12
internal/utils/strslice/strslice.go
Normal file
|
|
@ -0,0 +1,12 @@
|
||||||
|
package strslice
|
||||||
|
|
||||||
|
// Contains checks if slice of strings contains a string
|
||||||
|
func Contains(strings []string, contains string) bool {
|
||||||
|
for _, str := range strings {
|
||||||
|
if str == contains {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
@ -1,203 +0,0 @@
|
||||||
package utils
|
|
||||||
|
|
||||||
import (
|
|
||||||
"database/sql"
|
|
||||||
"fmt"
|
|
||||||
"github.com/go-jet/jet/v2/internal/3rdparty/snaker"
|
|
||||||
"go/format"
|
|
||||||
"os"
|
|
||||||
"path/filepath"
|
|
||||||
"reflect"
|
|
||||||
"strings"
|
|
||||||
"time"
|
|
||||||
)
|
|
||||||
|
|
||||||
// ToGoIdentifier converts database to Go identifier.
|
|
||||||
func ToGoIdentifier(databaseIdentifier string) string {
|
|
||||||
return snaker.SnakeToCamel(replaceInvalidChars(databaseIdentifier))
|
|
||||||
}
|
|
||||||
|
|
||||||
// ToGoFileName converts database identifier to Go file name.
|
|
||||||
func ToGoFileName(databaseIdentifier string) string {
|
|
||||||
return strings.ToLower(replaceInvalidChars(databaseIdentifier))
|
|
||||||
}
|
|
||||||
|
|
||||||
// SaveGoFile saves go file at folder dir, with name fileName and contents text.
|
|
||||||
func SaveGoFile(dirPath, fileName string, text []byte) error {
|
|
||||||
newGoFilePath := filepath.Join(dirPath, fileName)
|
|
||||||
|
|
||||||
if !strings.HasSuffix(newGoFilePath, ".go") {
|
|
||||||
newGoFilePath += ".go"
|
|
||||||
}
|
|
||||||
|
|
||||||
file, err := os.Create(newGoFilePath)
|
|
||||||
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
|
|
||||||
defer file.Close()
|
|
||||||
|
|
||||||
p, err := format.Source(text)
|
|
||||||
|
|
||||||
// if there is a format error we will write unformulated text for debug purposes
|
|
||||||
if err != nil {
|
|
||||||
file.Write(text)
|
|
||||||
return fmt.Errorf("failed to format '%s', check '%s' for syntax errors: %w", fileName, newGoFilePath, err)
|
|
||||||
}
|
|
||||||
|
|
||||||
_, err = file.Write(p)
|
|
||||||
if err != nil {
|
|
||||||
return fmt.Errorf("failed to save '%s' file: %w", newGoFilePath, err)
|
|
||||||
}
|
|
||||||
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// EnsureDirPath ensures dir path exists. If path does not exist, creates new path.
|
|
||||||
func EnsureDirPath(dirPath string) error {
|
|
||||||
if _, err := os.Stat(dirPath); os.IsNotExist(err) {
|
|
||||||
err := os.MkdirAll(dirPath, os.ModePerm)
|
|
||||||
|
|
||||||
if err != nil {
|
|
||||||
return fmt.Errorf("can't create directory - %s: %w", dirPath, err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// CleanUpGeneratedFiles deletes everything at folder dir.
|
|
||||||
func CleanUpGeneratedFiles(dir string) error {
|
|
||||||
exist, err := DirExists(dir)
|
|
||||||
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
|
|
||||||
if exist {
|
|
||||||
err := os.RemoveAll(dir)
|
|
||||||
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// DBClose closes non nil db connection
|
|
||||||
func DBClose(db *sql.DB) {
|
|
||||||
if db == nil {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
db.Close()
|
|
||||||
}
|
|
||||||
|
|
||||||
// DirExists checks if folder at path exist.
|
|
||||||
func DirExists(path string) (bool, error) {
|
|
||||||
_, err := os.Stat(path)
|
|
||||||
if err == nil {
|
|
||||||
return true, nil
|
|
||||||
}
|
|
||||||
if os.IsNotExist(err) {
|
|
||||||
return false, nil
|
|
||||||
}
|
|
||||||
return true, err
|
|
||||||
}
|
|
||||||
|
|
||||||
func replaceInvalidChars(str string) string {
|
|
||||||
str = strings.Replace(str, " ", "_", -1)
|
|
||||||
str = strings.Replace(str, "-", "_", -1)
|
|
||||||
str = strings.Replace(str, ".", "_", -1)
|
|
||||||
|
|
||||||
return str
|
|
||||||
}
|
|
||||||
|
|
||||||
// IsNil check if v is nil
|
|
||||||
func IsNil(v interface{}) bool {
|
|
||||||
return v == nil || (reflect.ValueOf(v).Kind() == reflect.Ptr && reflect.ValueOf(v).IsNil())
|
|
||||||
}
|
|
||||||
|
|
||||||
// MustBeTrue panics when condition is false
|
|
||||||
func MustBeTrue(condition bool, errorStr string) {
|
|
||||||
if !condition {
|
|
||||||
panic(errorStr)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// MustBe panics with errorStr error, if v interface is not of reflect kind
|
|
||||||
func MustBe(v interface{}, kind reflect.Kind, errorStr string) {
|
|
||||||
if reflect.TypeOf(v).Kind() != kind {
|
|
||||||
panic(errorStr)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// ValueMustBe panics with errorStr error, if v value is not of reflect kind
|
|
||||||
func ValueMustBe(v reflect.Value, kind reflect.Kind, errorStr string) {
|
|
||||||
if v.Kind() != kind {
|
|
||||||
panic(errorStr)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// TypeMustBe panics with errorStr error, if v type is not of reflect kind
|
|
||||||
func TypeMustBe(v reflect.Type, kind reflect.Kind, errorStr string) {
|
|
||||||
if v.Kind() != kind {
|
|
||||||
panic(errorStr)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// MustBeInitializedPtr panics with errorStr if val interface is nil
|
|
||||||
func MustBeInitializedPtr(val interface{}, errorStr string) {
|
|
||||||
if IsNil(val) {
|
|
||||||
panic(errorStr)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// ErrorCatch is used in defer to recover from panics and to set err
|
|
||||||
func ErrorCatch(err *error) {
|
|
||||||
recovered := recover()
|
|
||||||
|
|
||||||
if recovered == nil {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
recoveredErr, isError := recovered.(error)
|
|
||||||
|
|
||||||
if isError {
|
|
||||||
*err = recoveredErr
|
|
||||||
} else {
|
|
||||||
*err = fmt.Errorf("%v", recovered)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// StringSliceContains checks if slice of strings contains a string
|
|
||||||
func StringSliceContains(strings []string, contains string) bool {
|
|
||||||
for _, str := range strings {
|
|
||||||
if str == contains {
|
|
||||||
return true
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
|
|
||||||
// ExtractDateTimeComponents extracts number of days, hours, minutes, seconds, microseconds from duration
|
|
||||||
func ExtractDateTimeComponents(duration time.Duration) (days, hours, minutes, seconds, microseconds int64) {
|
|
||||||
days = int64(duration / (24 * time.Hour))
|
|
||||||
reminder := duration % (24 * time.Hour)
|
|
||||||
|
|
||||||
hours = int64(reminder / time.Hour)
|
|
||||||
reminder = reminder % time.Hour
|
|
||||||
|
|
||||||
minutes = int64(reminder / time.Minute)
|
|
||||||
reminder = reminder % time.Minute
|
|
||||||
|
|
||||||
seconds = int64(reminder / time.Second)
|
|
||||||
reminder = reminder % time.Second
|
|
||||||
|
|
||||||
microseconds = int64(reminder / time.Microsecond)
|
|
||||||
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
@ -2,11 +2,11 @@ package mysql
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
|
"github.com/go-jet/jet/v2/internal/utils/datetime"
|
||||||
"regexp"
|
"regexp"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"github.com/go-jet/jet/v2/internal/jet"
|
"github.com/go-jet/jet/v2/internal/jet"
|
||||||
"github.com/go-jet/jet/v2/internal/utils"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
type unitType string
|
type unitType string
|
||||||
|
|
@ -112,7 +112,7 @@ func INTERVALd(duration time.Duration) Interval {
|
||||||
duration = -duration
|
duration = -duration
|
||||||
}
|
}
|
||||||
|
|
||||||
days, hours, minutes, sec, microsec := utils.ExtractDateTimeComponents(duration)
|
days, hours, minutes, sec, microsec := datetime.ExtractTimeComponents(duration)
|
||||||
|
|
||||||
if days != 0 {
|
if days != 0 {
|
||||||
switch {
|
switch {
|
||||||
|
|
|
||||||
|
|
@ -3,7 +3,7 @@ package postgres
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
"github.com/go-jet/jet/v2/internal/jet"
|
"github.com/go-jet/jet/v2/internal/jet"
|
||||||
"github.com/go-jet/jet/v2/internal/utils"
|
"github.com/go-jet/jet/v2/internal/utils/datetime"
|
||||||
"strconv"
|
"strconv"
|
||||||
"strings"
|
"strings"
|
||||||
"time"
|
"time"
|
||||||
|
|
@ -148,9 +148,9 @@ func INTERVAL(quantityAndUnit ...quantityAndUnit) IntervalExpression {
|
||||||
|
|
||||||
// INTERVALd creates interval expression from time.Duration
|
// INTERVALd creates interval expression from time.Duration
|
||||||
func INTERVALd(duration time.Duration) IntervalExpression {
|
func INTERVALd(duration time.Duration) IntervalExpression {
|
||||||
days, hours, minutes, seconds, microseconds := utils.ExtractDateTimeComponents(duration)
|
days, hours, minutes, seconds, microseconds := datetime.ExtractTimeComponents(duration)
|
||||||
|
|
||||||
quantityAndUnits := []quantityAndUnit{}
|
var quantityAndUnits []quantityAndUnit
|
||||||
|
|
||||||
if days > 0 {
|
if days > 0 {
|
||||||
quantityAndUnits = append(quantityAndUnits, quantityAndUnit(days))
|
quantityAndUnits = append(quantityAndUnits, quantityAndUnit(days))
|
||||||
|
|
|
||||||
17
qrm/qrm.go
17
qrm/qrm.go
|
|
@ -5,9 +5,8 @@ import (
|
||||||
"database/sql"
|
"database/sql"
|
||||||
"errors"
|
"errors"
|
||||||
"fmt"
|
"fmt"
|
||||||
|
"github.com/go-jet/jet/v2/internal/utils/must"
|
||||||
"reflect"
|
"reflect"
|
||||||
|
|
||||||
"github.com/go-jet/jet/v2/internal/utils"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
// ErrNoRows is returned by Query when query result set is empty
|
// ErrNoRows is returned by Query when query result set is empty
|
||||||
|
|
@ -19,9 +18,9 @@ var ErrNoRows = errors.New("qrm: no rows in result set")
|
||||||
// If destination is pointer to struct and query result set is empty, method returns qrm.ErrNoRows.
|
// If destination is pointer to struct and query result set is empty, method returns qrm.ErrNoRows.
|
||||||
func Query(ctx context.Context, db Queryable, query string, args []interface{}, destPtr interface{}) (rowsProcessed int64, err error) {
|
func Query(ctx context.Context, db Queryable, query string, args []interface{}, destPtr interface{}) (rowsProcessed int64, err error) {
|
||||||
|
|
||||||
utils.MustBeInitializedPtr(db, "jet: db is nil")
|
must.BeInitializedPtr(db, "jet: db is nil")
|
||||||
utils.MustBeInitializedPtr(destPtr, "jet: destination is nil")
|
must.BeInitializedPtr(destPtr, "jet: destination is nil")
|
||||||
utils.MustBe(destPtr, reflect.Ptr, "jet: destination has to be a pointer to slice or pointer to struct")
|
must.BeTypeKind(destPtr, reflect.Ptr, "jet: destination has to be a pointer to slice or pointer to struct")
|
||||||
|
|
||||||
destinationPtrType := reflect.TypeOf(destPtr)
|
destinationPtrType := reflect.TypeOf(destPtr)
|
||||||
|
|
||||||
|
|
@ -64,8 +63,8 @@ func Query(ctx context.Context, db Queryable, query string, args []interface{},
|
||||||
|
|
||||||
// ScanOneRowToDest will scan one row into struct destination
|
// ScanOneRowToDest will scan one row into struct destination
|
||||||
func ScanOneRowToDest(scanContext *ScanContext, rows *sql.Rows, destPtr interface{}) error {
|
func ScanOneRowToDest(scanContext *ScanContext, rows *sql.Rows, destPtr interface{}) error {
|
||||||
utils.MustBeInitializedPtr(destPtr, "jet: destination is nil")
|
must.BeInitializedPtr(destPtr, "jet: destination is nil")
|
||||||
utils.MustBe(destPtr, reflect.Ptr, "jet: destination has to be a pointer to slice or pointer to struct")
|
must.BeTypeKind(destPtr, reflect.Ptr, "jet: destination has to be a pointer to slice or pointer to struct")
|
||||||
|
|
||||||
if len(scanContext.row) == 0 {
|
if len(scanContext.row) == 0 {
|
||||||
return errors.New("empty row slice")
|
return errors.New("empty row slice")
|
||||||
|
|
@ -149,7 +148,7 @@ func mapRowToSlice(
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
utils.TypeMustBe(sliceElemType, reflect.Struct, "jet: unsupported slice element type"+fieldToString(field))
|
must.TypeBeOfKind(sliceElemType, reflect.Struct, "jet: unsupported slice element type"+fieldToString(field))
|
||||||
|
|
||||||
structGroupKey := scanContext.getGroupKey(sliceElemType, field)
|
structGroupKey := scanContext.getGroupKey(sliceElemType, field)
|
||||||
|
|
||||||
|
|
@ -324,7 +323,7 @@ func mapRowToDestinationPtr(
|
||||||
destPtrValue reflect.Value,
|
destPtrValue reflect.Value,
|
||||||
structField *reflect.StructField) (updated bool, err error) {
|
structField *reflect.StructField) (updated bool, err error) {
|
||||||
|
|
||||||
utils.ValueMustBe(destPtrValue, reflect.Ptr, "jet: internal error. Destination is not pointer.")
|
must.ValueBeOfTypeKind(destPtrValue, reflect.Ptr, "jet: internal error. Destination is not pointer.")
|
||||||
|
|
||||||
destValueKind := destPtrValue.Elem().Kind()
|
destValueKind := destPtrValue.Elem().Kind()
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -3,7 +3,8 @@ package qrm
|
||||||
import (
|
import (
|
||||||
"database/sql"
|
"database/sql"
|
||||||
"fmt"
|
"fmt"
|
||||||
"github.com/go-jet/jet/v2/internal/utils"
|
"github.com/go-jet/jet/v2/internal/utils/must"
|
||||||
|
"github.com/go-jet/jet/v2/internal/utils/strslice"
|
||||||
"github.com/go-jet/jet/v2/qrm/internal"
|
"github.com/go-jet/jet/v2/qrm/internal"
|
||||||
"github.com/google/uuid"
|
"github.com/google/uuid"
|
||||||
"reflect"
|
"reflect"
|
||||||
|
|
@ -50,7 +51,7 @@ func getSliceElemPtrAt(slicePtrValue reflect.Value, index int) reflect.Value {
|
||||||
}
|
}
|
||||||
|
|
||||||
func appendElemToSlice(slicePtrValue reflect.Value, objPtrValue reflect.Value) error {
|
func appendElemToSlice(slicePtrValue reflect.Value, objPtrValue reflect.Value) error {
|
||||||
utils.MustBeTrue(!slicePtrValue.IsNil(), "jet: internal, slice is nil")
|
must.BeTrue(!slicePtrValue.IsNil(), "jet: internal, slice is nil")
|
||||||
|
|
||||||
sliceValue := slicePtrValue.Elem()
|
sliceValue := slicePtrValue.Elem()
|
||||||
sliceElemType := sliceValue.Type().Elem()
|
sliceElemType := sliceValue.Type().Elem()
|
||||||
|
|
@ -306,7 +307,7 @@ func setZeroValue(value reflect.Value) {
|
||||||
|
|
||||||
func isPrimaryKey(field reflect.StructField, primaryKeyOverwrites []string) bool {
|
func isPrimaryKey(field reflect.StructField, primaryKeyOverwrites []string) bool {
|
||||||
if len(primaryKeyOverwrites) > 0 {
|
if len(primaryKeyOverwrites) > 0 {
|
||||||
return utils.StringSliceContains(primaryKeyOverwrites, field.Name)
|
return strslice.Contains(primaryKeyOverwrites, field.Name)
|
||||||
}
|
}
|
||||||
|
|
||||||
sqlTag := field.Tag.Get("sql")
|
sqlTag := field.Tag.Get("sql")
|
||||||
|
|
|
||||||
|
|
@ -7,7 +7,7 @@ import (
|
||||||
mysql2 "github.com/go-jet/jet/v2/generator/mysql"
|
mysql2 "github.com/go-jet/jet/v2/generator/mysql"
|
||||||
"github.com/go-jet/jet/v2/generator/template"
|
"github.com/go-jet/jet/v2/generator/template"
|
||||||
"github.com/go-jet/jet/v2/internal/3rdparty/snaker"
|
"github.com/go-jet/jet/v2/internal/3rdparty/snaker"
|
||||||
"github.com/go-jet/jet/v2/internal/utils"
|
"github.com/go-jet/jet/v2/internal/utils/dbidentifier"
|
||||||
postgres2 "github.com/go-jet/jet/v2/postgres"
|
postgres2 "github.com/go-jet/jet/v2/postgres"
|
||||||
"github.com/go-jet/jet/v2/tests/dbconfig"
|
"github.com/go-jet/jet/v2/tests/dbconfig"
|
||||||
file2 "github.com/go-jet/jet/v2/tests/internal/utils/file"
|
file2 "github.com/go-jet/jet/v2/tests/internal/utils/file"
|
||||||
|
|
@ -157,17 +157,17 @@ func TestGeneratorTemplate_Model_RenameFilesAndTypes(t *testing.T) {
|
||||||
UseTable(func(table metadata.Table) template.TableModel {
|
UseTable(func(table metadata.Table) template.TableModel {
|
||||||
return template.DefaultTableModel(table).
|
return template.DefaultTableModel(table).
|
||||||
UseFileName(schemaMetaData.Name + "_" + table.Name).
|
UseFileName(schemaMetaData.Name + "_" + table.Name).
|
||||||
UseTypeName(utils.ToGoIdentifier(table.Name) + "Table")
|
UseTypeName(dbidentifier.ToGoIdentifier(table.Name) + "Table")
|
||||||
}).
|
}).
|
||||||
UseView(func(table metadata.Table) template.ViewModel {
|
UseView(func(table metadata.Table) template.ViewModel {
|
||||||
return template.DefaultViewModel(table).
|
return template.DefaultViewModel(table).
|
||||||
UseFileName(schemaMetaData.Name + "_" + table.Name + "_view").
|
UseFileName(schemaMetaData.Name + "_" + table.Name + "_view").
|
||||||
UseTypeName(utils.ToGoIdentifier(table.Name) + "View")
|
UseTypeName(dbidentifier.ToGoIdentifier(table.Name) + "View")
|
||||||
}).
|
}).
|
||||||
UseEnum(func(enumMetaData metadata.Enum) template.EnumModel {
|
UseEnum(func(enumMetaData metadata.Enum) template.EnumModel {
|
||||||
return template.DefaultEnumModel(enumMetaData).
|
return template.DefaultEnumModel(enumMetaData).
|
||||||
UseFileName(enumMetaData.Name + "_enum").
|
UseFileName(enumMetaData.Name + "_enum").
|
||||||
UseTypeName(utils.ToGoIdentifier(enumMetaData.Name) + "Enum")
|
UseTypeName(dbidentifier.ToGoIdentifier(enumMetaData.Name) + "Enum")
|
||||||
}),
|
}),
|
||||||
)
|
)
|
||||||
}),
|
}),
|
||||||
|
|
@ -256,19 +256,19 @@ func TestGeneratorTemplate_SQLBuilder_ChangeTypeAndFileName(t *testing.T) {
|
||||||
UseTable(func(table metadata.Table) template.TableSQLBuilder {
|
UseTable(func(table metadata.Table) template.TableSQLBuilder {
|
||||||
return template.DefaultTableSQLBuilder(table).
|
return template.DefaultTableSQLBuilder(table).
|
||||||
UseFileName(schemaMetaData.Name + "_" + table.Name + "_table").
|
UseFileName(schemaMetaData.Name + "_" + table.Name + "_table").
|
||||||
UseTypeName(utils.ToGoIdentifier(table.Name) + "TableSQLBuilder").
|
UseTypeName(dbidentifier.ToGoIdentifier(table.Name) + "TableSQLBuilder").
|
||||||
UseInstanceName("T_" + utils.ToGoIdentifier(table.Name))
|
UseInstanceName("T_" + dbidentifier.ToGoIdentifier(table.Name))
|
||||||
}).
|
}).
|
||||||
UseView(func(table metadata.Table) template.ViewSQLBuilder {
|
UseView(func(table metadata.Table) template.ViewSQLBuilder {
|
||||||
return template.DefaultViewSQLBuilder(table).
|
return template.DefaultViewSQLBuilder(table).
|
||||||
UseFileName(schemaMetaData.Name + "_" + table.Name + "_view").
|
UseFileName(schemaMetaData.Name + "_" + table.Name + "_view").
|
||||||
UseTypeName(utils.ToGoIdentifier(table.Name) + "ViewSQLBuilder").
|
UseTypeName(dbidentifier.ToGoIdentifier(table.Name) + "ViewSQLBuilder").
|
||||||
UseInstanceName("V_" + utils.ToGoIdentifier(table.Name))
|
UseInstanceName("V_" + dbidentifier.ToGoIdentifier(table.Name))
|
||||||
}).
|
}).
|
||||||
UseEnum(func(enum metadata.Enum) template.EnumSQLBuilder {
|
UseEnum(func(enum metadata.Enum) template.EnumSQLBuilder {
|
||||||
return template.DefaultEnumSQLBuilder(enum).
|
return template.DefaultEnumSQLBuilder(enum).
|
||||||
UseFileName(schemaMetaData.Name + "_" + enum.Name + "_enum").
|
UseFileName(schemaMetaData.Name + "_" + enum.Name + "_enum").
|
||||||
UseInstanceName(utils.ToGoIdentifier(enum.Name) + "EnumSQLBuilder")
|
UseInstanceName(dbidentifier.ToGoIdentifier(enum.Name) + "EnumSQLBuilder")
|
||||||
}),
|
}),
|
||||||
)
|
)
|
||||||
}),
|
}),
|
||||||
|
|
|
||||||
|
|
@ -8,7 +8,7 @@ import (
|
||||||
"github.com/go-jet/jet/v2/generator/template"
|
"github.com/go-jet/jet/v2/generator/template"
|
||||||
"github.com/go-jet/jet/v2/internal/3rdparty/snaker"
|
"github.com/go-jet/jet/v2/internal/3rdparty/snaker"
|
||||||
"github.com/go-jet/jet/v2/internal/testutils"
|
"github.com/go-jet/jet/v2/internal/testutils"
|
||||||
"github.com/go-jet/jet/v2/internal/utils"
|
"github.com/go-jet/jet/v2/internal/utils/dbidentifier"
|
||||||
postgres2 "github.com/go-jet/jet/v2/postgres"
|
postgres2 "github.com/go-jet/jet/v2/postgres"
|
||||||
"github.com/go-jet/jet/v2/tests/dbconfig"
|
"github.com/go-jet/jet/v2/tests/dbconfig"
|
||||||
file2 "github.com/go-jet/jet/v2/tests/internal/utils/file"
|
file2 "github.com/go-jet/jet/v2/tests/internal/utils/file"
|
||||||
|
|
@ -144,17 +144,17 @@ func TestGeneratorTemplate_Model_RenameFilesAndTypes(t *testing.T) {
|
||||||
UseTable(func(table metadata.Table) template.TableModel {
|
UseTable(func(table metadata.Table) template.TableModel {
|
||||||
return template.DefaultTableModel(table).
|
return template.DefaultTableModel(table).
|
||||||
UseFileName(schemaMetaData.Name + "_" + table.Name).
|
UseFileName(schemaMetaData.Name + "_" + table.Name).
|
||||||
UseTypeName(utils.ToGoIdentifier(table.Name) + "Table")
|
UseTypeName(dbidentifier.ToGoIdentifier(table.Name) + "Table")
|
||||||
}).
|
}).
|
||||||
UseView(func(table metadata.Table) template.ViewModel {
|
UseView(func(table metadata.Table) template.ViewModel {
|
||||||
return template.DefaultViewModel(table).
|
return template.DefaultViewModel(table).
|
||||||
UseFileName(schemaMetaData.Name + "_" + table.Name + "_view").
|
UseFileName(schemaMetaData.Name + "_" + table.Name + "_view").
|
||||||
UseTypeName(utils.ToGoIdentifier(table.Name) + "View")
|
UseTypeName(dbidentifier.ToGoIdentifier(table.Name) + "View")
|
||||||
}).
|
}).
|
||||||
UseEnum(func(enumMetaData metadata.Enum) template.EnumModel {
|
UseEnum(func(enumMetaData metadata.Enum) template.EnumModel {
|
||||||
return template.DefaultEnumModel(enumMetaData).
|
return template.DefaultEnumModel(enumMetaData).
|
||||||
UseFileName(enumMetaData.Name + "_enum").
|
UseFileName(enumMetaData.Name + "_enum").
|
||||||
UseTypeName(utils.ToGoIdentifier(enumMetaData.Name) + "Enum")
|
UseTypeName(dbidentifier.ToGoIdentifier(enumMetaData.Name) + "Enum")
|
||||||
}),
|
}),
|
||||||
)
|
)
|
||||||
}),
|
}),
|
||||||
|
|
@ -280,19 +280,19 @@ func TestGeneratorTemplate_SQLBuilder_ChangeTypeAndFileName(t *testing.T) {
|
||||||
UseTable(func(table metadata.Table) template.TableSQLBuilder {
|
UseTable(func(table metadata.Table) template.TableSQLBuilder {
|
||||||
return template.DefaultTableSQLBuilder(table).
|
return template.DefaultTableSQLBuilder(table).
|
||||||
UseFileName(schemaMetaData.Name + "_" + table.Name + "_table").
|
UseFileName(schemaMetaData.Name + "_" + table.Name + "_table").
|
||||||
UseTypeName(utils.ToGoIdentifier(table.Name) + "TableSQLBuilder").
|
UseTypeName(dbidentifier.ToGoIdentifier(table.Name) + "TableSQLBuilder").
|
||||||
UseInstanceName("T_" + utils.ToGoIdentifier(table.Name))
|
UseInstanceName("T_" + dbidentifier.ToGoIdentifier(table.Name))
|
||||||
}).
|
}).
|
||||||
UseView(func(table metadata.Table) template.ViewSQLBuilder {
|
UseView(func(table metadata.Table) template.ViewSQLBuilder {
|
||||||
return template.DefaultViewSQLBuilder(table).
|
return template.DefaultViewSQLBuilder(table).
|
||||||
UseFileName(schemaMetaData.Name + "_" + table.Name + "_view").
|
UseFileName(schemaMetaData.Name + "_" + table.Name + "_view").
|
||||||
UseTypeName(utils.ToGoIdentifier(table.Name) + "ViewSQLBuilder").
|
UseTypeName(dbidentifier.ToGoIdentifier(table.Name) + "ViewSQLBuilder").
|
||||||
UseInstanceName("V_" + utils.ToGoIdentifier(table.Name))
|
UseInstanceName("V_" + dbidentifier.ToGoIdentifier(table.Name))
|
||||||
}).
|
}).
|
||||||
UseEnum(func(enum metadata.Enum) template.EnumSQLBuilder {
|
UseEnum(func(enum metadata.Enum) template.EnumSQLBuilder {
|
||||||
return template.DefaultEnumSQLBuilder(enum).
|
return template.DefaultEnumSQLBuilder(enum).
|
||||||
UseFileName(schemaMetaData.Name + "_" + enum.Name + "_enum").
|
UseFileName(schemaMetaData.Name + "_" + enum.Name + "_enum").
|
||||||
UseInstanceName(utils.ToGoIdentifier(enum.Name) + "EnumSQLBuilder")
|
UseInstanceName(dbidentifier.ToGoIdentifier(enum.Name) + "EnumSQLBuilder")
|
||||||
}),
|
}),
|
||||||
)
|
)
|
||||||
}),
|
}),
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue