View Javadoc

1   /*
2    * Copyright (c) 2010.  The Apache Software Foundation
3    *
4    * Licensed to the Apache Software Foundation (ASF) under one
5    * or more contributor license agreements.  See the NOTICE file
6    * distributed with this work for additional information
7    * regarding copyright ownership.  The ASF licenses this file
8    * to you under the Apache License, Version 2.0 (the
9    * "License"); you may not use this file except in compliance
10   * with the License.  You may obtain a copy of the License at
11   *
12   *     http://www.apache.org/licenses/LICENSE-2.0
13   *
14   * Unless required by applicable law or agreed to in writing, software
15   * distributed under the License is distributed on an "AS IS" BASIS,
16   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
17   * See the License for the specific language governing permissions and
18   * limitations under the License.
19   */
20  
21  package org.apache.expreval.expr.casestmt;
22  
23  import org.apache.expreval.client.NullColumnValueException;
24  import org.apache.expreval.client.ResultMissingColumnException;
25  import org.apache.expreval.expr.TypeSupport;
26  import org.apache.expreval.expr.node.BooleanValue;
27  import org.apache.expreval.expr.node.DateValue;
28  import org.apache.expreval.expr.node.GenericValue;
29  import org.apache.expreval.expr.node.NumberValue;
30  import org.apache.expreval.expr.node.StringValue;
31  import org.apache.hadoop.hbase.hbql.client.HBqlException;
32  import org.apache.hadoop.hbase.hbql.impl.HConnectionImpl;
33  
34  public class DelegateCaseWhen extends GenericCaseWhen {
35  
36      public DelegateCaseWhen(final GenericValue arg0, final GenericValue arg1) {
37          super(null, arg0, arg1);
38      }
39  
40      public Class<? extends GenericValue> validateTypes(final GenericValue parentExpr,
41                                                         final boolean allowCollections) throws HBqlException {
42  
43          this.validateParentClass(BooleanValue.class, this.getExprArg(0).validateTypes(this, false));
44          final Class<? extends GenericValue> valueType = this.getExprArg(1).validateTypes(this, false);
45  
46          if (TypeSupport.isParentClass(StringValue.class, valueType))
47              this.setTypedExpr(new StringCaseWhen(this.getExprArg(0), this.getExprArg(1)));
48          else if (TypeSupport.isParentClass(NumberValue.class, valueType))
49              this.setTypedExpr(new NumberCaseWhen(this.getExprArg(0), this.getExprArg(1)));
50          else if (TypeSupport.isParentClass(DateValue.class, valueType))
51              this.setTypedExpr(new DateCaseWhen(this.getExprArg(0), this.getExprArg(1)));
52          else if (TypeSupport.isParentClass(BooleanValue.class, valueType))
53              this.setTypedExpr(new BooleanCaseWhen(this.getExprArg(0), this.getExprArg(1)));
54          else
55              this.throwInvalidTypeException(valueType);
56  
57          return this.getTypedExpr().validateTypes(parentExpr, false);
58      }
59  
60      public GenericValue getOptimizedValue() throws HBqlException {
61          this.optimizeAllArgs();
62          return this;
63      }
64  
65      public Object getValue(final HConnectionImpl conn, final Object object) throws HBqlException,
66                                                                                     ResultMissingColumnException,
67                                                                                     NullColumnValueException {
68          return this.getTypedExpr().getValue(conn, object);
69      }
70  }