소스 검색

对gorm生成的外键表多s问题作出兼容代码

SliverHorn 4 년 전
부모
커밋
2ed73cae18
1개의 변경된 파일7개의 추가작업 그리고 1개의 파일을 삭제
  1. 7 1
      server/gva/init_data/init_data.go

+ 7 - 1
server/gva/init_data/init_data.go

@@ -411,7 +411,13 @@ func InitSysDataAuthorityId() (err error) {
 		{"9528", "8881"},
 		{"9528", "9528"},
 	}
-	if tx.Table("sys_data_authority_ids").Create(&insert).Error != nil { // 遇到错误时回滚事务
+	if global.GVA_DB.Migrator().HasTable("sys_data_authority_ids") {
+		if tx.Table("sys_data_authority_ids").Create(&insert).Error != nil { // 遇到错误时回滚事务
+			tx.Rollback()
+		}
+		return tx.Commit().Error
+	}
+	if tx.Table("sys_data_authority_id").Create(&insert).Error != nil { // 遇到错误时回滚事务
 		tx.Rollback()
 	}
 	return tx.Commit().Error