com.palantir.atlasdb.table.description.render.NamedRowResultRenderer Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of atlasdb-client Show documentation
Palantir open source project
There is a newer version: 0.1193.0
/*
 * (c) Copyright 2018 Palantir Technologies Inc. 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 com.palantir.atlasdb.table.description.render;

import static com.palantir.atlasdb.table.description.render.ColumnRenderers.TypeName;
import static com.palantir.atlasdb.table.description.render.ColumnRenderers.VarName;
import static com.palantir.atlasdb.table.description.render.ColumnRenderers.short_name;

import com.palantir.atlasdb.table.description.NamedColumnDescription;
import java.util.SortedSet;

@SuppressWarnings("checkstyle:AvoidNestedBlocks")
class NamedRowResultRenderer extends Renderer {
    private final String row;
    private final String rowResult;
    private final SortedSet cols;

    NamedRowResultRenderer(Renderer parent, String name, SortedSet cols) {
        super(parent);
        this.row = name + "Row";
        this.rowResult = name + "RowResult";
        this.cols = cols;
    }

    @Override
    protected void run() {
        line("public static final class ", rowResult, " implements TypedRowResult {");
        {
            fields();
            line();
            staticFactory();
            line();
            constructor();
            line();
            getRowName();
            line();
            getRowNameFun();
            line();
            fromRawRowResultFun();
            line();
            for (NamedColumnDescription col : cols) {
                hasCol(col);
                line();
            }
            for (NamedColumnDescription col : cols) {
                getCol(col);
                line();
            }
            for (NamedColumnDescription col : cols) {
                getColFun(col);
                line();
            }
            renderToString();
        }
        line("}");
    }

    private void fields() {
        line("private final RowResult row;");
    }

    private void staticFactory() {
        line("public static ", rowResult, " of(RowResult row) {");
        {
            line("return new ", rowResult, "(row);");
        }
        line("}");
    }

    private void constructor() {
        line("private ", rowResult, "(RowResult row) {");
        {
            line("this.row = row;");
        }
        line("}");
    }

    private void getRowName() {
        line("@Override");
        line("public ", row, " getRowName() {");
        {
            line("return ", row, ".BYTES_HYDRATOR.hydrateFromBytes(row.getRowName());");
        }
        line("}");
    }

    private void getRowNameFun() {
        line("public static Function<", rowResult, ", ", row, "> getRowNameFun() {");
        {
            line("return new Function<", rowResult, ", ", row, ">() {");
            {
                line("@Override");
                line("public ", row, " apply(", rowResult, " rowResult) {");
                {
                    line("return rowResult.getRowName();");
                }
                line("}");
            }
            line("};");
        }
        line("}");
    }

    private void fromRawRowResultFun() {
        line("public static Function, ", rowResult, "> fromRawRowResultFun() {");
        {
            line("return new Function, ", rowResult, ">() {");
            {
                line("@Override");
                line("public ", rowResult, " apply(RowResult rowResult) {");
                {
                    line("return new ", rowResult, "(rowResult);");
                }
                line("}");
            }
            line("};");
        }
        line("}");
    }

    private void hasCol(NamedColumnDescription col) {
        line("public boolean has", VarName(col), "() {");
        {
            line("return row.getColumns().containsKey(PtBytes.toCachedBytes(", short_name(col), "));");
        }
        line("}");
    }

    private void getCol(NamedColumnDescription col) {
        line("public ", TypeName(col), " get", VarName(col), "() {");
        {
            line("byte[] bytes = row.getColumns().get(PtBytes.toCachedBytes(", short_name(col), "));");
            line("if (bytes == null) {");
            {
                line("return null;");
            }
            line("}");
            line(
                    Renderers.CamelCase(col.getLongName()),
                    " value = ",
                    Renderers.CamelCase(col.getLongName()),
                    ".BYTES_HYDRATOR.hydrateFromBytes(bytes);");
            line("return value.getValue();");
        }
        line("}");
    }

    private void getColFun(NamedColumnDescription col) {
        line("public static Function<", rowResult, ", ", TypeName(col), "> get", VarName(col), "Fun() {");
        {
            line("return new Function<", rowResult, ", ", TypeName(col), ">() {");
            {
                line("@Override");
                line("public ", TypeName(col), " apply(", rowResult, " rowResult) {");
                {
                    line("return rowResult.get", VarName(col), "();");
                }
                line("}");
            }
            line("};");
        }
        line("}");
    }

    private void renderToString() {
        line("@Override");
        line("public String toString() {");
        {
            line("return MoreObjects.toStringHelper(getClass().getSimpleName())");
            line("    .add(\"RowName\", getRowName())");
            for (NamedColumnDescription col : cols) {
                line("    .add(\"", VarName(col), "\", get", VarName(col), "())");
            }
            line("    .toString();");
        }
        line("}");
    }
}