org.soulwing.jdbc.source.SimpleSQLFormatter Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of fluent-jdbc Show documentation

A lightweight JDBC facade that provides a fluent API based on the Builder and Command design patterns.

There is a newer version: 2.0.3

/*
 * File created on Aug 19, 2015
 *
 * Copyright (c) 2015 Carl Harris, Jr
 * and others as noted
 *
 * 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.soulwing.jdbc.source;

import java.io.IOException;
import java.io.StringReader;

import org.soulwing.jdbc.SQLRuntimeException;

/**
 * A simple {@link SQLFormatter} that removes comments and cleans up whitespace
 * to make an SQL statement occupy a single line in a log file.
 *
 * @author Carl Harris
 */
public class SimpleSQLFormatter implements SQLFormatter {

  @Override
  public String format(String sql) {
    try {
      Lexer lexer = new Lexer(new StringReader(sql));
      Lexer.Token token = lexer.next();
      StringBuilder statement = new StringBuilder();
      while (token != null) {
        if (token.type == Lexer.Token.Type.WHITESPACE) {
          statement.append(' ');
        }
        else if (token.type != Lexer.Token.Type.COMMENT) {
          statement.append(token.value);
        }
        token = lexer.next();
      }
      return statement.toString();
    }
    catch (IOException ex) {
      throw new SQLRuntimeException(ex.getMessage(), ex);
    }
  }

}