com.google.api.codegen.packagegen.java.JavaGrpcPackageGenerator.java Source code

Introduction

Here is the source code for com.google.api.codegen.packagegen.java.JavaGrpcPackageGenerator.java

Source

/* Copyright 2017 Google LLC
 *
 * 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
 *
 *      https://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.google.api.codegen.packagegen.java;

import com.google.api.codegen.common.CodeGenerator;
import com.google.api.codegen.common.GeneratedResult;
import com.google.api.codegen.config.PackageMetadataConfig;
import com.google.api.codegen.config.ProtoApiModel;
import com.google.api.codegen.rendering.CommonSnippetSetRunner;
import com.google.api.codegen.viewmodel.metadata.PackageMetadataView;
import com.google.api.tools.framework.model.Model;
import com.google.api.tools.framework.snippet.Doc;
import com.google.common.collect.ImmutableList;
import com.google.common.collect.ImmutableMap;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;

/** Performs grpc package generation for Java */
public class JavaGrpcPackageGenerator implements CodeGenerator<Doc> {

    private final JavaPackageTransformer transformer;
    private final Model model;
    private final PackageMetadataConfig config;

    public JavaGrpcPackageGenerator(JavaPackageTransformer transformer, Model model, PackageMetadataConfig config) {
        this.transformer = transformer;
        this.model = model;
        this.config = config;
    }

    @Override
    public List<String> getInputFileNames() {
        return ImmutableList.of();
    }

    @Override
    public Map<String, GeneratedResult<Doc>> generate() {
        ImmutableMap.Builder<String, GeneratedResult<Doc>> results = new ImmutableMap.Builder<>();

        ProtoApiModel apiModel = new ProtoApiModel(model);
        ArrayList<PackageMetadataView> metadataViews = new ArrayList<>();
        metadataViews.addAll(transformer.transform(apiModel, config));

        for (PackageMetadataView view : metadataViews) {
            CommonSnippetSetRunner runner = new CommonSnippetSetRunner(view, false);
            results.putAll(runner.generate(view));
        }
        return results.build();
    }
}