Examples with AnomalyFeedbackBean com.linkedin.thirdeye.datalayer.pojo.AnomalyFeedbackBean used on opensource projects

Example 1 with AnomalyFeedbackBean

use of com.linkedin.thirdeye.datalayer.pojo.AnomalyFeedbackBean in project pinot by linkedin.

the class RawAnomalyResultManagerImpl method update.

public int update(RawAnomalyResultDTO entity) {
    RawAnomalyResultBean bean = (RawAnomalyResultBean) convertDTO2Bean(entity, RawAnomalyResultBean.class);
    if (entity.getFeedback() != null) {
        if (entity.getFeedback().getId() == null) {
            AnomalyFeedbackBean feedbackBean = (AnomalyFeedbackBean) convertDTO2Bean(entity.getFeedback(), AnomalyFeedbackBean.class);
            Long feedbackId = genericPojoDao.put(feedbackBean);
            entity.getFeedback().setId(feedbackId);
        }
        bean.setAnomalyFeedbackId(entity.getFeedback().getId());
    }
    if (entity.getFunction() != null) {
        bean.setFunctionId(entity.getFunction().getId());
    }
    return genericPojoDao.update(bean);
}

Example 2 with AnomalyFeedbackBean

use of com.linkedin.thirdeye.datalayer.pojo.AnomalyFeedbackBean in project pinot by linkedin.

the class RawAnomalyResultManagerImpl method save.

public Long save(RawAnomalyResultDTO entity) {
    if (entity.getId() != null) {
        //TODO: throw exception and force the caller to call update instead
        update(entity);
        return entity.getId();
    }
    RawAnomalyResultBean bean = (RawAnomalyResultBean) convertDTO2Bean(entity, RawAnomalyResultBean.class);
    if (entity.getFeedback() != null) {
        if (entity.getFeedback().getId() == null) {
            AnomalyFeedbackBean feedbackBean = (AnomalyFeedbackBean) convertDTO2Bean(entity.getFeedback(), AnomalyFeedbackBean.class);
            Long feedbackId = genericPojoDao.put(feedbackBean);
            entity.getFeedback().setId(feedbackId);
        }
        bean.setAnomalyFeedbackId(entity.getFeedback().getId());
    }
    if (entity.getFunction() != null) {
        bean.setFunctionId(entity.getFunction().getId());
    }
    Long id = genericPojoDao.put(bean);
    entity.setId(id);
    return id;
}

Example 3 with AnomalyFeedbackBean

use of com.linkedin.thirdeye.datalayer.pojo.AnomalyFeedbackBean in project pinot by linkedin.

the class MergedAnomalyResultManagerImpl method updateAnomalyFeedback.

public void updateAnomalyFeedback(MergedAnomalyResultDTO entity) {
    MergedAnomalyResultBean bean = convertDTO2Bean(entity, MergedAnomalyResultBean.class);
    if (entity.getFeedback() != null) {
        if (entity.getFeedback().getId() == null) {
            AnomalyFeedbackBean feedbackBean = (AnomalyFeedbackBean) convertDTO2Bean(entity.getFeedback(), AnomalyFeedbackBean.class);
            Long feedbackId = genericPojoDao.put(feedbackBean);
            entity.getFeedback().setId(feedbackId);
        } else {
            AnomalyFeedbackBean feedbackBean = genericPojoDao.get(entity.getFeedback().getId(), AnomalyFeedbackBean.class);
            feedbackBean.setStatus(entity.getFeedback().getStatus());
            feedbackBean.setFeedbackType(entity.getFeedback().getFeedbackType());
            feedbackBean.setComment(entity.getFeedback().getComment());
            genericPojoDao.update(feedbackBean);
        }
        bean.setAnomalyFeedbackId(entity.getFeedback().getId());
    }
    genericPojoDao.update(bean);
}

Example 4 with AnomalyFeedbackBean

use of com.linkedin.thirdeye.datalayer.pojo.AnomalyFeedbackBean in project pinot by linkedin.

the class AbstractManagerImpl method createRawAnomalyDTOFromBean.

