Apache Zeppelin基础工具类

@Component
public class ZeppelinUtil {

    public String createNote(Script record, String zeppelinUrl) throws Exception {
        Map<String, Object> params = new HashMap<>();
        params.put("name", record.getScriptName());
        String url = zeppelinUrl + Constant.CREATE_NOTE;
        Map<String, Object> headers = new HashMap<>();
        JSONObject jsonObject = HttpUtil.httpPostUtil(params, headers, url);
        checkResponse(jsonObject);
        return jsonObject.getString("body");
    }

    public String createParagraph(Script record, String zeppelinUrl) throws Exception {
        Map<String, Object> params = new HashMap<>();
        params.put("title", record.getScriptName());
        params.put("text", record.getScriptText());
        String url = zeppelinUrl + (Constant.CREATE_PARAGRAPH).replace("${NOTE}", record.getNoteId());
        Map<String, Object> headers = new HashMap<>();
        JSONObject jsonObject = HttpUtil.httpPostUtil(params, headers, url);
        checkResponse(jsonObject);
        return jsonObject.getString("body");
    }

    public void updateParagraph(Script record, String zeppelinUrl) throws Exception {
        Map<String, Object> params = new HashMap<>();
        params.put("title", record.getScriptName());
        params.put("text", record.getScriptText());
        String url = zeppelinUrl + (Constant.UPDATER_PARAGRAPH).replace("${NOTE}", record.getNoteId()) + record.getParagraphId();
        Map<String, Object> headers = new HashMap<>();
        JSONObject jsonObject = HttpUtil.httpPutUtil(params, headers, url);
        checkResponse(jsonObject);
    }

    public JSONObject execute(Script record, String zeppelinUrl) throws Exception {
        Map<String, Object> params = new HashMap<>();
        String url = zeppelinUrl + Constant.RUN_PARAGRAPH + record.getNoteId() + "/" + record.getParagraphId();
        Map<String, Object> headers = new HashMap<>();
        JSONObject jsonObject = HttpUtil.httpPostUtil(params, headers, url);
        checkResponse(jsonObject);
        return jsonObject.getJSONObject("body");
    }

    public void deleteNode(String noteId, String zeppelinUrl) throws Exception {
        String url = zeppelinUrl + Constant.DELETE_NOTE + noteId;
        Map<String, Object> headers = new HashMap<>();
        JSONObject jsonObject = HttpUtil.httpDeleteUtil(headers, url);
        checkResponse(jsonObject);
    }

    /**
     * Throw exception if the status code is not 200.
     *
     * @param jsonObject
     * @throws Exception
     */
    private void checkResponse(JSONObject jsonObject) throws Exception {
        if (null == jsonObject) {
            throw new Exception(Constant.ERROR_MESSAGE);
        }
        if (!"OK".equals(jsonObject.getString("status"))) {
            String message = jsonObject.getString("message");
            throw new Exception(String.format("执行异常, message: %s", message));
        }
    }
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值