org.pentaho.metaverse.impl.model.kettle.json.JobMetaJsonSerializer.java Source code

Introduction

Here is the source code for org.pentaho.metaverse.impl.model.kettle.json.JobMetaJsonSerializer.java

Source

/*! ******************************************************************************
 *
 * Pentaho Data Integration
 *
 * Copyright (C) 2002-2015 by Pentaho : http://www.pentaho.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 org.pentaho.metaverse.impl.model.kettle.json;

import com.fasterxml.jackson.core.JsonGenerator;
import com.fasterxml.jackson.databind.JavaType;
import org.pentaho.di.core.exception.KettleException;
import org.pentaho.di.job.Job;
import org.pentaho.di.job.JobHopMeta;
import org.pentaho.di.job.JobMeta;
import org.pentaho.di.job.entry.JobEntryBase;
import org.pentaho.di.job.entry.JobEntryCopy;
import org.pentaho.di.job.entry.JobEntryInterface;
import org.pentaho.di.repository.ObjectId;
import org.pentaho.di.repository.StringObjectId;
import org.pentaho.metaverse.api.model.kettle.HopInfo;
import org.pentaho.metaverse.impl.model.kettle.LineageRepository;
import org.pentaho.metaverse.messages.Messages;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

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

/**
 * User: RFellows Date: 12/15/14
 */
public class JobMetaJsonSerializer extends AbstractMetaJsonSerializer<JobMeta> {
    private static final Logger LOGGER = LoggerFactory.getLogger(JobMetaJsonSerializer.class);

    public JobMetaJsonSerializer(Class<JobMeta> aClass) {
        super(aClass);
    }

    public JobMetaJsonSerializer(JavaType javaType) {
        super(javaType);
    }

    public JobMetaJsonSerializer(Class<?> aClass, boolean b) {
        super(aClass, b);
    }

    @Override
    protected void serializeHops(JobMeta meta, JsonGenerator json) throws IOException {
        // Hops
        json.writeArrayFieldStart(JSON_PROPERTY_HOPS);
        int numberOfHops = meta.nrJobHops();
        for (int i = 0; i < numberOfHops; i++) {
            JobHopMeta hopMeta = meta.getJobHop(i);
            HopInfo hopInfo = new HopInfo(hopMeta);
            json.writeObject(hopInfo);
        }
        json.writeEndArray();
    }

    @Override
    protected void serializeSteps(JobMeta meta, JsonGenerator json) throws IOException {
        json.writeArrayFieldStart(JSON_PROPERTY_STEPS);
        int numberOfEntries = meta.nrJobEntries();
        for (int i = 0; i < numberOfEntries; i++) {
            JobEntryCopy jobEntry = meta.getJobEntry(i);
            LineageRepository repo = getLineageRepository();
            ObjectId jobId = meta.getObjectId() == null ? new StringObjectId(meta.getName()) : meta.getObjectId();
            ObjectId entryId = jobEntry.getObjectId() == null ? new StringObjectId(jobEntry.getName())
                    : jobEntry.getObjectId();

            JobEntryInterface jobEntryInterface = jobEntry.getEntry();
            JobEntryBase jobEntryBase = getJobEntryBase(jobEntryInterface);
            Job job = new Job(null, meta);
            jobEntryBase.setParentJob(job);
            jobEntryInterface.setObjectId(entryId);
            try {
                jobEntryInterface.saveRep(repo, null, jobId);
            } catch (KettleException e) {
                LOGGER.warn(Messages.getString("INFO.Serialization.Trans.Step", jobEntry.getName()), e);
            }
            json.writeObject(jobEntryBase);
        }
        json.writeEndArray();
    }

    protected JobEntryBase getJobEntryBase(JobEntryInterface jobEntryInterface) {
        JobEntryBase jobEntryBase = new JobEntryBase(jobEntryInterface.getName(),
                jobEntryInterface.getDescription());
        if (jobEntryInterface instanceof JobEntryBase) {
            jobEntryBase = (JobEntryBase) jobEntryInterface;
        }
        return jobEntryBase;
    }

    @Override
    protected List<String> getUsedVariables(JobMeta meta) {
        return meta.getUsedVariables();
    }

}