com.acmeair.mongo.services.FlightServiceImpl.java Source code

Introduction

Here is the source code for com.acmeair.mongo.services.FlightServiceImpl.java

Source

/*******************************************************************************
* Copyright (c) 2015 IBM Corp.
*
* 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 com.acmeair.mongo.services;

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

import java.util.ArrayList;
import java.util.Date;
import java.util.List;
import java.util.logging.Level;

import javax.annotation.PostConstruct;
import javax.inject.Inject;

import org.bson.Document;
import org.json.simple.JSONObject;
import org.json.simple.parser.JSONParser;
import org.json.simple.parser.ParseException;

import com.acmeair.AirportCodeMapping;
import com.acmeair.mongo.ConnectionManager;
import com.acmeair.mongo.MongoConstants;
import com.acmeair.service.DataService;
import com.acmeair.service.FlightService;
import com.acmeair.service.KeyGenerator;
import com.mongodb.BasicDBObject;
import com.mongodb.client.MongoCollection;
import com.mongodb.client.MongoCursor;
import com.mongodb.client.MongoDatabase;
//import com.mongodb.async.client.*;

@DataService(name = MongoConstants.KEY, description = MongoConstants.KEY_DESCRIPTION)
public class FlightServiceImpl extends FlightService implements MongoConstants {

    //private final static Logger logger = Logger.getLogger(FlightService.class.getName()); 

    private MongoCollection<Document> flight;
    private MongoCollection<Document> flightSegment;
    private MongoCollection<Document> airportCodeMapping;

    @Inject
    KeyGenerator keyGenerator;

    @PostConstruct
    public void initialization() {
        MongoDatabase database = ConnectionManager.getConnectionManager().getDB();
        flight = database.getCollection("flight");
        flightSegment = database.getCollection("flightSegment");
        airportCodeMapping = database.getCollection("airportCodeMapping");
    }

    @Override
    public Long countFlights() {
        return flight.count();
    }

    @Override
    public Long countFlightSegments() {
        return flightSegment.count();
    }

    @Override
    public Long countAirports() {
        return airportCodeMapping.count();
    }

    protected String getFlight(String flightId, String segmentId) {
        return flight.find(eq("_id", flightId)).first().toJson();
    }

    @Override
    protected String getFlightSegment(String fromAirport, String toAirport) {
        try {
            return flightSegment.find(new BasicDBObject("originPort", fromAirport).append("destPort", toAirport))
                    .first().toJson();
        } catch (java.lang.NullPointerException e) {
            if (logger.isLoggable(Level.FINE)) {
                logger.fine("getFlghSegment returned no flightSegment available");
            }
            return "";
        }
    }

    @Override
    protected List<String> getFlightBySegment(String segment, Date deptDate) {
        try {
            JSONObject segmentJson = (JSONObject) new JSONParser().parse(segment);
            MongoCursor<Document> cursor;

            if (deptDate != null) {
                if (logger.isLoggable(Level.FINE)) {
                    logger.fine("getFlghtBySegment Search String : "
                            + new BasicDBObject("flightSegmentId", segmentJson.get("_id"))
                                    .append("scheduledDepartureTime", deptDate).toJson());
                }
                cursor = flight.find(new BasicDBObject("flightSegmentId", segmentJson.get("_id"))
                        .append("scheduledDepartureTime", deptDate)).iterator();
            } else {
                cursor = flight.find(eq("flightSegmentId", segmentJson.get("_id"))).iterator();
            }

            List<String> flights = new ArrayList<String>();
            try {
                while (cursor.hasNext()) {
                    Document tempDoc = cursor.next();

                    if (logger.isLoggable(Level.FINE)) {
                        logger.fine("getFlghtBySegment Before : " + tempDoc.toJson());
                    }

                    Date deptTime = (Date) tempDoc.get("scheduledDepartureTime");
                    Date arvTime = (Date) tempDoc.get("scheduledArrivalTime");
                    tempDoc.remove("scheduledDepartureTime");
                    tempDoc.append("scheduledDepartureTime", deptTime.toString());
                    tempDoc.remove("scheduledArrivalTime");
                    tempDoc.append("scheduledArrivalTime", arvTime.toString());

                    if (logger.isLoggable(Level.FINE)) {
                        logger.fine("getFlghtBySegment after : " + tempDoc.toJson());
                    }

                    flights.add(tempDoc.append("flightSegment", segmentJson).toJson());
                }
            } finally {
                cursor.close();
            }
            return flights;
        } catch (ParseException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
            return null;
        }
    }

    @Override
    public void storeAirportMapping(AirportCodeMapping mapping) {
        Document airportDoc = new Document("_id", mapping.getAirportCode()).append("airportName",
                mapping.getAirportName());
        airportCodeMapping.insertOne(airportDoc);
    }

    @Override
    public AirportCodeMapping createAirportCodeMapping(String airportCode, String airportName) {
        return new AirportCodeMapping(airportCode, airportName);
    }

    @Override
    public void createNewFlight(String flightSegmentId, Date scheduledDepartureTime, Date scheduledArrivalTime,
            int firstClassBaseCost, int economyClassBaseCost, int numFirstClassSeats, int numEconomyClassSeats,
            String airplaneTypeId) {
        String id = keyGenerator.generate().toString();
        Document flightDoc = new Document("_id", id).append("firstClassBaseCost", firstClassBaseCost)
                .append("economyClassBaseCost", economyClassBaseCost)
                .append("numFirstClassSeats", numFirstClassSeats)
                .append("numEconomyClassSeats", numEconomyClassSeats).append("airplaneTypeId", airplaneTypeId)
                .append("flightSegmentId", flightSegmentId).append("scheduledDepartureTime", scheduledDepartureTime)
                .append("scheduledArrivalTime", scheduledArrivalTime);

        flight.insertOne(flightDoc);
    }

    @Override
    public void storeFlightSegment(String flightSeg) {
        try {
            JSONObject flightSegJson = (JSONObject) new JSONParser().parse(flightSeg);
            storeFlightSegment((String) flightSegJson.get("_id"), (String) flightSegJson.get("originPort"),
                    (String) flightSegJson.get("destPort"), (int) flightSegJson.get("miles"));
        } catch (ParseException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }
    }

    @Override
    public void storeFlightSegment(String flightName, String origPort, String destPort, int miles) {
        Document flightSegmentDoc = new Document("_id", flightName).append("originPort", origPort)
                .append("destPort", destPort).append("miles", miles);

        flightSegment.insertOne(flightSegmentDoc);
    }
}