davyxu / cellmesh Goto Github PK

import (
	"fmt"
	"reflect"	
	"github.com/davyxu/cellnet"
	"github.com/davyxu/cellnet/codec"
	"github.com/davyxu/cellmesh/service"
	{{if HasJsonCodec $}}_ "github.com/davyxu/cellnet/codec/json"{{end}}
	{{if HasGogopbCodec $}}"github.com/davyxu/cellmesh/demo/proto/pb"{{end}}
)

...
{{range .Structs}}
{{ if StructCodecNotPb . }}
{{ObjectLeadingComment .}}
type {{.Name}} struct{	{{range .Fields}}
	{{GoFieldName .}} {{GoTypeName .}} {{GoStructTag .}}{{FieldTrailingComment .}} {{end}}
}
{{end}}
{{end}}

{{range .Structs}}{{ if StructCodecNotPb . }}
func (self *{{.Name}}) String() string { return fmt.Sprintf("%+v",*self) } {{end}} {{end}}

...

{{ if StructCodecNotPb . }}
			case *{{.Name}}:
				Handle_{{ExportSymbolName $svcName}}_{{.Name}}(ev)
			{{else}}
			case *pb.{{.Name}}:
				Handle_{{ExportSymbolName $svcName}}_{{.Name}}(ev) 
			{{end}}


...
func init() {
	{{range $index, $ele := .Structs}} {{ if IsMessage $ele }}{{ if StructCodecNotPb $ele }}
	cellnet.RegisterMessageMeta(&cellnet.MessageMeta{
		Codec: codec.MustGetCodec("{{StructCodec $ele}}"),
		Type:  reflect.TypeOf((*{{$ele.Name}})(nil)).Elem(),
		ID:    {{StructMsgID $ele}},
	})
	{{end}} {{end}} {{end}}
}

FuncMap["StructCodecNotPb"] = func(d *model.Descriptor) bool {
		if  d.TagValueString("Codec") == "gogopb" {
			return false
		}else {
			return true
		}
	}

	FuncMap["HasGogopbCodec"] = func(ctx *gen.Context) bool {

		for _, d := range ctx.Structs() {
			if d.TagValueString("Codec") == "gogopb" {
				return true
			}
		}

		return true
	}