2017-02-24 165 views
0

我是相當新的apache spark技術,在嘗試分析我從文件中提取的數據時遇到一些問題。比較RDD對象-Apache Spark

我有大量的基因信息,我把這些信息提交給RDD,迄今爲止這麼好。

JavaRDD<Gene> inputfile = sc.textFile(logFile).map(
     new Function<String, Gene>() { 
      @Override 
      public Gene call(String line) throws Exception { 
       String[] values = line.split("\t"); 
       Gene gen = null; 

       //We are only interested in genes; 
       if(values.length > 2 && values[2].equalsIgnoreCase("gene") && !line.contains("#")){ 
        String[] infoGene = values[8].split(";"); 

        String geneId = StringUtils.substringBetween(infoGene[0], "\""); 
        String geneType = StringUtils.substringBetween(infoGene[2], "\""); 
        String geneName = StringUtils.substringBetween(infoGene[4], "\""); 
        gen = new Gene(geneName,values[3],values[4]); 

        return gen; 
       } 
       return gen; 
      } 
     } 
    ).filter(new Function<Gene, Boolean>() { 
     @Override 
     public Boolean call(Gene gene) throws Exception { 
      if(gene == null) 
       return false; 
      else 
       return true; 
     } 
    }); 

基因類:

public class Gene implements Serializable{ 
String firstBp; 
String lastBp; 
String name; 

public Gene(String name, String firstBp, String lastBp) { 
    this.name = name; 
    this.firstBp = firstBp; 
    this.lastBp = lastBp; 
} 

public String getFirstBp() { 
    return firstBp; 
} 

public String getLastBp() { 
    return lastBp; 
} 

public String getName() { 
    return name; 
} 

public String toString(){ 
    return name + " " + firstBp + " " + lastBp; 
}} 

問題從這裏開始,我需要的,如果2個基因疊加分析,併爲我做了這個簡單的實用功能:

public static Boolean isOverlay(Gene gene1, Gene gene2){ 
    int gene1First = Integer.parseInt(gene1.getFirstBp()); 
    int gene1Last = Integer.parseInt(gene1.getLastBp()); 
    int gene2First = Integer.parseInt(gene2.getFirstBp()); 
    int gene2Last = Integer.parseInt(gene2.getLastBp()); 

    if(gene2First >= gene1First && gene2First <= gene1Last) // FirstBp - Gene2 inside 
     return true; 
    else if (gene2Last >= gene1First && gene2Last <= gene1Last) // LastBP - Gene2 inside 
     return true; 
    else if (gene1First >= gene2First && gene1First <= gene2Last) // FirstBp - Gene1 inside 
     return true; 
    else if (gene1Last >= gene2First && gene1Last <= gene2Last) // LastBP - Gene1 inside 
     return true; 
    else 
     return false; 
} 

現在我在做什麼,我認爲是錯誤的是將RDD對象轉變爲列表:

List<Gene> genesList = inputfile.collect(); 

並遍歷該列表以檢查是否有重疊並將保存到文件的結果花費時間,因爲我沒有使用spark。

List<OverlayPair> overlayPairList= new ArrayList<OverlayPair>(); 
List<String> visitedGenes = new ArrayList<String>(); 

for (Gene gene1 : genesList){ 

     for (Gene gene2 : genesList) { 
      if (gene1.getName().equalsIgnoreCase(gene2.getName()) || visitedGenes.contains(gene2.getName())) { 
       continue; 
      } 

      if (isOverlay(gene1, gene2)) 
       overlayPairList.add(new OverlayPair(gene1.getName(), gene2.getName())); 

     } 
     visitedGenes.add(gene1.getName()); 
    } 

    JavaRDD<OverlayPair> overlayFile = sc.parallelize(overlayPairList); 

    //Export the results to the file 
    String outputDirectory = "/Users/joaoalmeida/Desktop/Dissertacao/sol/data/mitocondrias/feup-pp/project/data/output/overlays"; 
    overlayFile.coalesce(1).saveAsTextFile(outputDirectory); 

Overlay對基本上是一個具有2個基因名稱的對象。

有沒有辦法做這第二部分,同時利用火花?因爲這些2的時間複雜性對於我目前擁有的數據量來說是很大的。

回答

1

是的,有,你必須使用RDD.cartesian函數來獲得所有的對,然後你可以基本上應用你寫的函數。