org.xlcloud.service.heat.parsers.resources.WaitConditionParser Maven / Gradle / Ivy

/*
 * Copyright 2012 AMG.lab, a Bull Group Company
 * 
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * 
 *    http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package org.xlcloud.service.heat.parsers.resources;

import static org.xlcloud.service.heat.template.fields.IntrinsicFunctionField.REF;
import static org.xlcloud.service.heat.template.fields.ResourceFields.DEPENDS_ON;
import static org.xlcloud.service.heat.template.fields.ResourceFields.TYPE;
import static org.xlcloud.service.heat.template.fields.WaitConditionFields.COUNT;
import static org.xlcloud.service.heat.template.fields.WaitConditionFields.HANDLE;
import static org.xlcloud.service.heat.template.fields.WaitConditionFields.TIMEOUT;

import org.apache.commons.lang.StringUtils;
import org.json.JSONException;
import org.json.JSONObject;
import org.xlcloud.service.heat.parsers.JSONParser;
import org.xlcloud.service.heat.parsers.resources.properties.PropertiesParser;
import org.xlcloud.service.heat.parsers.utils.JsonObjectWrapper;
import org.xlcloud.service.heat.parsers.utils.JsonObjectWrapperImpl;
import org.xlcloud.service.heat.template.resources.WaitCondition;

/**
 * Implementation of {@link JSONParser} for {@link WaitCondition}.
 * 
 * @author Michał Kamiński, AMG.net
 */
public class WaitConditionParser extends BaseResourceParser implements JSONParser {

    @Override
    public WaitCondition fromJSON(JSONObject json) throws JSONException {
        WaitCondition pojo = new WaitCondition();
        JsonObjectWrapper jsonWrapper = JsonObjectWrapperImpl.newInstance(json);
        pojo.setDependsOn(jsonWrapper.getString(DEPENDS_ON));
        appendPropertiesToPojo(pojo, jsonWrapper);
        return pojo;
    }

    @Override
    public JSONObject fromPojo(WaitCondition pojo) throws JSONException {
        JsonObjectWrapper json = JsonObjectWrapperImpl.newInstance();
        json.put(TYPE, pojo.getType());
        json.put(DEPENDS_ON, pojo.getDependsOn());
        appendPropertiesToJSON(json, pojo);
        return json.toJsonObject();
    }

    @Override
    protected PropertiesParser getPropertiesParser() {
        return new PropertiesParser() {

            @Override
            public void appendPropertiesToJSON(JsonObjectWrapper propsWrapper, WaitCondition pojo) throws JSONException {
                if (pojo.getCount() != null) {
                    propsWrapper.put(COUNT, parseValue(pojo.getCount()));
                }
                if (pojo.getHandle() != null) {
                    propsWrapper.put(HANDLE, JsonObjectWrapperImpl.newInstance().put(REF, pojo.getHandle()));
                }
                if (pojo.getTimeout() != null) {
                    propsWrapper.put(TIMEOUT, pojo.getTimeout().toString());
                }

            }

            private Object parseValue(String value) {
                try {
                    return new JSONObject(value);
                } catch (JSONException e) {
                    return value;
                }
            }

            @Override
            public void appendPropertiesToPojo(WaitCondition pojo, JsonObjectWrapper propsWrapper) throws JSONException {
                String count = propsWrapper.getString(COUNT);
                if (StringUtils.isNotBlank(count)) {
                    pojo.setCount(count);
                }

                // FIXME can fail if the handle is not passed using "Ref"
                JsonObjectWrapper handle = propsWrapper.getWrapper(HANDLE);
                pojo.setHandle(handle.getString(REF));

                String timeout = propsWrapper.getString(TIMEOUT);
                if (StringUtils.isNotBlank(timeout)) {
                    pojo.setTimeout(Long.valueOf(timeout));
                }
            }
        };
    }

}