com.hazelcast.client.protocol.generator.CodecModel Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of hazelcast-code-generator Show documentation
The newest version!
/*
 * Copyright (c) 2008-2015, Hazelcast, 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.hazelcast.client.protocol.generator;

import com.hazelcast.annotation.EventResponse;
import com.hazelcast.annotation.GenerateCodec;
import com.hazelcast.annotation.Nullable;
import com.hazelcast.annotation.Request;

import javax.lang.model.element.ExecutableElement;
import javax.lang.model.element.TypeElement;
import javax.lang.model.element.VariableElement;
import javax.lang.model.util.Elements;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.LinkedList;
import java.util.List;
import java.util.Map;

public class CodecModel implements Model {

    static final Map CUSTOM_CODEC_MAP = new HashMap();

    private final Lang lang;
    private short requestId;
    private String id;
    private String name;
    private String className;
    private String parentName;
    private String packageName;
    private String partitionIdentifier;
    private String comment = "";

    private int retryable;
    private int response;

    private Elements elementUtil;

    private final List requestParams = new LinkedList();
    private final List responseParams = new LinkedList();
    private final List events = new LinkedList();

    //TEST ONLY MOCKUP CONSTRUCTOR
    public CodecModel(boolean mockup) {
        this.retryable = 1;
        this.lang = Lang.JAVA;

        name = "put";
        parentName = "Map";
        className =
                CodeGenerationUtils.capitalizeFirstLetter(parentName) + CodeGenerationUtils.capitalizeFirstLetter(name) + "Codec";
        packageName = "com.hazelcast.client.impl.protocol.codec";

        response = 104;

        //request parameters

        ParameterModel pm = new ParameterModel();
        pm.name = "name";
        pm.type = "java.lang.String";
        pm.lang = Lang.JAVA;
        pm.nullable = true;
        requestParams.add(pm);

        pm = new ParameterModel();
        pm.name = "val";
        pm.type = "int";
        pm.lang = Lang.JAVA;
        pm.nullable = false;
        requestParams.add(pm);

        pm = new ParameterModel();
        pm.name = "address";
        pm.type = "com.hazelcast.nio.Address";
        pm.lang = Lang.JAVA;
        pm.nullable = false;
        requestParams.add(pm);

        pm = new ParameterModel();
        pm.name = "arr";
        pm.type = "int[]";
        pm.lang = Lang.JAVA;
        pm.nullable = false;
        requestParams.add(pm);

        pm = new ParameterModel();
        pm.name = "setD";
        pm.type = "java.util.Set<" + CodeGenerationUtils.DATA_FULL_NAME + ">";
        pm.lang = Lang.JAVA;
        pm.nullable = false;
        requestParams.add(pm);

        pm = new ParameterModel();
        pm.name = "mapIS";
        pm.type = "java.util.Map";
        pm.lang = Lang.JAVA;
        pm.nullable = false;
        requestParams.add(pm);

        pm = new ParameterModel();
        pm.name = "mapDD";
        pm.type = "java.util.Map<" + CodeGenerationUtils.DATA_FULL_NAME + ", " + CodeGenerationUtils.DATA_FULL_NAME + ">";
        pm.lang = Lang.JAVA;
        pm.nullable = false;
        requestParams.add(pm);

        pm = new ParameterModel();
        pm.name = "entryView";
        pm.type = "com.hazelcast.map.impl.SimpleEntryView<" + CodeGenerationUtils.DATA_FULL_NAME + ", "
                + CodeGenerationUtils.DATA_FULL_NAME + ">";
        pm.lang = Lang.JAVA;
        pm.nullable = true;
        requestParams.add(pm);

        //response parameters
        pm = new ParameterModel();
        pm.name = "name";
        pm.type = "long";
        pm.lang = Lang.JAVA;
        pm.nullable = false;
        responseParams.add(pm);

        EventModel eventModel = new EventModel();
        eventModel.type = 104;
        eventModel.name = "";

        List eventParam = new ArrayList();
        pm = new ParameterModel();
        pm.name = "name";
        pm.type = "java.lang.String";
        pm.lang = Lang.JAVA;
        pm.nullable = true;
        eventParam.add(pm);
        eventModel.eventParams = eventParam;
        events.add(eventModel);
    }

    public CodecModel(TypeElement parent, ExecutableElement methodElement, ExecutableElement responseElement,
                      List eventElementList, boolean retryable, Lang lang, Elements docCommentUtil) {
        this.retryable = retryable ? 1 : 0;
        this.lang = lang;

        name = methodElement.getSimpleName().toString();
        requestId = methodElement.getAnnotation(Request.class).id();
        partitionIdentifier = methodElement.getAnnotation(Request.class).partitionIdentifier();
        short masterId = parent.getAnnotation(GenerateCodec.class).id();
        id = CodeGenerationUtils.addHexPrefix(CodeGenerationUtils.mergeIds(masterId, requestId));
        parentName = parent.getAnnotation(GenerateCodec.class).name();
        className =
                CodeGenerationUtils.capitalizeFirstLetter(parentName) + CodeGenerationUtils.capitalizeFirstLetter(name) + "Codec";
        packageName = "com.hazelcast.client.impl.protocol.codec";

        if (lang != Lang.JAVA) {
            packageName = parent.getAnnotation(GenerateCodec.class).ns();
        }

        response = methodElement.getAnnotation(Request.class).response();

        elementUtil = docCommentUtil;

        initParameters(methodElement, responseElement, eventElementList, lang);
    }

    private void initParameters(ExecutableElement methodElement, ExecutableElement responseElement,
                                List eventElementList, Lang lang) {
        //request parameters
        for (VariableElement param : methodElement.getParameters()) {
            final Nullable nullable = param.getAnnotation(Nullable.class);

            ParameterModel pm = new ParameterModel();
            pm.name = param.getSimpleName().toString();
            pm.type = param.asType().toString();
            pm.lang = lang;
            pm.nullable = nullable != null;
            requestParams.add(pm);
        }

        //response parameters
        for (VariableElement param : responseElement.getParameters()) {
            final Nullable nullable = param.getAnnotation(Nullable.class);
            ParameterModel pm = new ParameterModel();
            pm.name = param.getSimpleName().toString();
            pm.type = param.asType().toString();
            pm.lang = lang;
            pm.nullable = nullable != null;
            responseParams.add(pm);
        }

        //event parameters
        for (ExecutableElement element : eventElementList) {
            EventModel eventModel = new EventModel();
            eventModel.comment = elementUtil.getDocComment(element);

            List eventParam = new ArrayList();
            for (VariableElement param : element.getParameters()) {
                final Nullable nullable = param.getAnnotation(Nullable.class);
                ParameterModel pm = new ParameterModel();
                pm.name = param.getSimpleName().toString();
                pm.type = param.asType().toString();
                pm.lang = lang;
                pm.nullable = nullable != null;
                pm.description = CodeGenerationUtils.getDescription(pm.name, eventModel.comment);
                eventParam.add(pm);
            }

            eventModel.type = element.getAnnotation(EventResponse.class).value();
            eventModel.name = element.getSimpleName().toString();
            eventModel.eventParams = eventParam;

            events.add(eventModel);
        }
    }

    public String getName() {
        return name;
    }

    public Lang getLang() {
        return lang;
    }

    public short getRequestId() {
        return requestId;
    }

    public String getId() {
        return id;
    }

    public String getPartitionIdentifier() {
        return partitionIdentifier;
    }

    public String getComment() {
        return comment;
    }

    public String getClassName() {
        return className;
    }

    public String getParentName() {
        return parentName;
    }

    public String getPackageName() {
        return packageName;
    }

    @Override
    public boolean isEmpty() {
        return requestParams.isEmpty();
    }

    public int getResponse() {
        return response;
    }

    public String getHexadecimalResponseId(){
        return CodeGenerationUtils.addHexPrefix(Integer.toHexString(response));
    }

    public List getRequestParams() {
        return requestParams;
    }

    public List getResponseParams() {
        return responseParams;
    }

    public List getEvents() {
        return events;
    }

    public int getRetryable() {
        return retryable;
    }

    public void setComment(String comment) {
        this.comment = comment;
    }

    public static class EventModel {
        private String name;
        private List eventParams;
        private int type;
        String comment = "";

        public int getType() {
            return type;
        }

        public String getHexadecimalTypeId(){
            return CodeGenerationUtils.addHexPrefix(Integer.toHexString(type));
        }

        public String getName() {
            return name;
        }

        public List getEventParams() {
            return eventParams;
        }

        public String getComment() {
            return comment;
        }
    }

    public static class ParameterModel {
        private String name;
        private String type;
        private Lang lang;
        private boolean nullable;
        private String description = "";

        public String getName() {
            return name;
        }

        public boolean isNullable() {
            return nullable;
        }

        public String getType() {
            return type;
        }

        public Lang getLang() {
            return lang;
        }

        public String getDescription() {
            return description;
        }
    }
}