diff options
Diffstat (limited to 'internal/plan.go')
| -rw-r--r-- | internal/plan.go | 24 |
1 files changed, 24 insertions, 0 deletions
diff --git a/internal/plan.go b/internal/plan.go index ac199bf..7dc3361 100644 --- a/internal/plan.go +++ b/internal/plan.go @@ -23,6 +23,7 @@ type Plan struct { IndexByColumnName map[string][]int // Planned queries. + Select PlannedQuery // only `SELECT ... FROM ...` without WHERE or any of the other clauses Insert PlannedQuery Update PlannedQuery Delete PlannedQuery @@ -126,6 +127,7 @@ func buildPlan(t reflect.Type, dialect Dialect, opts PlanOpts) (Plan, error) { } // prepare query strings + p.Select = p.buildSelectQueryIfPossible(dialect) p.Insert = p.buildInsertQueryIfPossible(dialect) p.Update = p.buildUpdateQueryIfPossible(dialect) p.Delete = p.buildDeleteQueryIfPossible(dialect) @@ -153,6 +155,28 @@ func (p Plan) getNonPrimaryKeyColumnNames() []string { return result } +func (p Plan) buildSelectQueryIfPossible(dialect Dialect) PlannedQuery { + if p.TableName == "" { + return PlannedQuery{Query: ""} + } + + var ( + argumentIndexes = make([][]int, len(p.AllColumnNames)) + quotedColumnNames = make([]string, len(p.AllColumnNames)) + ) + for idx, columnName := range p.AllColumnNames { + argumentIndexes[idx] = p.IndexByColumnName[columnName] + quotedColumnNames[idx] = dialect.QuoteIdentifier(columnName) + } + + query := fmt.Sprintf( + `SELECT %s FROM %s WHERE `, + strings.Join(quotedColumnNames, ", "), + dialect.QuoteIdentifier(p.TableName), + ) + return PlannedQuery{query, argumentIndexes} +} + func (p Plan) buildInsertQueryIfPossible(dialect Dialect) PlannedQuery { if p.TableName == "" || len(p.AllColumnNames) == 0 { return PlannedQuery{Query: ""} |