io.warp10.ext.thrift.ThriftWarpScriptExtension Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of warp10-ext-thrift Show documentation

Thrift WarpScript Extension

There is a newer version: 1.1.1-uberjar

//
//   Copyright 2020  SenX S.A.S.
//
//   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 io.warp10.ext.thrift;

import java.util.HashMap;
import java.util.Map;

import io.warp10.script.functions.TYPEOF;
import io.warp10.script.functions.TYPEOF.TypeResolver;
import io.warp10.warp.sdk.WarpScriptExtension;

public class ThriftWarpScriptExtension extends WarpScriptExtension {
  private static Map functions;

  public static final String TENUM = "TENUM";
  public static final String TBASE = "TBASE";
  
  public static final String THRIFTC = "THRIFTC";
  
  static {
    functions = new HashMap();
    
    functions.put(THRIFTC, new THRIFTC(THRIFTC));
    functions.put("THRIFT->", new THRIFTTO("THRIFT->"));
    functions.put("->THRIFT", new TOTHRIFT("->THRIFT"));
    
    TYPEOF.addResolver(new TypeResolver() {            
      @Override
      public String typeof(Class clazz) {
        if (clazz.isAssignableFrom(DynamicEnum.class)) {
          return TENUM;
        } else if (clazz.isAssignableFrom(DynamicTBase.class)) {
          return TBASE;
        }
        return null;
      }
    });
  }
  
  @Override
  public Map getFunctions() {
    return functions;
  }
}