org.solmix.datax.wmix.serializer.DSResponseSerializer.java Source code

Introduction

Here is the source code for org.solmix.datax.wmix.serializer.DSResponseSerializer.java

Source

/*
 * Copyright 2015 The Solmix Project
 *
 * This is free software; you can redistribute it and/or modify it
 * under the terms of the GNU Lesser General Public License as
 * published by the Free Software Foundation; either version 2.1 of
 * the License, or (at your option) any later version.
 *
 * This software is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
 * Lesser General Public License for more details.
 *
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * http://www.gnu.org/licenses/ 
 * or see the FSF site: http://www.fsf.org. 
 */

package org.solmix.datax.wmix.serializer;

import java.io.IOException;
import java.util.List;

import org.solmix.datax.DATAX;
import org.solmix.datax.DSResponse;
import org.solmix.datax.DSResponse.Status;
import org.solmix.datax.attachment.Pageable;
import org.solmix.datax.wmix.Constants;

import com.fasterxml.jackson.core.JsonGenerator;
import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.databind.JsonSerializer;
import com.fasterxml.jackson.databind.SerializerProvider;

/**
 * 
 * @author solmix.f@gmail.com
 * @version $Id$ 2015819
 */

public class DSResponseSerializer extends JsonSerializer<ResultObject> {
    public static final String RESPONSE = "response", RESPONSES = "responses", STATUS = "status",
            ISDSRESPONSE = "isDSResponse", AUTH = "AUTH", AFFECTED_ROWS = "affectedRows",
            INVALIDATE_CACHE = "invalidateCache", START_ROW = "startRow", END_ROW = "endRow",
            TOTAL_ROWS = "totalRows", DATA = "data", ERRORS = "errors";

    @Override
    public void serialize(ResultObject response, JsonGenerator jgen, SerializerProvider provider)
            throws IOException, JsonProcessingException {
        if (response.isDSResponse()) {
            jgen.writeStartObject();
            jgen.writeFieldName(RESPONSE);
            serialize((DSResponse) response.getReal(), jgen, provider);
            jgen.writeEndObject();
        } else if (response.isDSResponseList()) {
            List<DSResponse> res = (List<DSResponse>) response.getReal();
            jgen.writeStartObject();
            jgen.writeArrayFieldStart(RESPONSES);
            for (DSResponse re : res) {
                serialize(re, jgen, provider);
            }
            jgen.writeEndArray();
            jgen.writeEndObject();

        } else {
            jgen.writeObject(response.getReal());
        }

    }

    public void serialize(DSResponse response, JsonGenerator jgen, SerializerProvider provider)
            throws IOException, JsonProcessingException {

        jgen.writeStartObject();
        Status status = response.getStatus();
        jgen.writeNumberField(STATUS, status.value());
        jgen.writeBooleanField(ISDSRESPONSE, true);
        if (status == Status.STATUS_LOGIN_REQUIRED) {
            jgen.writeObjectField(AUTH, DATAX.LOGIN_REQUIRED_MARKER);
        } else if (status == Status.STATUS_LOGIN_SUCCESS) {
            jgen.writeObjectField(AUTH, DATAX.LOGIN_SUCCESS_MARKER);
        } else if (status == Status.STATUS_MAX_LOGIN_ATTEMPTS_EXCEEDED) {
            jgen.writeObjectField(AUTH, DATAX.MAX_LOGIN_ATTEMPTS_EXCEEDED_MARKER);
        } else {
            if (response.getAffectedRows() != null) {
                jgen.writeNumberField(AFFECTED_ROWS, response.getAffectedRows());
            }
            Object invalidate = response.getAttribute(Constants.INVALIDATE_CACHE);
            Pageable page = response.getAttachment(Pageable.class);
            if (invalidate != null) {
                jgen.writeBooleanField(INVALIDATE_CACHE, Boolean.valueOf(invalidate.toString()));
            }
            if (page != null) {
                jgen.writeNumberField(START_ROW, page.getStartRow());
                jgen.writeNumberField(END_ROW, page.getEndRow());
                jgen.writeNumberField(TOTAL_ROWS, page.getTotalRow());
            }
            Object o = response.getRawData();
            if (o != null) {
                jgen.writeObjectField(DATA, o);
            }
            Object[] errors = response.getErrors();
            if (errors != null && errors.length > 0) {
                jgen.writeObjectField(ERRORS, errors);
            }
        }
        jgen.writeEndObject();

    }

    @Override
    public Class<ResultObject> handledType() {
        return ResultObject.class;
    }
}