Add Method SingularTable

README.md

@@ -70,6 +70,8 @@ db, err := Open("postgres", "user=gorm dbname=gorm sslmode=disable")
 // Set the maximum idle database connections
 db.SetPool(100)
 
+// By default, table name is plural of struct type, if you like singular table name
+db.SingularTable(true)
 
 // Gorm is goroutines friendly, so you can create a global variable to keep the connection and use it everywhere like this
 

chain.go

@@ -12,6 +12,7 @@ type Chain struct {
 	db                *sql.DB
 	driver            string
 	debug             bool
+	singularTableName bool
 	value             interface{}
 
 	Errors []error
@@ -68,6 +69,7 @@ func (s *Chain) do(value interface{}) *Do {
 	do.limitStr = s.limitStr
 	do.specifiedTableName = s.specifiedTableName
 	do.unscoped = s.unscoped
+	do.singularTableName = s.singularTableName
 
 	s.value = value
 	do.setModel(value)

do.go

@@ -35,12 +35,13 @@ type Do struct {
 	unscoped             bool
 	updateAttrs          map[string]interface{}
 	ignoreProtectedAttrs bool
+	singularTableName    bool
 }
 
 func (s *Do) tableName() string {
 	if s.specifiedTableName == "" {
 		var err error
-		s.guessedTableName, err = s.model.tableName()
+		s.guessedTableName, err = s.model.tableName(s.singularTableName)
 		s.err(err)
 		return s.guessedTableName
 	} else {

gorm_test.go

@@ -81,7 +81,6 @@ var (
 func init() {
 	var err error
 	db, err = Open("postgres", "user=gorm dbname=gorm sslmode=disable")
-
 	// CREATE USER 'gorm'@'localhost' IDENTIFIED BY 'gorm';
 	// CREATE DATABASE 'gorm';
 	// GRANT ALL ON gorm.* TO 'gorm'@'localhost';

main.go

@@ -6,6 +6,7 @@ type DB struct {
 	db                *sql.DB
 	driver            string
 	DebugMode         bool
+	SingularTableName bool
 }
 
 func Open(driver, source string) (db DB, err error) {
@@ -18,8 +19,12 @@ func (s *DB) SetPool(n int) {
 	s.db.SetMaxIdleConns(n)
 }
 
+func (s *DB) SingularTable(result bool) {
+	s.SingularTableName = result
+}
+
 func (s *DB) buildChain() *Chain {
-	return &Chain{db: s.db, driver: s.driver, debug: s.DebugMode}
+	return &Chain{db: s.db, driver: s.driver, debug: s.DebugMode, singularTableName: s.SingularTableName}
 }
 
 func (s *DB) Where(querystring interface{}, args ...interface{}) *Chain {

model.go

@@ -276,7 +276,7 @@ func (m *Model) typeName() string {
 	return typ.Name()
 }
 
-func (m *Model) tableName() (str string, err error) {
+func (m *Model) tableName(singularTableName bool) (str string, err error) {
 	if m.data == nil {
 		err = errors.New("Model haven't been set")
 		return
@@ -284,6 +284,7 @@ func (m *Model) tableName() (str string, err error) {
 
 	str = toSnake(m.typeName())
 
+	if !singularTableName {
 		pluralMap := map[string]string{"ch": "ches", "ss": "sses", "sh": "shes", "day": "days", "y": "ies", "x": "xes", "s?": "s"}
 		for key, value := range pluralMap {
 			reg := regexp.MustCompile(key + "$")
@@ -291,6 +292,7 @@ func (m *Model) tableName() (str string, err error) {
 				return reg.ReplaceAllString(str, value), err
 			}
 		}
+	}
 	return
 }