This commit is contained in:
Redkale
2020-01-30 11:35:05 +08:00
parent c36fc36964
commit ee8a18a528

View File

@@ -97,6 +97,9 @@ public final class EntityInfo<T> {
//用于存在database.table_20160202类似这种分布式表 //用于存在database.table_20160202类似这种分布式表
private final Set<String> tables = new CopyOnWriteArraySet<>(); private final Set<String> tables = new CopyOnWriteArraySet<>();
//不能为null的字段名
private final Set<String> notNullColumns = new CopyOnWriteArraySet<>();
//分表 策略 //分表 策略
private final DistributeTableStrategy<T> tableStrategy; private final DistributeTableStrategy<T> tableStrategy;
@@ -322,6 +325,9 @@ public final class EntityInfo<T> {
updateattrs.add(attr); updateattrs.add(attr);
updateAttributeMap.put(fieldname, attr); updateAttributeMap.put(fieldname, attr);
} }
if (col != null && !col.nullable()) {
notNullColumns.add(fieldname);
}
} }
queryattrs.add(attr); queryattrs.add(attr);
fields.add(fieldname); fields.add(fieldname);
@@ -1070,6 +1076,14 @@ public final class EntityInfo<T> {
return logger.isLoggable(l) && l.intValue() >= this.logLevel; return logger.isLoggable(l) && l.intValue() >= this.logLevel;
} }
public boolean isNotNullable(String fieldname) {
return notNullColumns.contains(fieldname);
}
public boolean isNotNullable(Attribute<T, Serializable> attr) {
return attr == null ? false : notNullColumns.contains(attr.field());
}
/** /**
* 判断日志级别 * 判断日志级别
* *