com.fluxtion.ext.streaming.builder.group.MultiKeyInfo Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of fluxtion-streaming-builder Show documentation
There is a newer version: 2.10.50
Show newest version
/* 
 * Copyright (C) 2018 V12 Technology Ltd.
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the Server Side Public License, version 1,
 * as published by MongoDB, Inc.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * Server Side Public License for more details.
 *
 * You should have received a copy of the Server Side Public License
 * along with this program.  If not, see 
 * .
 */
package com.fluxtion.ext.streaming.builder.group;

import com.fluxtion.ext.streaming.api.group.MultiKey;
import com.fluxtion.ext.streaming.api.numeric.MutableNumber;
import com.fluxtion.ext.streaming.builder.util.ImportMap;
import java.lang.reflect.Method;

/**
 * Meta data required to build a multi valued key, used in group by.
 *
 * @author Greg Higgins
 */
public class MultiKeyInfo {

    private String id;
    private Method sourceMethod;
    private String sourceMethodName;
    private String sourceType;
    private String mutNum = MutableNumber.class.getSimpleName();
    public final ImportMap importMap;

    public MultiKeyInfo(ImportMap importMap) {
        this.importMap = importMap;
        importMap.addImport(MutableNumber.class);
        importMap.addImport(MultiKey.class);
    }

    public void setSource(Method sourceMethod, String id) {
        this.id = id;
        this.sourceMethod = sourceMethod;
        this.sourceMethodName = sourceMethod.getName();
        this.sourceType = importMap.addImport(this.sourceMethod.getReturnType());//.getCanonicalName();
    }

    public String updateKeyMethod(String targetId) {
        String set = id + " = " + targetId + "." + sourceMethodName + "()";
        if (isNumeric()) {
            set = id + ".set(" + targetId + "." + sourceMethodName + "())";
        }
        return set;
    }

    public String initMethod(String targetId) {
        String set = sourceType + " " + id;
        if (isNumeric()) {
            set = mutNum + " " + id + " = new " + mutNum + "()";
        }
        return set;
    }

    public String resetMethod() {
        String set = id + " = null";
        if (isNumeric()) {
            set = id + ".set(0)";
        }
        return set;
    }

    public String copyMethod(String copyId) {
        String set = copyId + "." + id + " = " + id;
        if (isNumeric()) {
            set = copyId +"." + id + ".copyFrom(" + id + ")" ;
        }
        return set;
    }

    private boolean isNumeric() {
        return this.sourceMethod.getReturnType().isPrimitive();
    }

    public String getId() {
        return id;
    }

    @Override
    public String toString() {
        return "MultiKeyInfo{" + "id=" + id + ", sourceMethod=" + sourceMethod + ", sourceMethodName=" + sourceMethodName + ", sourceType=" + sourceType +'}';
    }

}