2016-03-28 5 views
0

Ich möchte mehrere Pig UDF definieren. Jeder von ihnen wird einen anderen Teil der Daten extrahieren. In meinem Fall sind die Daten JSON-Dokumente, die eine komplexe Struktur mit vielen verschachtelten JSON-Objekten haben.Schwein: Mehrere UDF in einer Klasse

Das Problem ist, dass ich für jetzt eine andere Eval-Klasse für jede Funktion, die ich brauche, erstellt habe. Jede dieser Klassen implementiert die exec(). Gibt es eine Möglichkeit, alle Funktionen auf die gleiche UDF-Klasse zu setzen und sie vom Schwein aus aufzurufen?

Beispiel eines meiner UDF:

public class PigGetTimestamps extends EvalFunc<Tuple>{ 
    public Tuple exec(org.apache.pig.data.Tuple input) throws IOException {   

    if (input == null || input.size() == 0){ 
     return null; 
    } 

    try { 

     String inputString = DataType.toString(input.get(0)); 
     try 
     { 
      String[] tokens=inputString.split("\t"); 
      if (tokens.length<1) 
       return null; 
      Document document=new Document(tokens[0], true, false); 
      long timestamp_fetch=document.getTimestamp_fetch(); 
      long timestamp_pub=document.getTimestampPub(); 
      Tuple output = TupleFactory.getInstance().newTuple(2); 
      output.set(0,timestamp_pub); 
      output.set(1,timestamp_fetch); 
      return output; 
     } 
     catch(Exception e) 
     { 
      return null; 
     } 

    } catch (Exception e) { 
     System.out.println("Can't extract field; error = " + e.getMessage()); 
     return null; 
    }   
} 

Antwort

0

Sie die gleiche Klasse verwenden können, erstellen Sie mehrere Instanzen davon - wenn im Konstruktor werden Sie angeben, über welche Funktion es

tun Also in Ihrem Schwein Sie werden die verschiedenen Instanzen definieren:

define udf1 my.package.udf.MultiFunc('1'); 
define udf2 my.package.udf.MultiFunc('2'); 

Und Sie werden UDF-Klasse wie folgt aussehen:

public class MultiFunc extends EvalFunc<String> { 

     private String operation; 
     public MultiFunc(String operation){ 
      this.data = operation; 

     } 

     @Override 
     public String exec(Tuple tuple) throws IOException { 
     switch (operation){ 
      case "+": 
       //your code here; 
       break; 

      case "-": 
       //your code here; 
       break; 

      break; 

     }   
} 
+0

Gibt es einen Standard Weg, könnten Sie das Ergebnis erhalten, aber es ist ein Hack – Athar