// ============================================================================ // // Copyright (C) 2006-2012 Talend Inc. - www.talend.com // // This source code is available under agreement available at // %InstallDIR%\features\org.talend.rcp.branding.%PRODUCTNAME%\%PRODUCTNAME%license.txt // // You should have received a copy of the agreement // along with this program; if not, write to Talend SA // 9 rue Pages 92150 Suresnes, France // // ============================================================================ package org.talend.repository.model.migration; import java.util.Arrays; import java.util.Date; import java.util.GregorianCalendar; import org.talend.commons.ui.runtime.exception.ExceptionHandler; import org.talend.core.language.ECodeLanguage; import org.talend.core.model.components.ComponentUtilities; import org.talend.core.model.components.ModifyComponentsAction; import org.talend.core.model.components.conversions.IComponentConversion; import org.talend.core.model.components.filters.IComponentFilter; import org.talend.core.model.components.filters.NameComponentFilter; import org.talend.core.model.migration.AbstractJobMigrationTask; import org.talend.core.model.properties.Item; import org.talend.designer.core.model.utils.emf.talendfile.NodeType; import org.talend.designer.core.model.utils.emf.talendfile.ProcessType; /** * change the FILE_ACTION type from checkbox to closed_list in tHDFSOutput, convert the value, see issue tdi bug 19870. */ public class ChangeValueTDIBug19870 extends AbstractJobMigrationTask { /* * (non-Javadoc) * * @seeorg.talend.core.model.migration.AbstractJobMigrationTask#executeOnProcess(org.talend.core.model.properties. * ProcessItem) */ @Override public ExecutionResult execute(Item item) { ProcessType processType = getProcessType(item); if (getProject().getLanguage() != ECodeLanguage.JAVA || processType == null) { return ExecutionResult.NOTHING_TO_DO; } String[] componentsName = new String[] { "tHDFSOutput" }; try { for (int i = 0; i < componentsName.length; i++) { IComponentFilter filter = new NameComponentFilter(componentsName[i]); ModifyComponentsAction.searchAndModify(item, processType, filter, Arrays.<IComponentConversion> asList(new IComponentConversion() { public void transform(NodeType node) { if (ComponentUtilities.getNodeProperty(node, "FILE_ACTION") == null || "false".equals(ComponentUtilities.getNodeProperty(node, "FILE_ACTION").getValue())) { ComponentUtilities.getNodeProperty(node, "FILE_ACTION").setValue("CREATE"); } else if ("true".equals(ComponentUtilities.getNodeProperty(node, "FILE_ACTION").getValue())) { ComponentUtilities.getNodeProperty(node, "FILE_ACTION").setValue("OVERWRITE"); } } })); } return ExecutionResult.SUCCESS_NO_ALERT; } catch (Exception e) { ExceptionHandler.process(e); return ExecutionResult.FAILURE; } } public Date getOrder() { GregorianCalendar gc = new GregorianCalendar(2012, 3, 6, 12, 0, 0); return gc.getTime(); } }