generator

package
v0.20.14 Latest Latest
Warning

This package is not in the latest version of its module.

 Go to latest Published: Aug 12, 2021 License: MIT Imports: 24 Imported by: 0

Documentation

Overview

The code generator for the plugin for the Google protocol buffer compiler. It generates Go code from the protocol buffer description files read by the main routine.

Index

Constants

View Source 
const (
	PackageType = 12
	OptionType  = 8
	ServiceType = 6
	MessageType = 4
)
View Source 
const NilTag = "nil"

Variables

This section is empty.

Functions

func CamelCase 

func CamelCase(s string) string

CamelCase returns the CamelCased name. If there is an interior underscore followed by a lower case letter, drop the underscore and convert the letter to upper case. There is a remote possibility of this rewrite causing a name collision, but it's so remote we're prepared to pretend it's nonexistent - since the C++ generator lowercases names, it's extremely unlikely to have two fields with different capitalizations. In short, _my_field_name_2 becomes XMyFieldName_2.

func CamelCaseSlice 

func CamelCaseSlice(elem []string) string

CamelCaseSlice is like CamelCase, but the argument is a slice of strings to be joined with "_".

func EmbedFieldName added in v0.10.19 

func EmbedFieldName(goTyp string) string

func FileName added in v0.10.19 

func FileName(file *FileDescriptor) string

func GetCustomType added in v0.10.19 

func GetCustomType(field *descriptor.FieldDescriptorProto) (packageName string, typ string, err error)

func GoMapValueTypes added in v0.10.19 

func GoMapValueTypes(mapField, valueField *descriptor.FieldDescriptorProto, goValueType, goValueAliasType string) (nullable bool, outGoType string, outGoAliasType string)

GoMapValueTypes returns the map value Go type and the alias map value Go type (for casting), taking into account whether the map is nullable or the value is a message.

func GoTypeToName added in v0.10.19 

func GoTypeToName(goTyp string) string

func IsScalar added in v0.10.19 

func IsScalar(field *descriptor.FieldDescriptorProto) bool

Is this field a scalar numeric type?

func NewPluginImports added in v0.10.19 

func NewPluginImports(generator *Generator) *pluginImports

func RegisterPlugin 

func RegisterPlugin(p Plugin)

RegisterPlugin installs a (second-order) plugin to be run when the Go output is generated. It is typically called during initialization.

func RegisterUniquePackageName 

func RegisterUniquePackageName(pkg string, f *FileDescriptor) string

RegisterUniquePackageName create and remember a guaranteed unique package name. Pkg is the candidate name. The FileDescriptor parameter is unused.

Types

type AnnotatedAtoms 

type AnnotatedAtoms struct {
	// contains filtered or unexported fields
}

AnnotatedAtoms is a list of atoms (as consumed by P) that records the file name and proto AST path from which they originated.

func Annotate 

func Annotate(file *FileDescriptor, path string, atoms ...interface{}) *AnnotatedAtoms

Annotate records the file name and proto AST path of a list of atoms so that a later call to P can emit a link from each atom to its origin.

type Comment 

type Comment struct {
	Tag  string
	Text string
}

type Descriptor 

type Descriptor struct {
	*descriptor.DescriptorProto
	// contains filtered or unexported fields
}

Descriptor represents a protocol buffer message.

func (*Descriptor) File 

func (c *Descriptor) File() *FileDescriptor

func (*Descriptor) GoImportPath 

func (c *Descriptor) GoImportPath() GoImportPath

GoImportPath is the import path of the Go package containing the type.

func (*Descriptor) IsGroup added in v0.10.19 

func (d *Descriptor) IsGroup() bool

func (*Descriptor) Path added in v0.10.19 

func (d *Descriptor) Path() string

func (*Descriptor) TypeName 

func (d *Descriptor) TypeName() []string

TypeName returns the elements of the dotted type name. The package name is not part of this name.

type EnumDescriptor 

type EnumDescriptor struct {
	*descriptor.EnumDescriptorProto
	// contains filtered or unexported fields
}

EnumDescriptor describes an enum. If it's at top level, its parent will be nil. Otherwise it will be the descriptor of the message in which it is defined.

func (*EnumDescriptor) File 

func (c *EnumDescriptor) File() *FileDescriptor

func (*EnumDescriptor) GoImportPath 

func (c *EnumDescriptor) GoImportPath() GoImportPath

GoImportPath is the import path of the Go package containing the type.

func (*EnumDescriptor) TypeName 

func (e *EnumDescriptor) TypeName() (s []string)

TypeName returns the elements of the dotted type name. The package name is not part of this name.

type ExtensionDescriptor 

