org.pentaho.mongo.wrapper.KerberosMongoClientWrapper.java Source code

Introduction

Here is the source code for org.pentaho.mongo.wrapper.KerberosMongoClientWrapper.java

Source

/*!
  * Copyright 2010 - 2014 Pentaho Corporation.  All rights reserved.
  *
  * 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.pentaho.mongo.wrapper;

import com.mongodb.DBCollection;
import com.mongodb.MongoClient;
import com.mongodb.MongoCredential;
import org.pentaho.mongo.AuthContext;
import org.pentaho.mongo.KerberosHelper;
import org.pentaho.mongo.MongoDbException;
import org.pentaho.mongo.MongoProp;
import org.pentaho.mongo.MongoProperties;
import org.pentaho.mongo.MongoUtilLogger;
import org.pentaho.mongo.wrapper.collection.KerberosMongoCollectionWrapper;
import org.pentaho.mongo.wrapper.collection.MongoCollectionWrapper;

import java.util.ArrayList;
import java.util.List;

/**
 * Implementation of MongoClientWrapper which uses the GSSAPI auth mechanism.
 * Should only be instantiated by MongoClientWrapperFactory.
 */
class KerberosMongoClientWrapper extends UsernamePasswordMongoClientWrapper {
    private final AuthContext authContext;

    public KerberosMongoClientWrapper(MongoProperties props, MongoUtilLogger log) throws MongoDbException {
        super(props, log);
        authContext = new AuthContext(KerberosHelper.login(getUser(), props));
    }

    KerberosMongoClientWrapper(MongoClient client, MongoUtilLogger log, String username, AuthContext authContext) {
        super(client, log, username);
        this.authContext = authContext;
    }

    @Override
    public List<MongoCredential> getCredentialList() {
        List<MongoCredential> credList = new ArrayList<MongoCredential>();
        credList.add(MongoCredential.createGSSAPICredential(props.get(MongoProp.USERNAME)));
        return credList;
    }

    @Override
    protected MongoCollectionWrapper wrap(DBCollection collection) {
        return KerberosInvocationHandler.wrap(MongoCollectionWrapper.class, authContext,
                new KerberosMongoCollectionWrapper(collection, authContext));
    }

    public AuthContext getAuthContext() {
        return authContext;
    }
}