    001    /*
    002     * Licensed to the Apache Software Foundation (ASF) under one or more
    003     * contributor license agreements. See the NOTICE file distributed with
    004     * this work for additional information regarding copyright ownership.
    005     * The ASF licenses this file to You under the Apache license, Version 2.0
    006     * (the "License"); you may not use this file except in compliance with
    007     * the License. You may obtain a copy of the License at
    008     *
    009     *      http://www.apache.org/licenses/LICENSE-2.0
    010     *
    011     * Unless required by applicable law or agreed to in writing, software
    012     * distributed under the License is distributed on an "AS IS" BASIS,
    013     * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
    014     * See the license for the specific language governing permissions and
    015     * limitations under the license.
    016     */
    017    package org.apache.logging.log4j.nosql.appender.mongodb;
    018
    019    import java.util.Collections;
    020
    021    import org.apache.logging.log4j.nosql.appender.NoSqlObject;
    022
    023    import com.mongodb.BasicDBList;
    024    import com.mongodb.BasicDBObject;
    025
    026    /**
    027     * The MongoDB implementation of {@link NoSqlObject}.
    028     */
    029    public final class MongoDbObject implements NoSqlObject<BasicDBObject> {
    030        private final BasicDBObject mongoObject;
    031
    032        public MongoDbObject() {
    033            this.mongoObject = new BasicDBObject();
    034        }
    035
    036        @Override
    037        public void set(final String field, final Object value) {
    038            this.mongoObject.append(field, value);
    039        }
    040
    041        @Override
    042        public void set(final String field, final NoSqlObject<BasicDBObject> value) {
    043            this.mongoObject.append(field, value.unwrap());
    044        }
    045
    046        @Override
    047        public void set(final String field, final Object[] values) {
    048            final BasicDBList list = new BasicDBList();
    049            Collections.addAll(list, values);
    050            this.mongoObject.append(field, list);
    051        }
    052
    053        @Override
    054        public void set(final String field, final NoSqlObject<BasicDBObject>[] values) {
    055            final BasicDBList list = new BasicDBList();
    056            for (final NoSqlObject<BasicDBObject> value : values) {
    057                list.add(value.unwrap());
    058            }
    059            this.mongoObject.append(field, list);
    060        }
    061
    062        @Override
    063        public BasicDBObject unwrap() {
    064            return this.mongoObject;
    065        }
    066    }