net.guerlab.cloud.searchparams.mybatisplus.DefaultHandler Maven / Gradle / Ivy

Go to download
/*
 * Copyright 2018-2025 guerlab.net and other contributors.
 *
 * Licensed under the GNU LESSER GENERAL PUBLIC LICENSE, Version 3 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      https://www.gnu.org/licenses/lgpl-3.0.html
 *
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package net.guerlab.cloud.searchparams.mybatisplus;

import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import jakarta.annotation.Nullable;
import org.apache.commons.lang3.StringUtils;

import net.guerlab.cloud.searchparams.JsonField;
import net.guerlab.cloud.searchparams.SearchModelType;
import net.guerlab.cloud.searchparams.SearchParamsHandler;

/**
 * 默认处理.
 *
 * @author guer
 */
public class DefaultHandler implements SearchParamsHandler {

	@Override
	public void setValue(Object object, String fieldName, String columnName, Object value,
			SearchModelType searchModelType, @Nullable String customSql, @Nullable JsonField jsonField) {
		QueryWrapper wrapper = (QueryWrapper) object;
		columnName = ColumnNameGetter.getColumnName(columnName, wrapper.getEntityClass());

		if (jsonField != null) {
			setValueWithJsonField(wrapper, columnName, value, searchModelType, jsonField);
		}
		else {
			setValueWithoutJsonField(wrapper, columnName, value, searchModelType, customSql);
		}
	}

	private void setValueWithJsonField(QueryWrapper wrapper, String columnName, Object value,
			SearchModelType searchModelType, JsonField jsonField) {
		DbType dbType = DbTypeUtils.getDbType(wrapper);
		String jsonPath = getJsonPath(jsonField);
		String sqlTemplate = dbType.formatJsonQuerySql(columnName, searchModelType, jsonPath, 1);
		if (sqlTemplate == null) {
			return;
		}

		wrapper.apply(sqlTemplate, dbType.jsonQueryValueFormat(value, searchModelType, jsonPath));
	}

	private String getJsonPath(JsonField jsonField) {
		String jsonPath = StringUtils.trimToEmpty(jsonField.jsonPath());
		if (jsonPath == null) {
			jsonPath = "$";
		}
		return jsonPath;
	}

	private void setValueWithoutJsonField(QueryWrapper wrapper, String columnName, Object value,
			SearchModelType searchModelType, @Nullable String customSql) {
		switch (searchModelType) {
		case IS_NULL -> {
			if (value instanceof Boolean bol) {
				if (bol) {
					wrapper.isNull(columnName);
				}
				else {
					wrapper.isNotNull(columnName);
				}
			}
			else {
				wrapper.isNull(columnName);
			}
		}
		case IS_NOT_NULL -> {
			if (value instanceof Boolean bol) {
				if (bol) {
					wrapper.isNotNull(columnName);
				}
				else {
					wrapper.isNull(columnName);
				}
			}
			else {
				wrapper.isNotNull(columnName);
			}
		}
		case GREATER_THAN -> wrapper.gt(columnName, value);
		case GREATER_THAN_OR_EQUAL_TO, START_WITH -> wrapper.ge(columnName, value);
		case LESS_THAN -> wrapper.lt(columnName, value);
		case LESS_THAN_OR_EQUAL_TO, END_WITH -> wrapper.le(columnName, value);
		case NOT_EQUAL_TO, NOT_LIKE, START_NOT_WITH, END_NOT_WITH -> wrapper.ne(columnName, value);
		case CUSTOM_SQL -> {
			if (customSql == null) {
				break;
			}
			CustomerSqlInfo info = new CustomerSqlInfo(customSql);
			if (info.batch) {
				wrapper.apply(info.sql.replaceAll(CustomerSqlInfo.BATCH_REG, "{0}"), value);
			}
			else if (info.matchFlag) {
				wrapper.apply(info.sql.replaceAll(CustomerSqlInfo.MATCH_REG, "{0}"), value);
			}
			else {
				wrapper.apply(info.sql);
			}
		}
		default -> wrapper.eq(columnName, value);
		}
	}
}