hkube.communication.SingleRequest Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of communication Show documentation
There is a newer version: 2.0.00
package hkube.communication;

import hkube.model.HeaderContentPair;
import hkube.model.ObjectAndSize;
import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;
import org.json.JSONObject;

import java.util.*;
import java.util.concurrent.TimeoutException;

public class SingleRequest extends DataRequest {
    public SingleRequest(IRequest requestAdapter, String taskId,   String encoding) {
        super(requestAdapter, null, encoding);
        List tasks = new ArrayList();
        tasks.add(taskId);
        super.tasks = tasks;
    }
    private static final Logger logger = LogManager.getLogger();
    public Object send() throws TimeoutException {
        logger.info("Invoking single from peer");
        HashMap map = new HashMap();
        if (tasks != null) {
            map.put("tasks", tasks);
        }
        List  headerContentPairs = requestAdapter.send(encoder.encodeNoHeader(map));
        HeaderContentPair pair = headerContentPairs.get(0);
        Integer size = pair.getContent().length;
        Object decoded = encoder.decodeSeparately(pair.getHeader(),pair.getContent());
        if (logger.isDebugEnabled()) {
            Object result = toJSON(decoded);
            logger.debug(result);
        }
        if(decoded instanceof Map){
            Map result = (Map)decoded;
                if (result.get("hkube_error")!=null) {
                logger.warn(result.toString());
                throw new RuntimeException(result.toString());
            }
            if (tasks != null ){
                if(result.get("errors")!=null && (Boolean) result.get("errors")==true){
                    if(result.get("items")!=null){
                        Collection items = (Collection)result.get("items");
                        Iterator itemIterator = items.iterator();
                        Map listResult = new HashMap<>();
                        int i=0;
                       while(itemIterator.hasNext()){
                            Object currentItem = itemIterator.next();
                            if(currentItem instanceof JSONObject && !((JSONObject) currentItem).has("hkube_error")){
                                listResult.put(tasks.get(i),currentItem);
                            }
                        }
                        decoded = listResult;
                    }
                }
            }
        }
        ObjectAndSize result = new ObjectAndSize(decoded,size);
        return result;
    }
}