org.yx.db.sql.Update Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of sumk Show documentation
Show all versions of sumk Show documentation
A quick developing framewort for internet company
/**
* Copyright (C) 2016 - 2030 youtongluan.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.yx.db.sql;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import org.yx.common.ItemJoiner;
import org.yx.db.enums.ColumnType;
import org.yx.db.event.UpdateEvent;
import org.yx.db.kit.NumUtil;
import org.yx.db.visit.SumkDbVisitor;
import org.yx.exception.SumkException;
import org.yx.util.CollectionUtil;
public class Update extends AbstractSqlBuilder implements Executable {
private boolean fullUpdate;
private Map updateTo;
private boolean updateDBID;
private Map incrMap;
/**
* @param update
* 如果为false，则数据库主键不会被更新。默认为false。
* @return 当前对象
*/
public Update updateDBID(boolean update) {
this.updateDBID = update;
return this;
}
/**
*
* @param fail
* 如果为true，会验证map参数中，是否存在无效的key，预防开发人员将key写错。默认为true
* @return 当前对象
*/
public Update failIfPropertyNotMapped(boolean fail) {
this.failIfPropertyNotMapped = fail;
return this;
}
/**
* 设置where条件，如果没有设置该条件。就用pojo的数据库主键或者redis主键
*
* - 本方法可以被多次调用，多次调用之间是OR关系
* - 注意：如果本表使用了缓存，本参数必须包含所有redis主键
* - 注意：如果pojo是map类型，那么它的null值是有效条件
*
*
* @param pojo
* bean类型或Map.如果是pojo对象，其中的null字段会被忽略掉
* @return 当前对象
*/
public Update addWhere(Object pojo) {
this._addIn(pojo);
return this;
}
@Override
public int execute() {
return this.accept(visitor);
}
public Update(SumkDbVisitor visitor) {
super(visitor);
}
/**
* @param fullUpdate
* 设置为true的话，整条记录全部更新，包括null字段。默认为false
* @return 当前对象
*/
public Update fullUpdate(boolean fullUpdate) {
this.fullUpdate = fullUpdate;
return this;
}
/**
* 分表的情况下，设置分区名。这个方法只能调用一次
*
* @param sub
* 分区名
* @return 当前对象
*/
public Update partition(String sub) {
sub(sub);
return this;
}
/**
* 记录被更新后的最终状态。
*
* - 有可能是部分字段，有可能是全部字段
* - 有可能只是单条记录变成这样，有可能是多条记录变成这样
*
*
* @param pojo
* Pojo或Map类型.如果是Map类型，要设置tableClass。
* 如果本表使用了缓存，并且没有where条件，本参数必须包含所有redis主键
* 如果本字段包含在自增长里面，那它将会被排除掉
* @return 当前对象
*/
@SuppressWarnings("unchecked")
public Update updateTo(Object pojo) {
if (Map.class.isInstance(pojo)) {
this.updateTo = new HashMap<>((Map) pojo);
return this;
}
this.pojoMeta = PojoMetaHolder.getPojoMeta(pojo.getClass(), this.sub);
if (this.pojoMeta == null) {
SumkException.throwException(36541, pojo.getClass() + " does not config as a table");
}
try {
this.updateTo = this.pojoMeta.populate(pojo, true);
} catch (Exception e) {
SumkException.throwException(-345461, e.getMessage(), e);
}
return this;
}
public Update tableClass(Class tableClass) {
this.tableClass = tableClass;
return this;
}
public MapedSql toMapedSql() throws Exception {
if (this.updateTo == null || this.updateTo.isEmpty()) {
SumkException.throwException(-34601, "updateTo is null or empty");
}
this.pojoMeta = this.parsePojoMeta(true);
this.checkMap(this.updateTo, this.pojoMeta);
if (CollectionUtil.isEmpty(this.in)) {
this.addDBIDs2Where();
}
return _toMapedSql();
}
protected MapedSql _toMapedSql() throws Exception {
MapedSql ms = new MapedSql();
StringBuilder sb = new StringBuilder(32);
List fms = pojoMeta.fieldMetas;
SoftDeleteMeta softDelete = pojoMeta.softDelete;
sb.append("UPDATE ").append(pojoMeta.getTableName());
boolean notFirst = false;
Map to = new HashMap<>(this.updateTo);
for (ColumnMeta fm : fms) {
String fieldName = fm.getFieldName();
if (this.incrMap != null && this.incrMap.containsKey(fieldName)) {
to.remove(fieldName);
sb.append(notFirst ? " , " : " SET ").append(fm.dbColumn).append('=').append(fm.dbColumn)
.append(" +? ");
notFirst = true;
ms.addParam(this.incrMap.get(fieldName));
continue;
}
Object value = fm.value(this.updateTo);
if (value == null && !this.fullUpdate) {
continue;
}
if (fm.accept(ColumnType.ID_DB) && !this.updateDBID) {
continue;
}
sb.append(notFirst ? " , " : " SET ");
sb.append(fm.dbColumn).append("=? ");
notFirst = true;
ms.addParam(value);
}
ItemJoiner orItem = new ItemJoiner(" OR ", " WHERE ", null);
boolean isSingle = in.size() == 1;
for (Map where : this.in) {
if (where.isEmpty()) {
continue;
}
this.checkMap(where, this.pojoMeta);
ItemJoiner andItem = isSingle ? new ItemJoiner(" AND ", null, null) : new ItemJoiner(" AND ", " ( ", " ) ");
for (ColumnMeta fm : fms) {
Object value = null;
if (!where.containsKey(fm.getFieldName())) {
continue;
}
value = where.get(fm.getFieldName());
if (value == null) {
andItem.item().append(fm.dbColumn).append(" IS NULL ");
continue;
}
andItem.item().append(fm.dbColumn).append("=? ");
ms.addParam(value);
}
if (andItem.isEmpty()) {
continue;
}
if (softDelete != null) {
if (softDelete.equalValid) {
andItem.item().append(softDelete.columnName).append("=? ");
ms.addParam(softDelete.validValue);
} else {
andItem.item().append(softDelete.columnName).append(" != ? ");
ms.addParam(softDelete.inValidValue);
}
}
CharSequence one = andItem.toCharSequence();
orItem.item().append(one);
}
CharSequence whereStr = orItem.toCharSequence(true);
if (whereStr == null || whereStr.length() == 0) {
SumkException.throwException(345445, "where cannot be null");
}
sb.append(whereStr);
ms.sql = sb.toString();
ms.event = new UpdateEvent(pojoMeta.getTableName(), to, this.incrMap, this.in, this.fullUpdate,
this.updateDBID);
return ms;
}
protected void addDBIDs2Where() throws Exception {
List whereColumns = this.pojoMeta.getPrimaryIDs();
Map paramMap = new HashMap<>();
for (ColumnMeta fm : whereColumns) {
String fieldName = fm.getFieldName();
paramMap.put(fieldName, fm.value(this.updateTo));
}
this._addIn(paramMap, true);
}
/**
* 增加或减少表中数字类型字段的值
*
* @param fieldName
* 需要增长或减少的字段的名字,不能是redis主键
* @param v
* 如果是减少，直接用负数就行了
* @return 当前对象
*/
public Update incrNum(String fieldName, Number v) {
if (v == null) {
throw new SumkException(5349238, "cannot incr " + fieldName + "(java) by null");
}
PojoMeta pm = this.parsePojoMeta(true);
ColumnMeta columnMeta = pm.getByFieldName(fieldName);
if (columnMeta == null) {
throw new SumkException(5912239, "cannot found java field " + fieldName + " in " + pm.pojoClz);
}
v = NumUtil.toType(v, columnMeta.field.getType(), true);
if (this.incrMap == null) {
this.incrMap = new HashMap<>();
}
this.incrMap.put(fieldName, v);
return this;
}
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy