diff --git a/options.go b/options.go index a5705cc..a4720ae 100644 --- a/options.go +++ b/options.go @@ -5,7 +5,7 @@ type clientOptions struct { Mysql *MysqlOptions Sqlite *SqliteOptions Prefix string // 表名前缀,`Article` 的表名应该是 `it_articles` - SingularTable bool // 使用单数表名,启用该选项,此时,`Article` 的表名应该是 `it_article` + SingularTable bool // 使用单数表名,启用该选项,此时,`Article` 的表名应该是 `it_article` } type MysqlOptions struct { @@ -38,18 +38,27 @@ func SetMysql(user, password, host, database string, port int) Option { Database: database, Password: password, } + if o.Mysql.Port == 0 { + o.Mysql.Port = 3306 + } } } +// 使用sqlite数据库 func SetSqlite(dbPath string) Option { return func(o *clientOptions) { o.Db = "sqlite" o.Sqlite = &SqliteOptions{ DbPath: dbPath, } + if o.Sqlite.DbPath == "" { + o.Sqlite.DbPath = "sqlite.db" + } } } + +// 表名前缀 func SetTablePrefix(perfix string) Option { return func(o *clientOptions) { o.Prefix = perfix