You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
29 lines
770 B
29 lines
770 B
package dorm
|
|
|
|
import "reflect"
|
|
|
|
// Database 设置库名
|
|
func (c *MongoClient) Database(databaseName string) *MongoClient {
|
|
c.databaseName = databaseName
|
|
return c
|
|
}
|
|
|
|
// Collection 设置表名
|
|
func (c *MongoClient) Collection(collectionName string) *MongoClient {
|
|
c.collectionName = collectionName
|
|
return c
|
|
}
|
|
|
|
// Model 传入模型自动获取库名和表名
|
|
func (c *MongoClient) Model(value interface{}) *MongoClient {
|
|
// https://studygolang.com/articles/896
|
|
val := reflect.ValueOf(value)
|
|
if methodValue := val.MethodByName("Database"); methodValue.IsValid() {
|
|
c.databaseName = methodValue.Call(nil)[0].String()
|
|
}
|
|
if methodValue := val.MethodByName("TableName"); methodValue.IsValid() {
|
|
c.collectionName = methodValue.Call(nil)[0].String()
|
|
}
|
|
return c
|
|
}
|