com.javaetmoi.javabean.generator.NumberCodeGenerator Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of javaetmoi-javabean-marshaller Show documentation

The JavaBeanMarshaller class is responsible for serializing an object (graph) to it's Java code.

There is a newer version: 1.0.3

Show newest version

package com.javaetmoi.javabean.generator;

import com.javaetmoi.javabean.JavaBeanMarshaller;
import com.javaetmoi.javabean.bean.Item;
import com.javaetmoi.javabean.bean.SetterParam;
import com.squareup.javapoet.MethodSpec;

import java.math.BigDecimal;
import java.math.BigInteger;

public class NumberCodeGenerator implements CodeGenerator {

    @Override
    public boolean supports(Class valueClass) {
        return Number.class.isAssignableFrom(valueClass);
    }

    @Override
    public void generateSetter(MethodSpec.Builder method, SetterParam param) {
        if (Long.class.isAssignableFrom(param.getValueClass())) {
            method.addStatement("$L.$L($LL)", param.getVarName(), param.getSetterName(), param.getValue());
        } else if (Float.class.isAssignableFrom(param.getValueClass())) {
            method.addStatement("$L.$L($Lf)", param.getVarName(), param.getSetterName(), param.getValue());
        } else if (BigDecimal.class.isAssignableFrom(param.getValueClass())) {
            method.addStatement("$L.$L(new $T(\"$L\"))", param.getVarName(), param.getSetterName(), BigDecimal.class, param.getValue());
        } else if (BigInteger.class.isAssignableFrom(param.getValueClass())) {
            method.addStatement("$L.$L(new $T(\"$L\"))", param.getVarName(), param.getSetterName(), BigInteger.class, param.getValue());
        } else if (Short.class.isAssignableFrom(param.getValueClass())) {
            method.addStatement("$L.$L((short) $L)", param.getVarName(), param.getSetterName(), param.getValue());
        } else {
            method.addStatement("$L.$L($L)", param.getVarName(), param.getSetterName(), param.getValue());
        }
    }

    @Override
    public void refineItem(JavaBeanMarshaller marshaller, Item item) {
        if (Long.class.isAssignableFrom(item.getClazz())) {
            item.appendAfterVal("L");
        } else if (Float.class.isAssignableFrom(item.getClazz())) {
            item.appendAfterVal("f");
        }  else if (Short.class.isAssignableFrom(item.getClazz())) {
            item.appendBeforeVal("(short) ");
        } else if (BigDecimal.class.isAssignableFrom(item.getClazz())) {
            item.setPattern("new $T(\""+item.getVal()+"\")");
            item.setVal(BigDecimal.class);
        }

    }
}