com.wavemaker.runtime.data.util.DataServiceUtils Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of wavemaker-app-runtime-db Show documentation
wavemaker-app-runtime-db
The newest version!
/*******************************************************************************
 * Copyright (C) 2022-2023 WaveMaker, Inc.
 *
 * 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 com.wavemaker.runtime.data.util;

import java.util.regex.Matcher;
import java.util.regex.Pattern;

import org.springframework.dao.InvalidDataAccessResourceUsageException;

import com.wavemaker.commons.util.SystemUtils;
import com.wavemaker.runtime.data.exception.DataServiceRuntimeException;

/**
 * @author Simon Toens
 */
public class DataServiceUtils {

    private static final String WM_MYSQL_CLOUD_TOKEN_PATTERN_STRING = "\\{WM_CLOUD_MYSQL_HOST\\}(:[\\d]+)?";
    private static final Pattern WM_MYSQL_CLOUD_TOKEN_PATTERN = Pattern.compile(WM_MYSQL_CLOUD_TOKEN_PATTERN_STRING);

    public static RuntimeException unwrap(Throwable th) {
        if (th instanceof DataServiceRuntimeException) {
            return (DataServiceRuntimeException) th;
        } else {
            th = SystemUtils.getRootException(th);

            if (InvalidDataAccessResourceUsageException.class.isAssignableFrom(th.getClass())) {
                InvalidDataAccessResourceUsageException e = (InvalidDataAccessResourceUsageException) th;
                if (e.getRootCause() != null) {
                    th = e.getRootCause();
                }
            }
            if (th instanceof RuntimeException) {
                return (RuntimeException) th;
            } else {
                return new DataServiceRuntimeException(th);
            }
        }
    }

    public static boolean isDML(String query) {

        String q = query.trim().toLowerCase();

        return !q.startsWith("select");
    }

    public static String replaceMySqlCloudToken(String url, String hostPortString) {
        final Matcher matcher = WM_MYSQL_CLOUD_TOKEN_PATTERN.matcher(url);
        return matcher.replaceFirst(hostPortString);
    }

    private DataServiceUtils() {
    }

}