com.sborny.wordcountexample.MongoBolt.java Source code

Introduction

Here is the source code for com.sborny.wordcountexample.MongoBolt.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 com.sborny.wordcountexample;

import backtype.storm.topology.BasicOutputCollector;
import backtype.storm.topology.OutputFieldsDeclarer;
import backtype.storm.topology.base.BaseBasicBolt;
import backtype.storm.tuple.Fields;
import backtype.storm.tuple.Tuple;
import com.mongodb.MongoClient;
import com.mongodb.client.MongoDatabase;
import org.bson.Document;
import org.json.JSONException;
import org.json.JSONObject;

/**
 *
 * @author sander
 */
public class MongoBolt extends BaseBasicBolt {
    private MongoDatabase db;
    private String message;
    private String m_ip;
    private String m_port;
    private String dbName;
    private Boolean dbIsSet;

    public void prepareConfig(String mongo_ip, String mongoPort, String name, String m) {
        message = m;
        m_ip = mongo_ip;
        m_port = mongoPort;
        dbName = name;
        dbIsSet = false;
    }

    @Override
    public void declareOutputFields(OutputFieldsDeclarer declarer) {
        declarer.declare(new Fields(""));
    }

    @Override
    public void execute(Tuple input, BasicOutputCollector collector) {
        if (!dbIsSet) {
            MongoClient mongoClient = new MongoClient(m_ip, Integer.parseInt(m_port));
            db = mongoClient.getDatabase(dbName);
            System.out.println("GOT_DATABASE");
            dbIsSet = true;
        }

        System.out.println("DID_EXECUTE");
        JSONObject obj = new JSONObject(input.getStringByField(message));
        try {
            System.out.println("START_TRY");
            String collection = obj.getString("collectionId");
            JSONObject json = obj.getJSONObject(message);
            db.getCollection(collection).insertOne(Document.parse(json.toString()));
            System.out.println("END_TRY");
        } catch (JSONException ex) {
            System.out.println("START_EXC");
            String collection = "00000";
            db.getCollection(collection).insertOne(Document.parse(obj.toString()));
            System.out.println("END_EX");
        }
    }
}