fix: fix get primaryKey name error

acmestack · Jan 16, 2023 · 11546ce · 11546ce
1 parent af0c087
commit 11546ce
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/example/base/user.go b/example/base/user.go
@@ -24,7 +24,7 @@ import (
 // +gplus:column=true
 
 type User struct {
-	ID        int64
+	ID        int64  `gorm:"primaryKey"`
 	Username  string `gorm:"column:username"`
 	Password  string
 	Address   string

diff --git a/gplus/base_dao.go b/gplus/base_dao.go
@@ -235,7 +235,12 @@ func getPKColumn[T any]() string {
 		tagSetting := schema.ParseTagSetting(field.Tag.Get("gorm"), ";")
 		isPrimaryKey := utils.CheckTruth(tagSetting["PRIMARYKEY"], tagSetting["PRIMARY_KEY"])
 		if isPrimaryKey {
-			columnName = tagSetting["COLUMN"]
+			name, ok := tagSetting["COLUMN"]
+			if !ok {
+				namingStrategy := schema.NamingStrategy{}
+				name = namingStrategy.ColumnName("", field.Name)
+			}
+			columnName = name
 			break
 		}
 	}