com.dangdang.ddframe.job.cloud.scheduler.config.CloudJobConfigurationGsonFactory Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of elastic-job-cloud-scheduler Show documentation
There is a newer version: 2.1.5
Show newest version
/*
 * Copyright 1999-2015 dangdang.com.
 * 
 * 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 com.dangdang.ddframe.job.cloud.scheduler.config;

import com.dangdang.ddframe.job.config.JobTypeConfiguration;
import com.dangdang.ddframe.job.util.json.AbstractJobConfigurationGsonTypeAdapter;
import com.dangdang.ddframe.job.util.json.GsonFactory;
import com.google.common.base.Preconditions;
import com.google.gson.stream.JsonReader;
import com.google.gson.stream.JsonWriter;
import lombok.AccessLevel;
import lombok.NoArgsConstructor;

import java.io.IOException;
import java.util.Map;

/**
 * Cloud作业配置的Gson工厂.
 *
 * @author zhangliang
 * @author caohao
 */
@NoArgsConstructor(access = AccessLevel.PRIVATE)
public final class CloudJobConfigurationGsonFactory {
    
    static  {
        GsonFactory.registerTypeAdapter(CloudJobConfiguration.class, new CloudJobConfigurationGsonTypeAdapter());
    }
    
    /**
     * 将作业配置转换为JSON字符串.
     *
     * @param cloudJobConfig 作业配置对象
     * @return 作业配置JSON字符串
     */
    public static String toJson(final CloudJobConfiguration cloudJobConfig) {
        return GsonFactory.getGson().toJson(cloudJobConfig);
    }
    
    /**
     * 将JSON字符串转换为作业配置.
     *
     * @param cloudJobConfigJson 作业配置JSON字符串
     * @return 作业配置对象
     */
    public static CloudJobConfiguration fromJson(final String cloudJobConfigJson) {
        return GsonFactory.getGson().fromJson(cloudJobConfigJson, CloudJobConfiguration.class);
    }
    
    /**
     * Cloud作业配置的Json转换适配器.
     *
     * @author zhangliang
     */
    public static final class CloudJobConfigurationGsonTypeAdapter extends AbstractJobConfigurationGsonTypeAdapter {
        
        @Override
        protected void addToCustomizedValueMap(final String jsonName, final JsonReader in, final Map customizedValueMap) throws IOException {
            switch (jsonName) {
                case "cpuCount":
                    customizedValueMap.put("cpuCount", in.nextDouble());
                    break;
                case "memoryMB":
                    customizedValueMap.put("memoryMB", in.nextDouble());
                    break;
                case "appURL":
                    customizedValueMap.put("appURL", in.nextString());
                    break;
                case "bootstrapScript":
                    customizedValueMap.put("bootstrapScript", in.nextString());
                    break;
                case "jobExecutionType":
                    customizedValueMap.put("jobExecutionType", in.nextString());
                    break;
                case "beanName":
                    customizedValueMap.put("beanName", in.nextString());
                    break;
                case "applicationContext":
                    customizedValueMap.put("applicationContext", in.nextString());
                    break;
                default:
                    in.skipValue();
                    break;
            }
        }
        
        @Override
        protected CloudJobConfiguration getJobRootConfiguration(final JobTypeConfiguration typeConfig, final Map customizedValueMap) {
            Preconditions.checkNotNull(customizedValueMap.get("cpuCount"), "cpuCount cannot be null.");
            Preconditions.checkArgument((double) customizedValueMap.get("cpuCount") >= 0.01, "cpuCount cannot be less than 0.01");
            Preconditions.checkNotNull(customizedValueMap.get("memoryMB"), "memoryMB cannot be null.");
            Preconditions.checkArgument((double) customizedValueMap.get("memoryMB") >= 16, "memory cannot be less than 16");
            Preconditions.checkNotNull(customizedValueMap.get("appURL"), "appURL cannot be null.");
            Preconditions.checkNotNull(customizedValueMap.get("bootstrapScript"), "bootstrapScript cannot be null.");
            Preconditions.checkNotNull(customizedValueMap.get("jobExecutionType"), "jobExecutionType cannot be null.");
            if (customizedValueMap.containsKey("beanName") && customizedValueMap.containsKey("applicationContext")) {
                return new CloudJobConfiguration(typeConfig, (double) customizedValueMap.get("cpuCount"), (double) customizedValueMap.get("memoryMB"),
                        (String) customizedValueMap.get("appURL"), (String) customizedValueMap.get("bootstrapScript"),
                        JobExecutionType.valueOf(customizedValueMap.get("jobExecutionType").toString()), customizedValueMap.get("beanName").toString(),
                        customizedValueMap.get("applicationContext").toString());
            } else {
                return new CloudJobConfiguration(typeConfig, (double) customizedValueMap.get("cpuCount"), (double) customizedValueMap.get("memoryMB"),
                        (String) customizedValueMap.get("appURL"), (String) customizedValueMap.get("bootstrapScript"),
                        JobExecutionType.valueOf(customizedValueMap.get("jobExecutionType").toString()));
            }
        }
        
        @Override
        protected void writeCustomized(final JsonWriter out, final CloudJobConfiguration value) throws IOException {
            out.name("cpuCount").value(value.getCpuCount());
            out.name("memoryMB").value(value.getMemoryMB());
            out.name("appURL").value(value.getAppURL());
            out.name("bootstrapScript").value(value.getBootstrapScript());
            out.name("jobExecutionType").value(value.getJobExecutionType().name());
            out.name("beanName").value(value.getBeanName());
            out.name("applicationContext").value(value.getApplicationContext());
        }
    }
}