io.prediction.CreateUserRequestBuilder Maven / Gradle / Ivy

Go to download
package io.prediction;

import com.google.gson.JsonObject;
import com.ning.http.client.Request;
import com.ning.http.client.RequestBuilder;

/**
 * Class to build User requests
 *
 * @author The PredictionIO Team (http://prediction.io)
 * @version 0.6.1
 * @since 0.2
 */

public class CreateUserRequestBuilder {
    private String apiUrl;
    private String apiFormat;
    private String appkey;
    private String uid;
    private Double latitude;
    private Double longitude;

    /**
     * Instantiate a request builder with mandatory arguments.
     * 
     * Do not use this directly. Please refer to "See Also".
     *
     * @param apiUrl the API URL
     * @param apiFormat the return format of the API
     * @param appkey the new app key to be used
     * @param uid the user ID
     *
     * @see Client#getCreateUserRequestBuilder
     */
    public CreateUserRequestBuilder(String apiUrl, String apiFormat, String appkey, String uid) {
        this.apiUrl = apiUrl;
        this.apiFormat = apiFormat;
        this.appkey = appkey;
        this.uid = uid;
    }

    /**
     * Add the "latitude" optional argument to the request.
     * 

     * Only certain data backend support geospatial indexing.
     * Please refer to the main documentation for more information.
     *
     * @param latitude latitude
     */
    public CreateUserRequestBuilder latitude(double latitude) {
        this.latitude = new Double(latitude);
        return this;
    }

    /**
     * Add the "longitude" optional argument to the request.
     * 

     * Only certain data backend support geospatial indexing.
     * Please refer to the main documentation for more information.
     *
     * @param longitude longitude
     */
    public CreateUserRequestBuilder longitude(double longitude) {
        this.longitude = new Double(longitude);
        return this;
    }

    /**
     * Build a request.
     * 
     * Do not use this directly. Please refer to "See Also".
     *
     * @see Client#createUser(CreateUserRequestBuilder)
     * @see Client#createUserAsFuture(CreateUserRequestBuilder)
     */
    public Request build() {
        RequestBuilder builder = new RequestBuilder("POST");
        builder.setUrl(this.apiUrl + "/users." + this.apiFormat);

        JsonObject requestJson = new JsonObject();

        requestJson.addProperty("pio_appkey", this.appkey);
        requestJson.addProperty("pio_uid", this.uid);
        if (this.latitude != null && this.longitude != null) {
            requestJson.addProperty("pio_latlng", this.latitude.toString() + "," + this.longitude.toString());
        }

        String requestJsonString = requestJson.toString();

        builder.setBody(requestJsonString);
        builder.setHeader("Content-Type","application/json");
        builder.setHeader("Content-Length", ""+requestJsonString.length());

        return builder.build();
    }
}