|
23 | 23 | import com.datastax.driver.core.schemabuilder.SchemaBuilder; |
24 | 24 | import com.datastax.driver.core.utils.CassandraVersion; |
25 | 25 | import com.google.common.collect.ImmutableMap; |
26 | | -import com.google.common.io.ByteStreams; |
27 | 26 | import com.google.common.io.Closer; |
28 | | -import java.io.ByteArrayOutputStream; |
| 27 | +import java.io.BufferedReader; |
29 | 28 | import java.io.IOException; |
30 | 29 | import java.io.InputStream; |
31 | | -import java.io.PrintStream; |
| 30 | +import java.io.InputStreamReader; |
| 31 | +import java.io.StringWriter; |
32 | 32 | import java.util.Map; |
33 | 33 | import org.slf4j.Logger; |
34 | 34 | import org.slf4j.LoggerFactory; |
@@ -248,10 +248,19 @@ private String getExpectedCqlString() { |
248 | 248 | + ")") |
249 | 249 | .isNotNull(); |
250 | 250 | closer.register(is); |
251 | | - ByteArrayOutputStream baos = new ByteArrayOutputStream(); |
252 | | - PrintStream ps = new PrintStream(baos); |
253 | | - ByteStreams.copy(is, ps); |
254 | | - return baos.toString().trim(); |
| 251 | + |
| 252 | + BufferedReader in = new BufferedReader(new InputStreamReader(is)); |
| 253 | + StringWriter out = new StringWriter(); |
| 254 | + |
| 255 | + String line; |
| 256 | + while ((line = in.readLine()) != null) { |
| 257 | + |
| 258 | + String trimmedLine = line.trim(); |
| 259 | + if (trimmedLine.startsWith("/*") || trimmedLine.startsWith("*")) continue; |
| 260 | + out.write(line); |
| 261 | + out.write(System.getProperty("line.separator")); |
| 262 | + } |
| 263 | + return out.toString().trim(); |
255 | 264 | } catch (IOException e) { |
256 | 265 | logger.warn("Failure to read {}", resourceName, e); |
257 | 266 | fail("Unable to read " + resourceName + " is it defined?"); |
|
0 commit comments