/* * Licensed to the Apache Software Foundation (ASF) under one * or more contributor license agreements. See the NOTICE file * distributed with this work for additional information * regarding copyright ownership. The ASF licenses this file * to you 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.apache.pig.backend.hadoop.executionengine.physicalLayer.expressionOperators; import org.apache.pig.backend.executionengine.ExecException; import org.apache.pig.backend.hadoop.executionengine.physicalLayer.POStatus; import org.apache.pig.backend.hadoop.executionengine.physicalLayer.Result; import org.apache.pig.backend.hadoop.executionengine.physicalLayer.plans.PhyPlanVisitor; import org.apache.pig.data.DataType; import org.apache.pig.impl.plan.OperatorKey; import org.apache.pig.impl.plan.NodeIdGenerator; import org.apache.pig.impl.plan.VisitorException; public class Mod extends BinaryExpressionOperator { /** * */ private static final long serialVersionUID = 1L; public Mod(OperatorKey k) { super(k); } public Mod(OperatorKey k, int rp) { super(k, rp); } @Override public void visit(PhyPlanVisitor v) throws VisitorException { v.visitMod(this); } @Override public String name() { return "Mod" + "[" + DataType.findTypeName(resultType) + "]" +" - " + mKey.toString(); } @Override public Result getNext(Integer i) throws ExecException{ Result r = accumChild(null, i); if (r != null) { return r; } byte status; Result res; Integer left = null, right = null; res = lhs.getNext(left); status = res.returnStatus; if(status != POStatus.STATUS_OK || res.result == null) { return res; } left = (Integer) res.result; res = rhs.getNext(right); status = res.returnStatus; if(status != POStatus.STATUS_OK || res.result == null) { return res; } right = (Integer) res.result; res.result = Integer.valueOf(left % right); return res; } @Override public Result getNext(Long i) throws ExecException{ Result r = accumChild(null, i); if (r != null) { return r; } byte status; Result res; Long left = null, right = null; res = lhs.getNext(left); status = res.returnStatus; if(status != POStatus.STATUS_OK || res.result == null) { return res; } left = (Long) res.result; res = rhs.getNext(right); status = res.returnStatus; if(status != POStatus.STATUS_OK || res.result == null) { return res; } right = (Long) res.result; res.result = Long.valueOf(left % right); return res; } @Override public Mod clone() throws CloneNotSupportedException { Mod clone = new Mod(new OperatorKey(mKey.scope, NodeIdGenerator.getGenerator().getNextNodeId(mKey.scope))); clone.cloneHelper(this); return clone; } }