From 7d46f7178db663ce337060fb410138a40b6f42d3 Mon Sep 17 00:00:00 2001 From: "david.yan" Date: Fri, 28 Mar 2025 20:54:27 +0800 Subject: [PATCH] initial --- go.mod | 5 ++ go.sum | 6 ++ main.go | 202 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++ 3 files changed, 213 insertions(+) create mode 100644 go.mod create mode 100644 go.sum create mode 100644 main.go diff --git a/go.mod b/go.mod new file mode 100644 index 0000000..3630a92 --- /dev/null +++ b/go.mod @@ -0,0 +1,5 @@ +module protoc-gen-slc + +go 1.24.0 + +require google.golang.org/protobuf v1.36.6 diff --git a/go.sum b/go.sum new file mode 100644 index 0000000..4665b5a --- /dev/null +++ b/go.sum @@ -0,0 +1,6 @@ +github.com/google/go-cmp v0.5.5 h1:Khx7svrCpmxxtHBq5j2mp/xVjsi8hQMfNLvJFAlrGgU= +github.com/google/go-cmp v0.5.5/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE= +golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543 h1:E7g+9GITq07hpfrRu66IVDexMakfv52eLZ2CXBWiKr4= +golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0= +google.golang.org/protobuf v1.36.6 h1:z1NpPI8ku2WgiWnf+t9wTPsn6eP1L7ksHUlkfLvd9xY= +google.golang.org/protobuf v1.36.6/go.mod h1:jduwjTPXsFjZGTmRluh+L6NjiWu7pchiJ2/5YcXBHnY= diff --git a/main.go b/main.go new file mode 100644 index 0000000..efaf5aa --- /dev/null +++ b/main.go @@ -0,0 +1,202 @@ +package main + +import ( + "fmt" + "strings" + + "google.golang.org/protobuf/compiler/protogen" + "google.golang.org/protobuf/types/pluginpb" +) + +func main() { + protogen.Options{}.Run(func(gen *protogen.Plugin) error { + gen.SupportedFeatures = uint64(pluginpb.CodeGeneratorResponse_FEATURE_PROTO3_OPTIONAL) + for _, f := range gen.Files { + if len(f.Services) == 0 { + continue + } + if err := generateFiles(gen, f); err != nil { + return err + } + } + return nil + }) +} + +func generateFiles(gen *protogen.Plugin, file *protogen.File) error { + for _, service := range file.Services { + // Generate server file + if err := generateServerFile(gen, file, service); err != nil { + return err + } + + // Generate client file + if err := generateClientFile(gen, file, service); err != nil { + return err + } + + // Generate logic file + if err := generateLogicFile(gen, file, service); err != nil { + return err + } + } + return nil +} + +func generateServerFile(gen *protogen.Plugin, file *protogen.File, service *protogen.Service) error { + filename := fmt.Sprintf("%s_server.pb.go", strings.ToLower(service.GoName)) + g := gen.NewGeneratedFile(filename, file.GoImportPath) + + // Package declaration + g.P("// Code generated by protoc-gen-layered. DO NOT EDIT.") + g.P() + g.P("package ", file.GoPackageName) + g.P() + + // Imports + g.P("import (") + g.P("\t\"context\"") + g.P("\t\"errors\"") + g.P() + g.P("\t\"google.golang.org/grpc\"") + g.P("\t\"google.golang.org/grpc/codes\"") + g.P("\t\"google.golang.org/grpc/status\"") + g.P(")") + g.P() + + // Server struct + g.P("type ", service.GoName, "Server struct {") + g.P("\tUnimplemented", service.GoName, "Server") + g.P("\tlogic *", service.GoName, "Logic") + g.P("}") + g.P() + + // NewServer function + g.P("func New", service.GoName, "Server(logic *", service.GoName, "Logic) *", service.GoName, "Server {") + g.P("\treturn &", service.GoName, "Server{logic: logic}") + g.P("}") + g.P() + + // Register function + g.P("func Register", service.GoName, "Server(s *grpc.Server, logic *", service.GoName, "Logic) {") + g.P("\tserver := New", service.GoName, "Server(logic)") + g.P("\tRegister", service.GoName, "Server(s, server)") + g.P("}") + g.P() + + // Service methods + for _, method := range service.Methods { + g.P("func (s *", service.GoName, "Server) ", methodSignature(g, method), " {") + g.P("\t// Add your server-side logic here") + g.P("\tresp, err := s.logic.", method.GoName, "(ctx, req)") + g.P("\tif err != nil {") + g.P("\t\treturn nil, status.Errorf(codes.Internal, \"%v\", err)") + g.P("\t}") + g.P("\treturn resp, nil") + g.P("}") + g.P() + } + + return nil +} + +func generateClientFile(gen *protogen.Plugin, file *protogen.File, service *protogen.Service) error { + filename := fmt.Sprintf("%s_client.pb.go", strings.ToLower(service.GoName)) + g := gen.NewGeneratedFile(filename, file.GoImportPath) + + // Package declaration + g.P("// Code generated by protoc-gen-layered. DO NOT EDIT.") + g.P() + g.P("package ", file.GoPackageName) + g.P() + + // Imports + g.P("import (") + g.P("\t\"context\"") + g.P() + g.P("\t\"google.golang.org/grpc\"") + g.P(")") + g.P() + + // Client struct + g.P("type ", service.GoName, "Client struct {") + g.P("\tcc grpc.ClientConnInterface") + g.P("}") + g.P() + + // NewClient function + g.P("func New", service.GoName, "Client(cc grpc.ClientConnInterface) *", service.GoName, "Client {") + g.P("\treturn &", service.GoName, "Client{cc}") + g.P("}") + g.P() + + // Client methods + for _, method := range service.Methods { + g.P("func (c *", service.GoName, "Client) ", methodSignature(g, method), " {") + g.P("\tout := new(", method.Output.GoIdent, ")") + g.P("\terr := c.cc.Invoke(ctx, \"", fullMethodName(file, service, method), "\", req, out)") + g.P("\tif err != nil {") + g.P("\t\treturn nil, err") + g.P("\t}") + g.P("\treturn out, nil") + g.P("}") + g.P() + } + + return nil +} + +func generateLogicFile(gen *protogen.Plugin, file *protogen.File, service *protogen.Service) error { + filename := fmt.Sprintf("%s_logic.pb.go", strings.ToLower(service.GoName)) + g := gen.NewGeneratedFile(filename, file.GoImportPath) + + // Package declaration + g.P("// Code generated by protoc-gen-layered. DO NOT EDIT.") + g.P() + g.P("package ", file.GoPackageName) + g.P() + + // Imports + g.P("import (") + g.P("\t\"context\"") + g.P("\t\"errors\"") + g.P(")") + g.P() + + // Logic struct + g.P("type ", service.GoName, "Logic struct {") + g.P("\t// Add your dependencies here") + g.P("}") + g.P() + + // NewLogic function + g.P("func New", service.GoName, "Logic() *", service.GoName, "Logic {") + g.P("\treturn &", service.GoName, "Logic{}") + g.P("}") + g.P() + + // Logic methods + for _, method := range service.Methods { + g.P("func (l *", service.GoName, "Logic) ", method.GoName, "(ctx context.Context, req *", method.Input.GoIdent, ") (*", method.Output.GoIdent, ", error) {") + g.P("\t// Implement your business logic here") + g.P("\treturn nil, errors.New(\"not implemented\")") + g.P("}") + g.P() + } + + return nil +} + +func methodSignature(g *protogen.GeneratedFile, method *protogen.Method) string { + return fmt.Sprintf("%s(ctx context.Context, req *%s) (*%s, error)", + method.GoName, + method.Input.GoIdent, + method.Output.GoIdent) +} + +func fullMethodName(file *protogen.File, service *protogen.Service, method *protogen.Method) string { + return fmt.Sprintf("/%s.%s/%s", + file.Proto.GetPackage(), + service.GoName, + method.GoName) +}