添加日志级别

Author: dapeng

core/pom.xml

@@ -115,7 +115,11 @@
             <artifactId>flink-statebackend-rocksdb_2.11</artifactId>
             <version>${flink.version}</version>
         </dependency>
-
+        <dependency>
+            <groupId>ch.qos.logback</groupId>
+            <artifactId>logback-classic</artifactId>
+            <version>1.1.7</version>
+        </dependency>
 
     </dependencies>
 

core/src/main/java/com/dtstack/flink/sql/Main.java

@@ -77,6 +77,8 @@
 import java.util.Set;
 
 import com.dtstack.flink.sql.option.Options;
+import ch.qos.logback.classic.Level;
+import ch.qos.logback.classic.LoggerContext;
 
 /**
  * Date: 2018/6/26
@@ -105,9 +107,9 @@ public static void main(String[] args) throws Exception {
         String pluginLoadMode = options.getPluginLoadMode();
         String deployMode = options.getMode();
         String confProp = options.getConfProp();
-
         sql = URLDecoder.decode(sql, Charsets.UTF_8.name());
         SqlParser.setLocalSqlPluginRoot(localSqlPluginPath);
+        setLogLevel(options.getLogLevel());
 
         List<String> addJarFileList = Lists.newArrayList();
         if (!Strings.isNullOrEmpty(addJarListStr)) {
@@ -304,4 +306,10 @@ private static StreamExecutionEnvironment getStreamExeEnv(Properties confPropert
         StreamEnvConfigManager.streamExecutionEnvironmentConfig(env, confProperties);
         return env;
     }
+    private static void setLogLevel(String level){
+        LoggerContext loggerContext= (LoggerContext) LoggerFactory.getILoggerFactory();
+        //设置全局日志级别
+        ch.qos.logback.classic.Logger logger = loggerContext.getLogger("root");
+        logger.setLevel(Level.toLevel(level, Level.INFO));
+    }
 }

core/src/main/java/com/dtstack/flink/sql/option/Options.java

@@ -69,6 +69,8 @@ public class Options {
     @OptionRequired(description = "plugin load mode, by classpath or shipfile")
     private String pluginLoadMode = EPluginLoadMode.CLASSPATH.name();
 
+    private String logLevel = "logLevel";
+
     public String getMode() {
         return mode;
     }
@@ -172,4 +174,12 @@ public String getPluginLoadMode() {
     public void setPluginLoadMode(String pluginLoadMode) {
         this.pluginLoadMode = pluginLoadMode;
     }
+
+    public String getLogLevel() {
+        return logLevel;
+    }
+
+    public void setLogLevel(String logLevel) {
+        this.logLevel = logLevel;
+    }
 }