capstone.yelpmodel.JSONWrapper.java Source code

Introduction

Here is the source code for capstone.yelpmodel.JSONWrapper.java

Source

/*
 * To change this license header, choose License Headers in Project Properties.
 * To change this template file, choose Tools | Templates
 * and open the template in the editor.
 */

package capstone.yelpmodel;

import java.io.BufferedReader;
import java.io.File;
import java.io.FileInputStream;
import java.io.InputStreamReader;
import java.io.Serializable;
import org.json.simple.JSONArray;
import org.json.simple.JSONValue;

public class JSONWrapper implements Serializable {

    public static final int MAX_RECORDS = 12800;

    JSONArray json = null;

    public JSONWrapper(String src) {
        json = (JSONArray) JSONValue.parse(src);
    }

    public JSONWrapper(File f) {

        try {

            String src = "[";

            FileInputStream fis = new FileInputStream(f);
            InputStreamReader isr = new InputStreamReader(fis);
            BufferedReader br = new BufferedReader(isr);

            int count = 0;
            while (br.ready() && count < MAX_RECORDS) {
                String line = br.readLine();
                src += line + ",\n";
                count++;
            }

            src += "]";

            json = (JSONArray) JSONValue.parse(src);
        }

        catch (Exception e) {
            System.out.println(e);
        }
    }

    public int size() {
        return json != null ? json.size() : -1;
    }

}