com.nhaarman.ellie.query.Insert Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of core Show documentation
Compile-time active record ORM for Android.
The newest version!
/*
 * Copyright (C) 2014 Michael Pardo
 * Copyright (C) 2014 Niek Haarman
 *
 * 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.nhaarman.ellie.query;

import android.text.TextUtils;

import com.nhaarman.ellie.Ellie;
import com.nhaarman.ellie.Model;

@SuppressWarnings({"HardCodedStringLiteral", "PublicInnerClass"})
public final class Insert extends QueryBase {

    private Ellie mEllie;

    public Insert() {
        super(null, null);
        mEllie = Ellie.getInstance();
    }

    public Insert with(final Ellie ellie) {
        mEllie = ellie;
        return this;
    }

    public Into into(final Class table) {
        return new Into(this, table);
    }

    public Into into(final Class table, final String... columns) {
        return new Into(this, table, columns);
    }

    @Override
    public String getPartSql() {
        return "INSERT";
    }

    @Override
    public Ellie getEllie() {
        return mEllie;
    }

    public static final class Into extends QueryBase {

        private final String[] mColumns;

        private Into(final Insert parent, final Class table, final String... columns) {
            super(parent, table);
            mColumns = columns;
        }

        public Values values(final Object... args) {
            return new Values(this, mTable, args);
        }

        @Override
        protected String getPartSql() {
            StringBuilder builder = new StringBuilder();
            builder.append("INTO ");
            builder.append(((Insert) mParent).getEllie().getTableName(mTable));
            if (mColumns != null && mColumns.length > 0) {
                builder.append("(").append(TextUtils.join(", ", mColumns)).append(")");
            }

            return builder.toString();
        }

        @Override
        public Ellie getEllie() {
            return mParent.getEllie();
        }
    }

    public static final class Values extends ExecutableQueryBase {

        private final Object[] mValuesArgs;

        private Values(final Query parent, final Class table, final Object[] args) {
            super(parent, table);
            mValuesArgs = args;
        }

        @Override
        public void execute() {
            if (((Into) mParent).mColumns != null && ((Into) mParent).mColumns.length != mValuesArgs.length) {
                throw new MalformedQueryException("Number of columns does not match number of values.");
            }
            super.execute();
        }

        @Override
        protected String getPartSql() {
            StringBuilder builder = new StringBuilder();
            builder.append("VALUES(");
            for (int i = 0; i < mValuesArgs.length; i++) {
                if (i > 0) {
                    builder.append(", ");
                }
                builder.append("?");
            }
            builder.append(")");
            return builder.toString();
        }

        @Override
        protected String[] getPartArgs() {
            return toStringArray(mValuesArgs);
        }

        @Override
        public Ellie getEllie() {
            return mParent.getEllie();
        }
    }
}