type ExtensionDescriptor struct {
	*descriptor.FieldDescriptorProto
	// contains filtered or unexported fields
}

ExtensionDescriptor describes an extension. If it's at top level, its parent will be nil. Otherwise it will be the descriptor of the message in which it is defined.

func (*ExtensionDescriptor) DescName 

func (e *ExtensionDescriptor) DescName() string

DescName returns the variable name used for the generated descriptor.

func (*ExtensionDescriptor) File 

func (c *ExtensionDescriptor) File() *FileDescriptor

func (*ExtensionDescriptor) GoImportPath 

func (c *ExtensionDescriptor) GoImportPath() GoImportPath

GoImportPath is the import path of the Go package containing the type.

func (*ExtensionDescriptor) TypeName 

func (e *ExtensionDescriptor) TypeName() (s []string)

TypeName returns the elements of the dotted type name. The package name is not part of this name.

type FieldDescriptor 

type FieldDescriptor struct {
	Proto *descriptor.FieldDescriptorProto

	Comments []*Comment
}

type FileDescriptor 

type FileDescriptor struct {
	*descriptor.FileDescriptorProto

	PackageName  GoPackageName // Name of this file's Go package.
	PackageAlias GoPackageName
	// contains filtered or unexported fields
}

FileDescriptor describes an protocol buffer descriptor file (.proto). It includes slices of all the messages and enums defined within it. Those slices are constructed by WrapTypes.

func (*FileDescriptor) Comments 

func (d *FileDescriptor) Comments() map[string]*descriptor.SourceCodeInfo_Location

func (*FileDescriptor) Enums added in v0.10.19 

func (d *FileDescriptor) Enums() []*EnumDescriptor

func (*FileDescriptor) Ext added in v0.10.19 

func (d *FileDescriptor) Ext() []*ExtensionDescriptor

func (*FileDescriptor) GoPackageName added in v0.10.19 

func (d *FileDescriptor) GoPackageName() string

func (*FileDescriptor) Messages 

func (d *FileDescriptor) Messages() []*MessageDescriptor

func (*FileDescriptor) TagServices 

func (d *FileDescriptor) TagServices() []*ServiceDescriptor

func (*FileDescriptor) VarName 

func (d *FileDescriptor) VarName() string

VarName is the variable name we'll use in the generated code to refer to the compressed bytes of this descriptor. It is not exported, so it is only valid inside the generated package.

type FileOutPut added in v0.10.7 

type FileOutPut struct {
	Package       string
	Out           string
	SourcePkgPath string
	Load          bool
}

type Generator 

type Generator struct {
	*bytes.Buffer

	Request  *plugin.CodeGeneratorRequest  // The input.
	Response *plugin.CodeGeneratorResponse // The output.

	ValidateEnable bool // Whether enable validate

	Param             map[string]string // Command-line parameters.
	PackageImportPath string            // Go import path of the package we're generating code for
	ImportPrefix      string            // String to prefix to imported package file names.
	ImportMap         map[string]string // Mapping from .proto file name to import path

	Pkg map[string]string // The names under which we import support packages

	OutPut *FileOutPut // What's file generate output
	// contains filtered or unexported fields
}

Generator is the type whose methods generate the output, stored in the associated response structure.

func New 

func New(name string) *Generator

New creates a new generator and allocates the request and response protobufs.

func (*Generator) AddImport 

func (g *Generator) AddImport(importPath GoImportPath) GoPackageName

AddImport adds a package to the generated file's import section. It returns the name used for the package.

func (*Generator) AllFiles added in v0.8.3 

func (g *Generator) AllFiles() []*FileDescriptor

func (*Generator) BuildFieldGoType added in v0.18.2 

func (g *Generator) BuildFieldGoType(file *FileDescriptor, field *descriptor.FieldDescriptorProto) string

BuildFieldGoType get go type

func (*Generator) BuildTypeNameMap 

func (g *Generator) BuildTypeNameMap()

BuildTypeNameMap builds the map from fully qualified type names to objects. The key names for the map come from the input data, which puts a period at the beginning. It should be called after SetPackageNames and before GenerateAllFiles.

func (*Generator) CommandLineParameters 

func (g *Generator) CommandLineParameters(parameter string)

CommandLineParameters breaks the comma-separated list of key=value pairs in the parameter (a member of the request protobuf) into a key/value map. It then sets file name mappings defined by those entries.

func (*Generator) Comments added in v0.10.19 

func (g *Generator) Comments(path string) string

Comments returns any comments from the source .proto file and empty string if comments not found. The path is a comma-separated list of intergers. See descriptor.proto for its format.

func (*Generator) DefaultPackageName 

