apex.benchmark.RedisHelper.java Source code

Introduction

Here is the source code for apex.benchmark.RedisHelper.java

Source

/**
 * Licensed under the terms of the Apache License 2.0. Please see LICENSE file in the project root for terms.
 */

package apex.benchmark;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.List;
import java.util.Map;

import org.apache.hadoop.conf.Configuration;
import org.apache.hadoop.fs.FSDataInputStream;
import org.apache.hadoop.fs.FileSystem;
import org.apache.hadoop.fs.Path;
import redis.clients.jedis.Jedis;

public class RedisHelper {
    private String host;
    private Jedis jedis;

    public void init(String host) {
        this.host = host;
        jedis = new Jedis(host);
    }

    public void clear(Integer db) {
        jedis.flushDB();
    }

    public void fillDB(String fileName) throws IOException {
        Path filePath = new Path(fileName);
        Configuration configuration = new Configuration();
        FileSystem fs;
        fs = FileSystem.newInstance(filePath.toUri(), configuration);
        FSDataInputStream inputStream = fs.open(filePath);
        BufferedReader bufferedReader;

        try {

            bufferedReader = new BufferedReader(new InputStreamReader(inputStream));

            String line;
            while ((line = bufferedReader.readLine()) != null) {

                String[] mapping = line.split("\\s+");

                if (mapping.length != 2) {
                    continue;
                }

                jedis.sadd("campaigns", mapping[0]);
                jedis.set(mapping[1], mapping[0]);
            }
        } catch (Exception e) {
            throw e;
        }
    }

    public void prepareRedis(Map<String, List<String>> campaigns) {
        jedis.select(0);
        jedis.flushAll();

        for (Map.Entry<String, List<String>> entry : campaigns.entrySet()) {
            String campaign = entry.getKey();
            jedis.sadd("campaigns", campaign);
            for (String ad : entry.getValue()) {
                jedis.set(ad, campaign);
            }
        }

        jedis.close();
    }
}