org.yx.db.kit.DBKits Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of sumk Show documentation
A quick developing framewort for internet company
There is a newer version: 4.0.2
/**
 * 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.kit;

import java.sql.PreparedStatement;
import java.util.List;
import java.util.Objects;
import java.util.function.Function;

import org.yx.db.sql.PojoMeta;
import org.yx.db.sql.PojoMetaHolder;
import org.yx.exception.SumkException;
import org.yx.exception.SumkExceptionCode;
import org.yx.redis.RecordRepository;
import org.yx.util.StringUtil;

public final class DBKits {

	private static Function plainSqlParse = statement -> {
		String sql = String.valueOf(statement);
		int index = sql.indexOf(": ");
		if (index < 10 || index + 10 > sql.length()) {
			return sql;
		}
		return sql.substring(index + 2);
	};

	public static void setPlainSqlParse(Function plainSqlParse) {
		DBKits.plainSqlParse = Objects.requireNonNull(plainSqlParse);
	}

	public static String getSqlOfStatement(PreparedStatement statement) {
		return plainSqlParse.apply(statement);
	}

	public static  T queryOne(List list) {
		if (list == null || list.isEmpty()) {
			return null;
		}
		if (list.size() > 1) {
			throw new SumkException(SumkExceptionCode.DB_TOO_MANY_RESULTS,
					"result size:" + list.size() + " more than one");
		}
		return list.get(0);
	}

	@SafeVarargs
	public static  int clearCache(T... pojos) throws Exception {
		int total = 0;
		if (pojos == null || pojos.length == 0) {
			return total;
		}
		T t = null;
		for (int i = 0; i < pojos.length; i++) {
			t = pojos[i];
			if (t != null) {
				break;
			}
		}
		if (t == null) {
			return total;
		}
		PojoMeta pm = PojoMetaHolder.getPojoMeta(t.getClass(), null);
		if (pm == null || pm.isNoCache()) {
			return total;
		}
		for (T src : pojos) {
			if (src == null) {
				continue;
			}
			String id = pm.getCacheID(src, false);
			if (StringUtil.isEmpty(id)) {
				continue;
			}
			RecordRepository.del(pm, id);
			total++;
		}
		return total;
	}
}