forked from go-gorp/gorp
-
Notifications
You must be signed in to change notification settings - Fork 1
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
In #1 we removed the Postgres dialect for simplicity. Per #7396 we'd like to be able to readily experiment with Postgres, so this restores it.
- Loading branch information
Showing
12 changed files
with
450 additions
and
10 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
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
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 |
---|---|---|
@@ -0,0 +1,150 @@ | ||
// Copyright 2012 James Cooper. All rights reserved. | ||
// Use of this source code is governed by a MIT-style | ||
// license that can be found in the LICENSE file. | ||
|
||
package borp | ||
|
||
import ( | ||
"context" | ||
"fmt" | ||
"reflect" | ||
"strings" | ||
"time" | ||
) | ||
|
||
type PostgresDialect struct { | ||
suffix string | ||
LowercaseFields bool | ||
} | ||
|
||
func (d PostgresDialect) QuerySuffix() string { return ";" } | ||
|
||
func (d PostgresDialect) ToSqlType(val reflect.Type, maxsize int, isAutoIncr bool) string { | ||
switch val.Kind() { | ||
case reflect.Ptr: | ||
return d.ToSqlType(val.Elem(), maxsize, isAutoIncr) | ||
case reflect.Bool: | ||
return "boolean" | ||
case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Uint, reflect.Uint8, reflect.Uint16, reflect.Uint32: | ||
if isAutoIncr { | ||
return "serial" | ||
} | ||
return "integer" | ||
case reflect.Int64, reflect.Uint64: | ||
if isAutoIncr { | ||
return "bigserial" | ||
} | ||
return "bigint" | ||
case reflect.Float64: | ||
return "double precision" | ||
case reflect.Float32: | ||
return "real" | ||
case reflect.Slice: | ||
if val.Elem().Kind() == reflect.Uint8 { | ||
return "bytea" | ||
} | ||
} | ||
|
||
switch val.Name() { | ||
case "NullInt64": | ||
return "bigint" | ||
case "NullFloat64": | ||
return "double precision" | ||
case "NullBool": | ||
return "boolean" | ||
case "Time", "NullTime": | ||
return "timestamp with time zone" | ||
} | ||
|
||
if maxsize > 0 { | ||
return fmt.Sprintf("varchar(%d)", maxsize) | ||
} else { | ||
return "text" | ||
} | ||
|
||
} | ||
|
||
// Returns empty string | ||
func (d PostgresDialect) AutoIncrStr() string { | ||
return "" | ||
} | ||
|
||
func (d PostgresDialect) AutoIncrBindValue() string { | ||
return "default" | ||
} | ||
|
||
func (d PostgresDialect) AutoIncrInsertSuffix(col *ColumnMap) string { | ||
return " returning " + d.QuoteField(col.ColumnName) | ||
} | ||
|
||
// Returns suffix | ||
func (d PostgresDialect) CreateTableSuffix() string { | ||
return d.suffix | ||
} | ||
|
||
func (d PostgresDialect) CreateIndexSuffix() string { | ||
return "using" | ||
} | ||
|
||
func (d PostgresDialect) DropIndexSuffix() string { | ||
return "" | ||
} | ||
|
||
func (d PostgresDialect) TruncateClause() string { | ||
return "truncate" | ||
} | ||
|
||
func (d PostgresDialect) SleepClause(s time.Duration) string { | ||
return fmt.Sprintf("pg_sleep(%f)", s.Seconds()) | ||
} | ||
|
||
// Returns "$(i+1)" | ||
func (d PostgresDialect) BindVar(i int) string { | ||
return fmt.Sprintf("$%d", i+1) | ||
} | ||
|
||
func (d PostgresDialect) InsertAutoIncrToTarget(ctx context.Context, exec SqlExecutor, insertSql string, target interface{}, params ...interface{}) error { | ||
rows, err := exec.QueryContext(ctx, insertSql, params...) | ||
if err != nil { | ||
return err | ||
} | ||
defer rows.Close() | ||
|
||
if !rows.Next() { | ||
return fmt.Errorf("No serial value returned for insert: %s Encountered error: %s", insertSql, rows.Err()) | ||
} | ||
if err := rows.Scan(target); err != nil { | ||
return err | ||
} | ||
if rows.Next() { | ||
return fmt.Errorf("more than two serial value returned for insert: %s", insertSql) | ||
} | ||
return rows.Err() | ||
} | ||
|
||
func (d PostgresDialect) QuoteField(f string) string { | ||
if d.LowercaseFields { | ||
return `"` + strings.ToLower(f) + `"` | ||
} | ||
return `"` + f + `"` | ||
} | ||
|
||
func (d PostgresDialect) QuotedTableForQuery(schema string, table string) string { | ||
if strings.TrimSpace(schema) == "" { | ||
return d.QuoteField(table) | ||
} | ||
|
||
return schema + "." + d.QuoteField(table) | ||
} | ||
|
||
func (d PostgresDialect) IfSchemaNotExists(command, schema string) string { | ||
return fmt.Sprintf("%s if not exists", command) | ||
} | ||
|
||
func (d PostgresDialect) IfTableExists(command, schema, table string) string { | ||
return fmt.Sprintf("%s if exists", command) | ||
} | ||
|
||
func (d PostgresDialect) IfTableNotExists(command, schema, table string) string { | ||
return fmt.Sprintf("%s if not exists", command) | ||
} |
Oops, something went wrong.