com.alipay.sofa.tracer.plugins.datasource.utils.SqlUtils Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of sofa-tracer-datasource-plugin Show documentation
The newest version!
/*
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You 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 com.alipay.sofa.tracer.plugins.datasource.utils;

import com.alipay.common.tracer.core.utils.StringUtils;

/**
 * @author shusong.yss
 * @author qilong.zql
 * @since 2.2.0
 */
public class SqlUtils {
    private static final String DEFAULT_SEPARATOR        = ",";
    private static final String DEFAULT_SEPARATOR_ESCAPE = "%2C";
    private static final String DEFAULT_NEW_LINE         = "\n";
    private static final String DEFAULT_RETURN           = "\r";
    private static final String EMPTY_STRING             = "";
    private static final int    DIGEST_LOG_SQL_LIMIT     = 4096;

    public static String getSqlEscaped(String sql) {
        String limitSql = sql;
        if (limitSql != null && limitSql.length() > DIGEST_LOG_SQL_LIMIT) {
            limitSql = limitSql.substring(0, DIGEST_LOG_SQL_LIMIT) + " ...";
        }
        return escape(escapeWithSpecialCheck(limitSql), DEFAULT_SEPARATOR, DEFAULT_SEPARATOR_ESCAPE);
    }

    private static String escape(String str, String oldStr, String newStr) {
        if (str == null) {
            return EMPTY_STRING;
        }
        return StringUtils.replace(str, oldStr, newStr);
    }

    private static String escapeWithSpecialCheck(String str) {
        if (str == null) {
            return "";
        }
        StringBuilder appender = new StringBuilder();
        int len = str.length();
        appender.ensureCapacity(appender.length() + len);
        for (int i = 0; i < len; i++) {
            char c = str.charAt(i);
            if (c == '\n' || c == '\r' || c == '|') {
                c = ' ';
            }
            if (c != ' ' || c == ' ' && appender.length() > 0
                && appender.charAt(appender.length() - 1) != ' ') {
                appender.append(c);
            }
        }
        return appender.toString().trim();
    }
}