func (g *Generator) DefaultPackageName(obj Object) string

DefaultPackageName returns the package name printed for the object. If its file is in a different package, it returns the package name we're using for this file, plus ".". Otherwise it returns the empty string.

func (*Generator) EnableEdit added in v0.14.2 

func (g *Generator) EnableEdit()

func (*Generator) Error 

func (g *Generator) Error(err error, msgs ...string)

Error reports a problem, including an error, and exits the program.

func (*Generator) ExtractEnum added in v0.10.20 

func (g *Generator) ExtractEnum(name string) *EnumDescriptor

ExtractEnum extract EnumDescriptor by name

func (*Generator) ExtractMessage added in v0.10.19 

func (g *Generator) ExtractMessage(name string) *MessageDescriptor

ExtractMessage extract MessageDescriptor by name

func (*Generator) Fail 

func (g *Generator) Fail(msgs ...string)

Fail reports a problem and exits the program.

func (*Generator) File added in v0.8.3 

func (g *Generator) File() *FileDescriptor

func (*Generator) GenerateAllFiles 

func (g *Generator) GenerateAllFiles()

GenerateAllFiles generates the output for all the files we're outputting.

func (*Generator) GenerateAllFilesWithOutPut added in v0.10.7 

func (g *Generator) GenerateAllFilesWithOutPut(put *FileOutPut)

GenerateAllFilesWithOutPut generates the output for all the files we're outputting and changes output path and package name.

func (*Generator) GenerateExtension added in v0.10.19 

func (g *Generator) GenerateExtension(ext *ExtensionDescriptor)

func (*Generator) GenerateFileDescriptor added in v0.10.19 

func (g *Generator) GenerateFileDescriptor(file *FileDescriptor)

func (*Generator) GenerateMessage added in v0.10.19 

func (g *Generator) GenerateMessage(message *MessageDescriptor)

GenerateMessage generates the type and default constant definitions for this Descriptor.

func (*Generator) GeneratePlugin added in v0.10.19 

func (g *Generator) GeneratePlugin(p Plugin)

func (*Generator) GetFieldName added in v0.10.19 

func (g *Generator) GetFieldName(message *Descriptor, f *FieldDescriptor) string

func (*Generator) GetMapKeyField added in v0.10.19 

func (g *Generator) GetMapKeyField(field, keyField *descriptor.FieldDescriptorProto) *descriptor.FieldDescriptorProto

func (*Generator) GetMapValueField added in v0.10.19 

func (g *Generator) GetMapValueField(field, valField *descriptor.FieldDescriptorProto) *descriptor.FieldDescriptorProto

func (*Generator) GetOneOfFieldName added in v0.10.19 

func (g *Generator) GetOneOfFieldName(msg *MessageDescriptor, f *FieldDescriptor) string

func (*Generator) GoMapType added in v0.10.19 

func (g *Generator) GoMapType(d *Descriptor, field *descriptor.FieldDescriptorProto) *GoMapDescriptor

func (*Generator) GoPackageName 

func (g *Generator) GoPackageName(importPath GoImportPath) GoPackageName

GoPackageName returns the name used for a package.

func (*Generator) GoType 

func (g *Generator) GoType(message *Descriptor, field *descriptor.FieldDescriptorProto) (typ string, wire string)

GoType returns a string representing the type name, and the wire type

func (*Generator) In 

func (g *Generator) In()

In Indents the output one tab stop.

func (*Generator) IsGroup added in v0.10.19 

func (g *Generator) IsGroup(field *descriptor.FieldDescriptorProto) bool

func (*Generator) IsMap added in v0.10.19 

func (g *Generator) IsMap(field *descriptor.FieldDescriptorProto) bool

func (*Generator) ObjectNamed 

func (g *Generator) ObjectNamed(typeName string) Object

ObjectNamed given a fully-qualified input type name as it appears in the input data, returns the descriptor for the message or enum with that name.

func (*Generator) OneOfTypeName added in v0.10.19 

func (g *Generator) OneOfTypeName(msg *MessageDescriptor, field *FieldDescriptor) string

func (*Generator) Out 

func (g *Generator) Out()

Out unindents the output one tab stop.

func (*Generator) P 

func (g *Generator) P(str ...interface{})

P prints the arguments to the generated output. It handles strings and int32s, plus handling indirections because they may be *string, etc. Any inputs of type AnnotatedAtoms may emit annotations in a .meta file in addition to outputting the atoms themselves (if g.annotateCode is true).

func (*Generator) PrintComments 

func (g *Generator) PrintComments(path string) bool

PrintComments prints any comments from the source .proto file. The path is a comma-separated list of integers. It returns an indication of whether any comments were printed. See descriptor.proto for its format.

