package com.atlas;

import java.io.FileWriter;

import org.json.simple.JSONArray;
import org.json.simple.JSONObject;
import org.json.simple.parser.JSONParser;

import com.sun.jersey.api.client.Client;
import com.sun.jersey.api.client.ClientResponse;
import com.sun.jersey.api.client.WebResource;

public class AtlasClient {

	public static void main(String[] args) {
		try {
			String url = "";
			String currentUrl = "http://<Atlas Host>:21000/api/atlas/entities?type=hive_table";
	//		FileWriter writer = new FileWriter(
	//				"C:/Users/prasanta_sahoo1/Documents/DataLake/Healthcare_data/Condition2.json", true);
			Client client = Client.create();
			while (!url.equals(currentUrl)) {
				url = currentUrl;
				try {
					WebResource webResource = client.resource(currentUrl);
					ClientResponse response = webResource.accept("application/json").get(ClientResponse.class);
					if (response.getStatus() != 200) {
						throw new RuntimeException("Failed : HTTP error code : " + response.getStatus());
					}
					StringBuffer sb = new StringBuffer();
					String output = response.getEntity(String.class);
					//System.out.println(output);
					System.out.println("=================================================");
					String[] keyValue = output.split(":");
					for (String s : keyValue) {
						sb.append(s.trim());
					//	System.out.println(s);
					}
					System.out.println("Output from Server .... \n ");
					JSONParser parser = new JSONParser();
					JSONObject json = (JSONObject) parser.parse(output);
					
					JSONArray resultJson=(JSONArray)json.get("results");
					for(int i = 0; i < resultJson.size(); i++) {
						System.out.println(resultJson.get(i));
					}
					
					
					//System.out.println(output);

					/*JSONParser parser = new JSONParser();
					JSONObject json = (JSONObject) parser.parse(output);

					JSONArray entry = (JSONArray) json.get("entry");
					for (int i = 0; i < entry.size(); i++) {
						JSONObject entryJson = (JSONObject) entry.get(i);
						//System.out.println(entryJson);
						JSONObject resource = (JSONObject) entryJson.get("resource");
						//System.out.println(resource + "\n");
						writer.write(resource.toString());
						writer.write("\n");
						writer.flush();
					}
*/
				/*	JSONArray link = (JSONArray) json.get("link");
					for (int i = 0; i < link.size(); i++) {
						JSONObject linkJson = (JSONObject) link.get(i);
						//System.out.println(linkJson);
						System.out.println("................................................");
						String relation = (String) linkJson.get("relation");
						System.out.println(relation);
						if (relation.equals("next")) {
							String nextUrl = (String) linkJson.get("url");
							currentUrl = nextUrl;
							System.out.println(nextUrl);
						}
					}*/
					response.close();
					// System.out.println(sb.toString().replace("\n", ""));
				} catch (RuntimeException e) {
				}
			}
			//writer.close();

		} catch (Exception e) {
			e.printStackTrace();
		}

	}

}