org.kaaproject.kaa.server.datamigration.EndpointProfileMigration.java Source code

Introduction

Here is the source code for org.kaaproject.kaa.server.datamigration.EndpointProfileMigration.java

Source

/*
 * Copyright 2014-2016 CyberVision, Inc.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package org.kaaproject.kaa.server.datamigration;

import static com.mongodb.client.model.Filters.eq;

import com.datastax.driver.core.Cluster;
import com.datastax.driver.core.Session;
import com.mongodb.MongoClient;
import com.mongodb.client.MongoCollection;
import com.mongodb.client.MongoDatabase;

import org.bson.Document;
import org.kaaproject.kaa.server.datamigration.utils.Options;

public class EndpointProfileMigration {
    private String host;
    private String dbName;
    private String nosql;

    /**
     * Create new instance of object that responsible for migration endpoint profiles.
     *
     * @param options the options
     */
    public EndpointProfileMigration(Options options) {
        dbName = options.getDbName();
        this.host = options.getHost();
        this.nosql = options.getNoSql();
    }

    /**
     * Add field use_raw_configuration_schema to endpointProfile that used to support devices using
     * SDK version 0.9.0
     */
    public void transform() {
        //mongo
        MongoClient client = new MongoClient(host);
        MongoDatabase database = client.getDatabase(dbName);
        MongoCollection<Document> endpointProfile = database.getCollection("endpoint_profile");
        endpointProfile.updateMany(new Document(), eq("$set", eq("use_raw_schema", false)));

        //cassandra
        Cluster cluster = Cluster.builder().addContactPoint(host).build();
        Session session = cluster.connect(dbName);
        session.execute("ALTER TABLE ep_profile ADD use_raw_schema boolean");
        session.close();
        cluster.close();

    }
}