func (*Generator) PrintImport added in v0.10.19 

func (g *Generator) PrintImport(alias GoPackageName, pkg GoImportPath)

func (*Generator) RecordTypeUse 

func (g *Generator) RecordTypeUse(t string)

func (*Generator) SetFile added in v0.10.19 

func (g *Generator) SetFile(filename string)

func (*Generator) SetPackageNames 

func (g *Generator) SetPackageNames()

SetPackageNames sets the package name for this run. The package name must agree across all files being generated. It also defines unique package names for all imported files.

func (*Generator) TypeName 

func (g *Generator) TypeName(obj Object) string

TypeName is the printed name appropriate for an item. If the object is in the current file, TypeName drops the package name and underscores the rest. Otherwise the object is from another package; and the result is the underscored package name followed by the item name. The result always has an initial capital.

func (*Generator) TypeNameByObject added in v0.10.19 

func (g *Generator) TypeNameByObject(typeName string) Object

func (*Generator) WrapTypes 

func (g *Generator) WrapTypes()

WrapTypes walks the incoming data, wrapping DescriptorProtos, EnumDescriptorProtos and FileDescriptorProtos into file-referenced objects within the Generator. It also creates the list of files to generate and so should be called before GenerateAllFiles.

type GoImportPath 

type GoImportPath string

A GoImportPath is the import path of a Go package. e.g., "google.golang.org/genproto/protobuf".

func (GoImportPath) String 

func (p GoImportPath) String() string

type GoMapDescriptor added in v0.10.19 

type GoMapDescriptor struct {
	GoType string

	KeyField      *descriptor.FieldDescriptorProto
	KeyAliasField *descriptor.FieldDescriptorProto
	KeyTag        string

	ValueField      *descriptor.FieldDescriptorProto
	ValueAliasField *descriptor.FieldDescriptorProto
	ValueTag        string
}

GoMapDescriptor is a full description of the map output struct.

type GoPackageName 

type GoPackageName string

A GoPackageName is the name of a Go package. e.g., "protobuf".

type ImportedDescriptor 

type ImportedDescriptor struct {
	// contains filtered or unexported fields
}

ImportedDescriptor describes a type that has been publicly imported from another file.

func (*ImportedDescriptor) File 

func (c *ImportedDescriptor) File() *FileDescriptor

func (*ImportedDescriptor) GoImportPath 

func (c *ImportedDescriptor) GoImportPath() GoImportPath

GoImportPath is the import path of the Go package containing the type.

func (*ImportedDescriptor) TypeName 

func (id *ImportedDescriptor) TypeName() []string

type MessageDescriptor 

type MessageDescriptor struct {
	Proto *Descriptor

	Fields []*FieldDescriptor

	Comments []*Comment
}

type MethodDescriptor 

type MethodDescriptor struct {
	Proto *descriptor.MethodDescriptorProto

	Comments []*Comment
}

type Object 

type Object interface {
	GoImportPath() GoImportPath
	TypeName() []string
	File() *FileDescriptor
}

Object is an interface abstracting the abilities shared by enums, messages, extensions and imported objects.

type Plugin 

type Plugin interface {
	// Name identifies the plugin.
	Name() string
	// Init is called once after data structures are built but before
	// code generation begins.
	Init(g *Generator)
	// Generate produces the code generated by the plugin for this file,
	// except for the imports, by calling the generator's methods P, In, and Out.
	Generate(file *FileDescriptor)
	// GenerateImports produces the import declarations for this file.
	// It is called after Generate.
	GenerateImports(file *FileDescriptor, imports map[GoImportPath]GoPackageName)
}

A Plugin provides functionality to add to the output during Go code generation, such as to produce RPC stubs.

type PluginImports added in v0.10.19 

type PluginImports interface {
	NewImport(pkg, alias string) Single
	GenerateImports(file *FileDescriptor, imports map[GoImportPath]GoPackageName)
}

type ServiceDescriptor 

type ServiceDescriptor struct {
	Proto *descriptor.ServiceDescriptorProto

	Methods []*MethodDescriptor

	Comments []*Comment
}

type Single added in v0.10.19 

type Single interface {
	Use() string
	IsUsed() bool
	Name() string
	Location() string
}

Source Files

View all Source files

Directories

Path Synopsis
internal
remap
Package remap handles tracking the locations of Go tokens in a source text across a rewrite by the Go formatter.
 Package remap handles tracking the locations of Go tokens in a source text across a rewrite by the Go formatter.
command

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL
go.dev uses cookies from Google to deliver and enhance the quality of its services and to analyze traffic. Learn more.