com.strapdata.transformer.StringLocaleTransformer Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of javassist-maven-plugin-core Show documentation
There is a newer version: 2.7
Show newest version
/*
 * Copyright (c) 2016 Strapdata ([email protected]).
 * 
 * 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.strapdata.transformer;

import javassist.CannotCompileException;
import javassist.ClassPool;
import javassist.CtClass;
import javassist.CtMethod;
import javassist.expr.ExprEditor;
import javassist.expr.MethodCall;
import javassist.expr.NewExpr;
import nl.topicus.plugins.maven.javassist.ClassTransformer;
import nl.topicus.plugins.maven.javassist.ILogger;
import nl.topicus.plugins.maven.javassist.TransformationException;


/**
 * See https://issues.apache.org/jira/browse/CASSANDRA-9626.
 * Modify Cassandra bytes code to be Locale independent.
 * @author vroyer
 *
 */
public class StringLocaleTransformer extends ClassTransformer {

    public void applyTransformations(ClassPool pool, CtClass classToTransform) throws TransformationException {
        try {
            final CtMethod[] targetMethods = classToTransform.getDeclaredMethods();
            for (int i = 0; i < targetMethods.length; i++) {
                targetMethods[i].instrument(new ExprEditor() {
                    
                    public void edit(final MethodCall m) throws CannotCompileException {
                        ILogger logger = getLogger();
                        if ("java.lang.String".equals(m.getClassName())) {
                           if ("format".equals(m.getMethodName()) && m.getSignature().startsWith("(Ljava/lang/String;")) {
                               logger.info("Modifing format() @ "+m.getFileName()+":"+m.getLineNumber());
                               m.replace("{$_ = java.lang.String.format(java.util.Locale.ROOT, $$);}");
                           } else if ("toUpperCase".equals(m.getMethodName()) && m.getSignature().startsWith("()")) {
                               logger.info("Modifing toUpperCase() @ "+m.getFileName()+":"+m.getLineNumber());
                               m.replace("{$_ = $proceed(java.util.Locale.ROOT);}");
                           } else if ("toLowerCase".equals(m.getMethodName()) && m.getSignature().startsWith("()")) {
                               logger.info("Modifing toLowerCase() @ "+m.getFileName()+":"+m.getLineNumber());
                               m.replace("{$_ = $proceed(java.util.Locale.ROOT);}");
                           }
                        }
                    }
                    
                    // see http://tutorials.jenkov.com/java-internationalization/decimalformat.html#decimalformatsymbols
                    public void edit(NewExpr e) throws CannotCompileException {
                        ILogger logger = getLogger();
                        if ("java.text.DecimalFormat".equals(e.getClassName())) {
                            if (e.getSignature().startsWith("(Ljava/lang/String;)")) {
                                logger.info("Modifing DecimalFormat(String) @ "+e.getFileName()+":"+e.getLineNumber());
                                e.replace("{$_ = $proceed($1, java.text.DecimalFormatSymbols.getInstance(java.util.Locale.ROOT) );}");
                            }
                        }
                    }
                });
            }
        } catch (CannotCompileException e) {
            e.printStackTrace();
        }
    }
}