Home | History | Annotate | Download | only in sherlockholmes 
      1 /*
      2  * Copyright (C) 2017 The Android Open Source Project
      3  *
      4  * Licensed under the Apache License, Version 2.0 (the "License");
      5  * you may not use this file except in compliance with the License.
      6  * You may obtain a copy of the License at
      7  *
      8  *      http://www.apache.org/licenses/LICENSE-2.0
      9  *
     10  * Unless required by applicable law or agreed to in writing, software
     11  * distributed under the License is distributed on an "AS IS" BASIS,
     12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
     13  * See the License for the specific language governing permissions and
     14  * limitations under the License.
     15  */
     16 package androidx.work.integration.testapp.sherlockholmes;
     17 
     18 import android.support.annotation.NonNull;
     19 import android.util.Log;
     20 
     21 import androidx.work.Data;
     22 import androidx.work.Worker;
     23 import androidx.work.integration.testapp.db.TestDatabase;
     24 import androidx.work.integration.testapp.db.WordCount;
     25 
     26 import java.io.DataInputStream;
     27 import java.io.FileInputStream;
     28 import java.io.IOException;
     29 import java.util.ArrayList;
     30 import java.util.Collections;
     31 import java.util.HashMap;
     32 import java.util.List;
     33 import java.util.Map;
     34 
     35 /**
     36  * A Worker that combines the word counts of various works and outputs them.
     37  */
     38 public class TextReducingWorker extends Worker {
     39 
     40     private static final String INPUT_FILE = "input_file";
     41 
     42     private Map<String, Integer> mWordCount = new HashMap<>();
     43 
     44     @Override
     45     public @NonNull Result doWork() {
     46         Data input = getInputData();
     47         String[] inputFiles = input.getStringArray(INPUT_FILE);
     48         if (inputFiles == null) {
     49             throw new IllegalArgumentException();
     50         }
     51 
     52         for (int i = 0; i < inputFiles.length; ++i) {
     53             FileInputStream fileInputStream = null;
     54             DataInputStream dataInputStream = null;
     55             try {
     56                 fileInputStream = getApplicationContext().openFileInput(inputFiles[i]);
     57                 dataInputStream = new DataInputStream(fileInputStream);
     58                 while (dataInputStream.available() > 0) {
     59                     String word = dataInputStream.readUTF();
     60                     int count = dataInputStream.readInt();
     61                     if (mWordCount.containsKey(word)) {
     62                         count += mWordCount.get(word);
     63                     }
     64                     mWordCount.put(word, count);
     65                 }
     66             } catch (IOException e) {
     67                 return Result.FAILURE;
     68             } finally {
     69                 if (dataInputStream != null) {
     70                     try {
     71                         dataInputStream.close();
     72                     } catch (IOException e) {
     73                         // Do nothing.
     74                     }
     75                 }
     76                 if (fileInputStream != null) {
     77                     try {
     78                         fileInputStream.close();
     79                     } catch (IOException e) {
     80                         // Do nothing.
     81                     }
     82                 }
     83             }
     84         }
     85 
     86         List<Map.Entry<String, Integer>> sortedList = new ArrayList<>(mWordCount.size());
     87         sortedList.addAll(mWordCount.entrySet());
     88         Collections.sort(sortedList, (o1, o2) -> o2.getValue().compareTo(o1.getValue()));
     89 
     90         TestDatabase db = TestDatabase.getInstance(getApplicationContext());
     91         db.beginTransaction();
     92         try {
     93             for (Map.Entry<String, Integer> entry : sortedList) {
     94                 WordCount wc = new WordCount();
     95                 wc.mWord = entry.getKey();
     96                 wc.mCount = entry.getValue();
     97                 db.getWordCountDao().insertWordCount(wc);
     98             }
     99             db.setTransactionSuccessful();
    100         } finally {
    101             db.endTransaction();
    102         }
    103 
    104         Log.d("Reduce", "Reduction finished");
    105         return Result.SUCCESS;
    106     }
    107 }
    108