123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110 |
- package mysql
- import (
- "database/sql"
- "fmt"
- "math"
- _ "github.com/go-sql-driver/mysql"
- "github.com/jinzhu/gorm"
- "github.com/jinzhu/gorm/callbacks"
- "github.com/jinzhu/gorm/migrator"
- "github.com/jinzhu/gorm/schema"
- )
- type Dialector struct {
- DSN string
- }
- func Open(dsn string) gorm.Dialector {
- return &Dialector{DSN: dsn}
- }
- func (dialector Dialector) Initialize(db *gorm.DB) (err error) {
- // register callbacks
- callbacks.RegisterDefaultCallbacks(db)
- db.DB, err = sql.Open("mysql", dialector.DSN)
- return
- }
- func (dialector Dialector) Migrator(db *gorm.DB) gorm.Migrator {
- return Migrator{migrator.Migrator{Config: migrator.Config{
- DB: db,
- Dialector: dialector,
- }}}
- }
- func (dialector Dialector) BindVar(stmt *gorm.Statement, v interface{}) string {
- return "?"
- }
- func (dialector Dialector) QuoteChars() [2]byte {
- return [2]byte{'`', '`'} // `name`
- }
- func (dialector Dialector) DataTypeOf(field *schema.Field) string {
- switch field.DataType {
- case schema.Bool:
- return "boolean"
- case schema.Int, schema.Uint:
- sqlType := "int"
- switch {
- case field.Size <= 8:
- sqlType = "tinyint"
- case field.Size <= 16:
- sqlType = "smallint"
- case field.Size <= 32:
- sqlType = "int"
- default:
- sqlType = "bigint"
- }
- if field.DataType == schema.Uint {
- sqlType += " unsigned"
- }
- if field.AutoIncrement {
- sqlType += " AUTO_INCREMENT"
- }
- return sqlType
- case schema.Float:
- if field.Size <= 32 {
- return "float"
- }
- return "double"
- case schema.String:
- size := field.Size
- if field.PrimaryKey && size == 0 {
- size = 256
- }
- if size >= 65536 && size <= int(math.Pow(2, 24)) {
- return "mediumtext"
- } else if size > int(math.Pow(2, 24)) || size <= 0 {
- return "longtext"
- }
- return fmt.Sprintf("varchar(%d)", size)
- case schema.Time:
- precision := ""
- if field.Precision > 0 {
- precision = fmt.Sprintf("(%d)", field.Precision)
- }
- if field.NotNull || field.PrimaryKey {
- return "datetime" + precision
- }
- return "datetime" + precision + " NULL"
- case schema.Bytes:
- if field.Size > 0 && field.Size < 65536 {
- return fmt.Sprintf("varbinary(%d)", field.Size)
- }
- if field.Size >= 65536 && field.Size <= int(math.Pow(2, 24)) {
- return "mediumblob"
- }
- return "longblob"
- }
- return ""
- }
|