com.nhaarman.ellie.internal.codegen.step.ColumnStep Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of compiler 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.internal.codegen.step;

import com.google.common.collect.Sets;
import com.nhaarman.ellie.annotation.Column;
import com.nhaarman.ellie.internal.codegen.Registry;
import com.nhaarman.ellie.internal.codegen.validator.ColumnValidator;
import com.nhaarman.ellie.internal.codegen.validator.Validator;

import java.util.Set;

import javax.annotation.processing.RoundEnvironment;
import javax.lang.model.element.Element;
import javax.lang.model.element.TypeElement;

public class ColumnStep implements ProcessingStep {

    private final Validator mValidator;

    public ColumnStep(final Registry registry) {
        mValidator = new ColumnValidator(registry);
    }

    @Override
    public void process(final Set annotations, final RoundEnvironment roundEnv) {
        final Set elements = Sets.newHashSet(roundEnv.getElementsAnnotatedWith(Column.class));
        for (Element element : elements) {
            mValidator.validate(element.getEnclosingElement(), element);
        }
    }
}