protected RawAnomalyResultDTO createRawAnomalyDTOFromBean(RawAnomalyResultBean rawAnomalyResultBean) {
    RawAnomalyResultDTO rawAnomalyResultDTO;
    rawAnomalyResultDTO = MODEL_MAPPER.map(rawAnomalyResultBean, RawAnomalyResultDTO.class);
    if (rawAnomalyResultBean.getFunctionId() != null) {
        AnomalyFunctionBean anomalyFunctionBean = genericPojoDao.get(rawAnomalyResultBean.getFunctionId(), AnomalyFunctionBean.class);
        if (anomalyFunctionBean == null) {
            LOG.error("this anomaly function bean should not be null");
        }
        AnomalyFunctionDTO anomalyFunctionDTO = MODEL_MAPPER.map(anomalyFunctionBean, AnomalyFunctionDTO.class);
        rawAnomalyResultDTO.setFunction(anomalyFunctionDTO);
    }
    if (rawAnomalyResultBean.getAnomalyFeedbackId() != null) {
        AnomalyFeedbackBean anomalyFeedbackBean = genericPojoDao.get(rawAnomalyResultBean.getAnomalyFeedbackId(), AnomalyFeedbackBean.class);
        AnomalyFeedbackDTO anomalyFeedbackDTO = MODEL_MAPPER.map(anomalyFeedbackBean, AnomalyFeedbackDTO.class);
        rawAnomalyResultDTO.setFeedback(anomalyFeedbackDTO);
    }
    return rawAnomalyResultDTO;
}

Example 5 with AnomalyFeedbackBean

use of com.linkedin.thirdeye.datalayer.pojo.AnomalyFeedbackBean in project pinot by linkedin.

the class MergedAnomalyResultManagerImpl method convertMergedAnomalyBean2DTO.

protected MergedAnomalyResultDTO convertMergedAnomalyBean2DTO(MergedAnomalyResultBean mergedAnomalyResultBean, boolean loadRawAnomalies) {
    MergedAnomalyResultDTO mergedAnomalyResultDTO;
    mergedAnomalyResultDTO = MODEL_MAPPER.map(mergedAnomalyResultBean, MergedAnomalyResultDTO.class);
    if (mergedAnomalyResultBean.getFunctionId() != null) {
        AnomalyFunctionBean anomalyFunctionBean = genericPojoDao.get(mergedAnomalyResultBean.getFunctionId(), AnomalyFunctionBean.class);
        AnomalyFunctionDTO anomalyFunctionDTO = MODEL_MAPPER.map(anomalyFunctionBean, AnomalyFunctionDTO.class);
        mergedAnomalyResultDTO.setFunction(anomalyFunctionDTO);
    }
    if (mergedAnomalyResultBean.getAnomalyFeedbackId() != null) {
        AnomalyFeedbackBean anomalyFeedbackBean = genericPojoDao.get(mergedAnomalyResultBean.getAnomalyFeedbackId(), AnomalyFeedbackBean.class);
        AnomalyFeedbackDTO anomalyFeedbackDTO = MODEL_MAPPER.map(anomalyFeedbackBean, AnomalyFeedbackDTO.class);
        mergedAnomalyResultDTO.setFeedback(anomalyFeedbackDTO);
    }
    if (loadRawAnomalies && mergedAnomalyResultBean.getRawAnomalyIdList() != null && !mergedAnomalyResultBean.getRawAnomalyIdList().isEmpty()) {
        List<RawAnomalyResultDTO> anomalyResults = new ArrayList<>();
        List<RawAnomalyResultBean> list = genericPojoDao.get(mergedAnomalyResultBean.getRawAnomalyIdList(), RawAnomalyResultBean.class);
        for (RawAnomalyResultBean rawAnomalyResultBean : list) {
            anomalyResults.add(createRawAnomalyDTOFromBean(rawAnomalyResultBean));
        }
        mergedAnomalyResultDTO.setAnomalyResults(anomalyResults);
    }
    return mergedAnomalyResultDTO;
}