502647092/TabooLib
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

.

Browse Source
This commit is contained in:
Izzel_Aliz
2018-05-09 13:04:58 +08:00
parent 3f4940c6b7
commit 2b14457309
72 changed files with 1249 additions and 1318 deletions
Download Patch File Download Diff File Expand all files Collapse all files

240
src/main/java/me/skymc/taboolib/csvutils/CsvReader.java
View File

@@ -242,17 +242,16 @@ public class CsvReader {
return array;
}
public void setHeaders(final String[] headers) {
this.headersHolder.Headers = headers;
this.headersHolder.IndexByName.clear();
if (headers != null) {
this.headersHolder.Length = headers.length;
private static char hexToDec(final char c) {
char c2;
if (c >= 'a') {
c2 = (char) (c - 'a' + '\n');
} else if (c >= 'A') {
c2 = (char) (c - 'A' + '\n');
} else {
this.headersHolder.Length = 0;
}
for (int i = 0; i < this.headersHolder.Length; ++i) {
this.headersHolder.IndexByName.put(headers[i], i);
c2 = (char) (c - '0');
}
return c2;
}
public String[] getValues() throws IOException {
@@ -282,6 +281,19 @@ public class CsvReader {
return new CsvReader(new StringReader(s));
}
public void setHeaders(final String[] headers) {
this.headersHolder.Headers = headers;
this.headersHolder.IndexByName.clear();
if (headers != null) {
this.headersHolder.Length = headers.length;
} else {
this.headersHolder.Length = 0;
}
for (int i = 0; i < this.headersHolder.Length; ++i) {
this.headersHolder.IndexByName.put(headers[i], i);
}
}
public boolean readRecord() throws IOException {
this.checkClosed();
this.columnsCount = 0;
@@ -716,6 +728,35 @@ public class CsvReader {
return this.hasReadNextLine;
}
public boolean readHeaders() throws IOException {
final boolean record = this.readRecord();
this.headersHolder.Length = this.columnsCount;
this.headersHolder.Headers = new String[this.columnsCount];
for (int i = 0; i < this.headersHolder.Length; ++i) {
final String value = this.get(i);
this.headersHolder.Headers[i] = value;
this.headersHolder.IndexByName.put(value, i);
}
if (record) {
--this.currentRecord;
}
this.columnsCount = 0;
return record;
}
public String getHeader(final int n) throws IOException {
this.checkClosed();
if (n > -1 && n < this.headersHolder.Length) {
return this.headersHolder.Headers[n];
}
return "";
}
public boolean isQualified(final int n) throws IOException {
this.checkClosed();
return n < this.columnsCount && n > -1 && this.isQualified[n];
}
private void checkDataLength() throws IOException {
if (!this.initialized) {
if (this.fileName != null) {
@@ -749,79 +790,6 @@ public class CsvReader {
this.dataBuffer.ColumnStart = 0;
}
public boolean readHeaders() throws IOException {
final boolean record = this.readRecord();
this.headersHolder.Length = this.columnsCount;
this.headersHolder.Headers = new String[this.columnsCount];
for (int i = 0; i < this.headersHolder.Length; ++i) {
final String value = this.get(i);
this.headersHolder.Headers[i] = value;
this.headersHolder.IndexByName.put(value, i);
}
if (record) {
--this.currentRecord;
}
this.columnsCount = 0;
return record;
}
public String getHeader(final int n) throws IOException {
this.checkClosed();
if (n > -1 && n < this.headersHolder.Length) {
return this.headersHolder.Headers[n];
}
return "";
}
public boolean isQualified(final int n) throws IOException {
this.checkClosed();
return n < this.columnsCount && n > -1 && this.isQualified[n];
}
public void endColumn() throws IOException {
String s = "";
if (this.startedColumn) {
if (this.columnBuffer.Position == 0) {
if (this.dataBuffer.ColumnStart < this.dataBuffer.Position) {
int n = this.dataBuffer.Position - 1;
if (this.userSettings.TrimWhitespace && !this.startedWithQualifier) {
while (n >= this.dataBuffer.ColumnStart && (this.dataBuffer.Buffer[n] == ' ' || this.dataBuffer.Buffer[n] == '\t')) {
--n;
}
}
s = new String(this.dataBuffer.Buffer, this.dataBuffer.ColumnStart, n - this.dataBuffer.ColumnStart + 1);
}
} else {
this.updateCurrentValue();
int n2 = this.columnBuffer.Position - 1;
if (this.userSettings.TrimWhitespace && !this.startedWithQualifier) {
while (n2 >= 0 && (this.columnBuffer.Buffer[n2] == ' ' || this.columnBuffer.Buffer[n2] == ' ')) {
--n2;
}
}
s = new String(this.columnBuffer.Buffer, 0, n2 + 1);
}
}
this.columnBuffer.Position = 0;
this.startedColumn = false;
if (this.columnsCount >= 100000 && this.userSettings.SafetySwitch) {
this.close();
throw new IOException("Maximum column count of 100,000 exceeded in record " + NumberFormat.getIntegerInstance().format(this.currentRecord) + ". Set the SafetySwitch property to false" + " if you're expecting more than 100,000 columns per record to" + " avoid this error.");
}
if (this.columnsCount == this.values.length) {
final int n3 = this.values.length * 2;
final String[] values = new String[n3];
System.arraycopy(this.values, 0, values, 0, this.values.length);
this.values = values;
final boolean[] isQualified = new boolean[n3];
System.arraycopy(this.isQualified, 0, isQualified, 0, this.isQualified.length);
this.isQualified = isQualified;
}
this.values[this.columnsCount] = s;
this.isQualified[this.columnsCount] = this.startedWithQualifier;
++this.columnsCount;
}
private void appendLetter(final char c) {
if (this.columnBuffer.Position == this.columnBuffer.Buffer.length) {
final char[] buffer = new char[this.columnBuffer.Buffer.length * 2];
@@ -872,35 +840,48 @@ public class CsvReader {
return record;
}
public boolean skipLine() throws IOException {
this.checkClosed();
this.columnsCount = 0;
boolean b = false;
if (this.hasMoreData) {
boolean b2 = false;
do {
if (this.dataBuffer.Position == this.dataBuffer.Count) {
this.checkDataLength();
} else {
b = true;
final char lastLetter = this.dataBuffer.Buffer[this.dataBuffer.Position];
if (lastLetter == '\r' || lastLetter == '\n') {
b2 = true;
public void endColumn() throws IOException {
String s = "";
if (this.startedColumn) {
if (this.columnBuffer.Position == 0) {
if (this.dataBuffer.ColumnStart < this.dataBuffer.Position) {
int n = this.dataBuffer.Position - 1;
if (this.userSettings.TrimWhitespace && !this.startedWithQualifier) {
while (n >= this.dataBuffer.ColumnStart && (this.dataBuffer.Buffer[n] == ' ' || this.dataBuffer.Buffer[n] == '\t')) {
--n;
}
}
this.lastLetter = lastLetter;
if (b2) {
continue;
}
final DataBuffer dataBuffer = this.dataBuffer;
++dataBuffer.Position;
s = new String(this.dataBuffer.Buffer, this.dataBuffer.ColumnStart, n - this.dataBuffer.ColumnStart + 1);
}
} while (this.hasMoreData && !b2);
this.columnBuffer.Position = 0;
this.dataBuffer.LineStart = this.dataBuffer.Position + 1;
} else {
this.updateCurrentValue();
int n2 = this.columnBuffer.Position - 1;
if (this.userSettings.TrimWhitespace && !this.startedWithQualifier) {
while (n2 >= 0 && (this.columnBuffer.Buffer[n2] == ' ' || this.columnBuffer.Buffer[n2] == ' ')) {
--n2;
}
}
s = new String(this.columnBuffer.Buffer, 0, n2 + 1);
}
}
this.rawBuffer.Position = 0;
this.rawRecord = "";
return b;
this.columnBuffer.Position = 0;
this.startedColumn = false;
if (this.columnsCount >= 100000 && this.userSettings.SafetySwitch) {
this.close();
throw new IOException("Maximum column count of 100,000 exceeded in record " + NumberFormat.getIntegerInstance().format(this.currentRecord) + ". Set the SafetySwitch property to false" + " if you're expecting more than 100,000 columns per record to" + " avoid this error.");
}
if (this.columnsCount == this.values.length) {
final int n3 = this.values.length * 2;
final String[] values = new String[n3];
System.arraycopy(this.values, 0, values, 0, this.values.length);
this.values = values;
final boolean[] isQualified = new boolean[n3];
System.arraycopy(this.isQualified, 0, isQualified, 0, this.isQualified.length);
this.isQualified = isQualified;
}
this.values[this.columnsCount] = s;
this.isQualified[this.columnsCount] = this.startedWithQualifier;
++this.columnsCount;
}
public void close() {
@@ -937,23 +918,42 @@ public class CsvReader {
}
}
public boolean skipLine() throws IOException {
this.checkClosed();
this.columnsCount = 0;
boolean b = false;
if (this.hasMoreData) {
boolean b2 = false;
do {
if (this.dataBuffer.Position == this.dataBuffer.Count) {
this.checkDataLength();
} else {
b = true;
final char lastLetter = this.dataBuffer.Buffer[this.dataBuffer.Position];
if (lastLetter == '\r' || lastLetter == '\n') {
b2 = true;
}
this.lastLetter = lastLetter;
if (b2) {
continue;
}
final DataBuffer dataBuffer = this.dataBuffer;
++dataBuffer.Position;
}
} while (this.hasMoreData && !b2);
this.columnBuffer.Position = 0;
this.dataBuffer.LineStart = this.dataBuffer.Position + 1;
}
this.rawBuffer.Position = 0;
this.rawRecord = "";
return b;
}
@Override
protected void finalize() {
this.close(false);
}
private static char hexToDec(final char c) {
char c2;
if (c >= 'a') {
c2 = (char) (c - 'a' + '\n');
} else if (c >= 'A') {
c2 = (char) (c - 'A' + '\n');
} else {
c2 = (char) (c - '0');
}
return c2;
}
private class StaticSettings {
public static final int MAX_BUFFER_SIZE = 1024;
public static final int MAX_FILE_BUFFER_SIZE = 4096;

42
src/main/java/me/skymc/taboolib/csvutils/CsvWriter.java
View File

@@ -121,6 +121,27 @@ public class CsvWriter {
this.userSettings.ForceQualifier = forceQualifier;
}
public static String replace(final String s, final String s2, final String s3) {
final int length = s2.length();
int i = s.indexOf(s2);
if (i > -1) {
final StringBuilder sb = new StringBuilder();
int n;
for (n = 0; i != -1; i = s.indexOf(s2, n)) {
sb.append(s, n, i);
sb.append(s3);
n = i + length;
}
sb.append(s.substring(n));
return sb.toString();
}
return s;
}
public void write(final String s) throws IOException {
this.write(s, false);
}
public void write(String s, final boolean b) throws IOException {
this.checkClosed();
this.checkInit();
@@ -181,10 +202,6 @@ public class CsvWriter {
this.firstColumn = false;
}
public void write(final String s) throws IOException {
this.write(s, false);
}
public void writeComment(final String s) throws IOException {
this.checkClosed();
this.checkInit();
@@ -198,23 +215,6 @@ public class CsvWriter {
this.firstColumn = true;
}
public static String replace(final String s, final String s2, final String s3) {
final int length = s2.length();
int i = s.indexOf(s2);
if (i > -1) {
final StringBuilder sb = new StringBuilder();
int n;
for (n = 0; i != -1; i = s.indexOf(s2, n)) {
sb.append(s.substring(n, i));
sb.append(s3);
n = i + length;
}
sb.append(s.substring(n));
return sb.toString();
}
return s;
}
public void writeRecord(final String[] array) throws IOException {
this.writeRecord(array, false);
}