调整提示

mybatis-plus/src/main/java/com/baomidou/framework/service/impl/ServiceImpl.java

@@ -73,11 +73,7 @@ public class ServiceImpl<M extends BaseMapper<T, PK>, T, PK extends Serializable
 			TableInfo tableInfo = TableInfoHelper.getTableInfo(cls);
 			if (null != tableInfo) {
 				try {
-					String keyProperty = tableInfo.getKeyProperty();
-					if (keyProperty == null) {
-						throw new MybatisPlusException("Error:  Cannot execute. Could not find @TableId.");
-					}
-					Method m = cls.getMethod("get" + StringUtils.capitalize(keyProperty));
+					Method m = cls.getMethod("get" + StringUtils.capitalize(tableInfo.getKeyProperty()));
 					Object idVal = m.invoke(entity);
 					if (null != idVal) {
 						return selective ? updateSelectiveById(entity) : updateById(entity);
@@ -87,6 +83,8 @@ public class ServiceImpl<M extends BaseMapper<T, PK>, T, PK extends Serializable
 				} catch (Exception e) {
 					e.printStackTrace();
 				}
+			} else {
+				throw new MybatisPlusException("Error:  Cannot execute. Could not find @TableId.");
 			}
 		}
 		return false;