com.gemstone.gemfire.cache.query.internal.Functions Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of gemfire-core Show documentation
SnappyData store based off Pivotal GemFireXD
The newest version!
/*
 * Copyright (c) 2010-2015 Pivotal Software, Inc. All rights reserved.
 *
 * 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. See accompanying
 * LICENSE file.
 */

package com.gemstone.gemfire.cache.query.internal;

import java.util.*;

import com.gemstone.gemfire.cache.query.*;
import com.gemstone.gemfire.internal.i18n.LocalizedStrings;

import java.text.*;

/**
 * Class Description
 *
 * @version     $Revision: 1.1 $
 * @author      ericz
 */

public class Functions
{
    public static final int ELEMENT = 1;

    public static Object nvl(CompiledValue arg1, CompiledValue arg2, ExecutionContext context)
      throws FunctionDomainException, TypeMismatchException, NameResolutionException,
      QueryInvocationTargetException {
      Object value = arg1.evaluate(context);
      if(value == null) {
        return arg2.evaluate(context);
      }
      return value;
    }

    public static Date to_date (CompiledValue cv1, CompiledValue cv2, ExecutionContext context)
    throws FunctionDomainException, TypeMismatchException, NameResolutionException,
    QueryInvocationTargetException {
      Object value1 = cv1.evaluate(context);
      Object value2 = cv2.evaluate(context);
      if(!(value1 instanceof String) || !(value2 instanceof String)) {
		throw new QueryInvalidException(LocalizedStrings.Functions_PARAMETERS_TO_THE_TO_DATE_FUNCTION_SHOULD_BE_STRICTLY_SIMPLE_STRINGS.toLocalizedString());
      }
      String dateStr = (String) value1;
      String format = (String) value2;
      Date dt=null;		
      try {
        // Removed the following line so that to data format conforms to
        // SimpleDateFormat exactly (bug 39144)
        //format = ((format.replaceAll("Y", "y")).replaceAll("m", "M")).replaceAll("D", "d");

        /*if((format.indexOf("MM") == -1 && format.indexOf("M") == -1) || (format.
  indexOf("dd") == -1 && format.indexOf("d") == -1) || (format.indexOf("yyyy") ==
  -1 && format.indexOf("yy") == -1))
        {
          throw new QueryInvalidException("Malformed date format string");
        }
        if(format.indexOf("MMM") != -1 || format.indexOf("ddd") != -1 || format.in
  dexOf("yyyyy") != -1)
        {
          throw new QueryInvalidException("Malformed date format string");
        } */

        SimpleDateFormat sdf1 = new SimpleDateFormat(format);
        dt = sdf1.parse(dateStr);

      }
      catch (Exception ex){
        throw new QueryInvalidException(LocalizedStrings.Functions_MALFORMED_DATE_FORMAT_STRING_AS_THE_FORMAT_IS_0.toLocalizedString(format), ex);
      }
      return dt;
    }
    //end of to_date

    public static Object element(Object arg)
        throws FunctionDomainException, TypeMismatchException
    {
        if (arg == null || arg == QueryService.UNDEFINED)
            return QueryService.UNDEFINED;

        if (arg instanceof Collection)
        {
            Collection c = (Collection)arg;
            checkSingleton(c.size());
            return c.iterator().next();
        }

        // not a Collection, must be an array
        Class clazz = arg.getClass();
        if (!clazz.isArray())
            throw new TypeMismatchException(LocalizedStrings.Functions_THE_ELEMENT_FUNCTION_CANNOT_BE_APPLIED_TO_AN_OBJECT_OF_TYPE_0.toLocalizedString(clazz.getName()));

        // handle arrays
        if (arg instanceof Object[])
        {
            Object[] a = (Object[])arg;
            checkSingleton(a.length);
            return a[0];
        }

        if (arg instanceof int[])
        {
            int[] a = (int[])arg;
            checkSingleton(a.length);
            return Integer.valueOf(a[0]);
        }

        if (arg instanceof long[])
        {
            long[] a = (long[])arg;
            checkSingleton(a.length);
            return Long.valueOf(a[0]);
        }


        if (arg instanceof boolean[])
        {
            boolean[] a = (boolean[])arg;
            checkSingleton(a.length);
            return Boolean.valueOf(a[0]);
        }

        if (arg instanceof byte[])
        {
            byte[] a = (byte[])arg;
            checkSingleton(a.length);
            return Byte.valueOf(a[0]);
        }

        if (arg instanceof char[])
        {
            char[] a = (char[])arg;
            checkSingleton(a.length);
            return Character.valueOf(a[0]);
        }

        if (arg instanceof double[])
        {
            double[] a = (double[])arg;
            checkSingleton(a.length);
            return Double.valueOf(a[0]);
        }

        if (arg instanceof float[])
        {
            float[] a = (float[])arg;
            checkSingleton(a.length);
            return new Float(a[0]);
        }


        if (arg instanceof short[])
        {
            short[] a = (short[])arg;
            checkSingleton(a.length);
            return Short.valueOf(a[0]);
        }

        // did I miss something?
        throw new TypeMismatchException(LocalizedStrings.Functions_THE_ELEMENT_FUNCTION_CANNOT_BE_APPLIED_TO_AN_OBJECT_OF_TYPE_0.toLocalizedString(clazz.getName()));
    }

    private static void checkSingleton(int size)
        throws FunctionDomainException
    {
        if (size != 1)
            throw new FunctionDomainException(LocalizedStrings.Functions_ELEMENT_APPLIED_TO_PARAMETER_OF_SIZE_0.toLocalizedString(Integer.valueOf(size)));
    }


}