-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
10 changed files
with
440 additions
and
102 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
|
|
@@ -39,3 +39,4 @@ build | |
| .env | ||
| Dockerfile-dev | ||
| sqlc.yaml | ||
| pgxgen.yaml | ||
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,14 +1,108 @@ | ||
| package config | ||
|
|
||
| import ( | ||
| "strings" | ||
|
|
||
| "github.com/tkcrm/pgxgen/utils" | ||
| ) | ||
|
|
||
| type Config struct { | ||
| Sqlc SqlcConfig | ||
| Pgxgen PgxgenConfig | ||
| } | ||
|
|
||
| type SqlcConfig struct { | ||
| Version int `yaml:"version"` | ||
| Packages []Packages `yaml:"packages"` | ||
| Version int `yaml:"version"` | ||
| Packages []Package `yaml:"packages"` | ||
| } | ||
|
|
||
| type Package struct { | ||
| EmitPreparedQueries bool `yaml:"emit_prepared_queries"` | ||
| Path string `yaml:"path"` | ||
| Engine string `yaml:"engine"` | ||
| EmitDbTags bool `yaml:"emit_db_tags"` | ||
| Name string `yaml:"name"` | ||
| EmitJsonTags bool `yaml:"emit_json_tags"` | ||
| EmitExactTableNames bool `yaml:"emit_exact_table_names"` | ||
| EmitEmptySlices bool `yaml:"emit_empty_slices"` | ||
| EmitResultStructPointers bool `yaml:"emit_result_struct_pointers"` | ||
| EmitParamsStructPointers bool `yaml:"emit_params_struct_pointers"` | ||
| Schema string `yaml:"schema"` | ||
| Queries string `yaml:"queries"` | ||
| SqlPackage string `yaml:"sql_package"` | ||
| EmitExportedQueries bool `yaml:"emit_exported_queries"` | ||
| EmitInterface bool `yaml:"emit_interface"` | ||
| } | ||
|
|
||
| type PgxgenConfig struct { | ||
| Version int `yaml:"version"` | ||
| OutputCrudSqlFileName string `yaml:"output_crud_sql_file_name"` | ||
| CrudParams CrudParams `yaml:"crud_params"` | ||
| } | ||
|
|
||
| type CrudParams struct { | ||
| Limit []string `yaml:"limit"` | ||
| OrderBy []OrderBy `yaml:"order_by"` | ||
| Where []Where `yaml:"where"` | ||
| } | ||
|
|
||
| // OrderBy used only for `Find` method | ||
| type OrderBy struct { | ||
| By string `yaml:"by"` | ||
| Order string `yaml:"order"` | ||
| Tables []string `yaml:"tables"` | ||
| } | ||
|
|
||
| // `Where` used for all method expect `create`. Instead of listing all methods, you can use an asterisk: * | ||
| type Where struct { | ||
| Methods string `yaml:"methods"` | ||
| Tables []string `yaml:"tables"` | ||
| Params []string `yaml:"params"` | ||
| } | ||
|
|
||
| func (p *PgxgenConfig) GetWhereParams(table, method string) []string { | ||
| params := []string{} | ||
|
|
||
| for _, item := range p.CrudParams.Where { | ||
| method = strings.ToLower(method) | ||
| if method == "c" || (method != "*" && !strings.Contains(strings.ToLower(item.Methods), method)) { | ||
| continue | ||
| } | ||
| if !utils.ExistInArray(item.Tables, "*") && !utils.ExistInArray(item.Tables, table) { | ||
| continue | ||
| } | ||
| for _, param := range item.Params { | ||
| if utils.ExistInArray(params, param) { | ||
| continue | ||
| } | ||
| params = append(params, param) | ||
| } | ||
| } | ||
|
|
||
| return params | ||
| } | ||
|
|
||
| func (p *PgxgenConfig) GetOrderByParams(table string) *OrderBy { | ||
|
|
||
| for _, item := range p.CrudParams.OrderBy { | ||
| if !utils.ExistInArray(item.Tables, "*") && !utils.ExistInArray(item.Tables, table) { | ||
| continue | ||
| } | ||
| if item.By == "" { | ||
| continue | ||
| } | ||
| if item.Order == "" { | ||
| item.Order = "DESC" | ||
| } | ||
| return &item | ||
| } | ||
|
|
||
| return nil | ||
| } | ||
|
|
||
| type Packages struct { | ||
| Path string `yaml:"path"` | ||
| Name string `yaml:"name"` | ||
| Engine string `yaml:"engine"` | ||
| Schema string `yaml:"schema"` | ||
| Queries string `yaml:"queries"` | ||
| func (p *PgxgenConfig) GetLimitParam(table string) bool { | ||
| if utils.ExistInArray(p.CrudParams.Limit, "*") { | ||
| return true | ||
| } | ||
| return utils.ExistInArray(p.CrudParams.Limit, table) | ||
| } |
Oops, something went wrong.