add --log-level=debug/info/warn/error flag, so we can close the log print during db gen
This commit is contained in:
parent
ed57fa5c52
commit
bff6b5661c
|
|
@ -21,10 +21,11 @@ public class MakerTest {
|
|||
System.out.println("options:");
|
||||
System.out.println(" --src string source ip text file path");
|
||||
System.out.println(" --dst string destination binary xdb file path");
|
||||
System.out.println(" --log-level string set the log level, options: debug/info/warn/error");
|
||||
}
|
||||
|
||||
public static void genDb(String[] args) throws Exception {
|
||||
String srcFile = "", dstFile = "";
|
||||
String srcFile = "", dstFile = "", logLevel = "info";
|
||||
int indexPolicy = IndexPolicy.Vector;
|
||||
for (final String r : args) {
|
||||
if (r.length() < 5) {
|
||||
|
|
@ -48,25 +49,23 @@ public class MakerTest {
|
|||
srcFile = val;
|
||||
} else if ("dst".equals(key)) {
|
||||
dstFile = val;
|
||||
} else if ("index".equals(key)) {
|
||||
try {
|
||||
indexPolicy = IndexPolicy.parse(val);
|
||||
} catch (Exception e) {
|
||||
System.out.println("parse policy " + e);
|
||||
}
|
||||
} else if ("log-level".equals(key)) {
|
||||
logLevel = val;
|
||||
} else {
|
||||
System.out.printf("undefined option `%s`\n", r);
|
||||
return;
|
||||
}
|
||||
}
|
||||
|
||||
if (srcFile.length() < 1 || dstFile.length() < 1) {
|
||||
if (srcFile.isEmpty() || dstFile.isEmpty()) {
|
||||
printHelp(args);
|
||||
return;
|
||||
}
|
||||
|
||||
long tStart = System.currentTimeMillis();
|
||||
Maker maker = new Maker(indexPolicy, srcFile, dstFile);
|
||||
final Maker maker = new Maker(indexPolicy, srcFile, dstFile);
|
||||
log.infof("Generating xdb with src=%s, dst=%s, logLevel=%s\n", srcFile, dstFile, logLevel);
|
||||
Maker.log.setLevel(logLevel);
|
||||
maker.init();
|
||||
maker.start();
|
||||
maker.end();
|
||||
|
|
|
|||
|
|
@ -28,7 +28,7 @@ public class Log {
|
|||
};
|
||||
|
||||
public final Class<?> baseClass;
|
||||
private static int level;
|
||||
private int level = INFO;
|
||||
|
||||
public Log(Class<?> baseClass) {
|
||||
this.baseClass = baseClass;
|
||||
|
|
@ -56,7 +56,7 @@ public class Log {
|
|||
}
|
||||
|
||||
// level filter
|
||||
if (level < Log.level) {
|
||||
if (level < this.level) {
|
||||
return;
|
||||
}
|
||||
|
||||
|
|
@ -96,21 +96,24 @@ public class Log {
|
|||
printf(ERROR, format, args);
|
||||
}
|
||||
|
||||
public static void setLevel(int level) {
|
||||
Log.level = level;
|
||||
public Log setLevel(int level) {
|
||||
this.level = level;
|
||||
return this;
|
||||
}
|
||||
|
||||
public static void setLevel(String level) {
|
||||
public Log setLevel(String level) {
|
||||
String v = level.toLowerCase();
|
||||
if ("debug".equals(v)) {
|
||||
Log.level = DEBUG;
|
||||
this.level = DEBUG;
|
||||
} else if ("info".equals(v)) {
|
||||
Log.level = INFO;
|
||||
this.level = INFO;
|
||||
} else if ("warn".equals(v)) {
|
||||
Log.level = WARN;
|
||||
this.level = WARN;
|
||||
} else if ("error".equals(v)) {
|
||||
Log.level = ERROR;
|
||||
this.level = ERROR;
|
||||
}
|
||||
|
||||
return this;
|
||||
}
|
||||
|
||||
}
|
||||
|
|
@ -68,7 +68,7 @@ public class Maker {
|
|||
public static final int SegmentIndexSize = 14;
|
||||
public static final int VectorIndexLength = VectorIndexRows * VectorIndexCols * VectorIndexSize;
|
||||
|
||||
private static final Log log = Log.getLogger(Maker.class);
|
||||
public static final Log log = Log.getLogger(Maker.class);
|
||||
|
||||
// source text file handle
|
||||
private final File srcFile;
|
||||
|
|
@ -199,7 +199,7 @@ public class Maker {
|
|||
|
||||
// start to make the binary file
|
||||
public void start() throws Exception {
|
||||
if (segments.size() == 0) {
|
||||
if (segments.isEmpty()) {
|
||||
throw new Exception("empty segment list");
|
||||
}
|
||||
|
||||
|
|
|
|||
Loading…
Reference in New Issue