com.testvagrant.monitor.radiator.MongoIO Maven / Gradle / Ivy

Go to download
/*
 * Copyright (c) 2017.  TestVagrant Technologies
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see .
 *
 */

package com.testvagrant.monitor.radiator;


import com.mongodb.BasicDBObject;
import com.mongodb.DBObject;
import com.mongodb.MongoClient;
import com.mongodb.client.MongoCollection;
import com.mongodb.gridfs.GridFS;
import com.mongodb.gridfs.GridFSDBFile;
import com.mongodb.gridfs.GridFSInputFile;
import com.mongodb.util.JSON;
import com.testvagrant.monitor.MongoBase;
import com.testvagrant.monitor.clients.BuildsClient;
import com.testvagrant.monitor.entities.DeviceDetails;
import com.testvagrant.monitor.entities.device.DeviceType;
import com.testvagrant.monitor.entities.device.Platform;
import com.testvagrant.monitor.entities.device.Status;
import org.bson.Document;

import java.util.List;

import static com.testvagrant.monitor.constants.MongoDB.*;

public class MongoIO {
    protected static String latestBuildID;
    protected MongoClient mongoClient;

    public MongoIO() {
        mongoClient = MongoBase.getInstance();
        System.out.println("MongoClient:"+mongoClient.hashCode());
        if(latestBuildID==null) {
            getLatestBuild();
        }
    }

    protected static boolean screenshotExist(MongoCollection coll, String udid) {
        BasicDBObject query = new BasicDBObject(KEY_SCREENSHOTS_SCREENSHOT, new BasicDBObject(QUERY_NE, "")
                .append(QUERY_EXISTS, true)).append(KEY_DEVICES_UDID, udid).append(KEY_BUILD_ID,latestBuildID);
        return coll.count(query) == 1;
    }

    String getLatestBuild() {
        latestBuildID = new BuildsClient().getLatestBuildId();
        return latestBuildID;
    }

    private boolean fileExists(GridFS gridFS, GridFSInputFile file) {
        List gridFSDBFiles = gridFS.find((DBObject) JSON
                .parse(String.format("{ _id : '%s' }", file.getId())));
        if (gridFSDBFiles.size() > 0) {
            return true;
        }
        return false;
    }



    protected Document createDocument(DeviceDetails deviceDetails) {
        return new Document(KEY_DEVICES_PLATFORM,deviceDetails.getPlatform().name())
                .append(KEY_DEVICES_DEVICENAME,deviceDetails.getDeviceName())
                .append(KEY_DEVICES_STATUS,deviceDetails.getStatus().name())
                .append(KEY_DEVICES_PLATFORM_VERSION,deviceDetails.getPlatformVersion())
                .append(KEY_DEVICES_RUNSON,deviceDetails.getRunsOn().name())
                .append(KEY_DEVICES_UDID,deviceDetails.getUdid());
    }

    protected DeviceDetails readDocument(Document document) {
        Platform platform = Platform.valueOf(document.getString(KEY_DEVICES_PLATFORM));
        Status status = Status.valueOf(document.getString(KEY_DEVICES_STATUS));
        DeviceType deviceType = DeviceType.valueOf(document.getString("deviceType"));
        DeviceDetails deviceDetails = new DeviceDetails();
        deviceDetails.setPlatform(platform);
        deviceDetails.setStatus(status);
        deviceDetails.setPlatformVersion(document.getString(KEY_DEVICES_PLATFORM_VERSION));
        deviceDetails.setDeviceName(document.getString(KEY_DEVICES_DEVICENAME));
        deviceDetails.setRunsOn(deviceType);
        deviceDetails.setUdid(document.getString(KEY_DEVICES_UDID));
        return deviceDetails;
    }
}