package com.zhangxiaofan.sparkproject.util;
import com.zhangxiaofan.sparkproject.conf.ConfigurationManager;
import com.zhangxiaofan.sparkproject.constant.Constants;
import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;
/**
* 参数工具类
**/
public class ParamUtils {
/**
* 从命令行参数中提取任务id
*
* @param args 命令行参数
* @return 任务id
*/
public static Long getTaskIdFromArgs(String[] args, String taskType) {
Boolean local = ConfigurationManager.getBoolean(Constants.SPARK_LOCAL);
if (local) {
return ConfigurationManager.getLong(taskType);
} else {
try {
if (args != null && args.length > 0) {
return Long.valueOf(args[0]);
}
} catch (NumberFormatException e) {
e.printStackTrace();
}
}
return null;
}
/**
* 从JSON对象中提取参数
*
* @param jsonObject JSON对象
* @return 参数
*/
public static String getParam(JSONObject jsonObject, String field) {
JSONArray jsonArray = jsonObject.getJSONArray(field);
if (jsonArray != null && jsonArray.size() > 0) {
return jsonArray.getString(0);
}
return null;
}
}