protoc-gen-kit/example/main.go

package main

import (
	"context"
	"flag"
	"fmt"
	"git.diulo.com/mogfee/kit"
	user "git.diulo.com/mogfee/kit/api"
	"git.diulo.com/mogfee/kit/example/service"
	"git.diulo.com/mogfee/kit/middleware/jwt"
	"git.diulo.com/mogfee/kit/middleware/logging"
	"git.diulo.com/mogfee/kit/middleware/validate"
	"git.diulo.com/mogfee/kit/registry/etcd"
	"git.diulo.com/mogfee/kit/transport/http"
	clientv3 "go.etcd.io/etcd/client/v3"
)

var host string

func init() {
	flag.StringVar(&host, "h", "localhost:9093", "")
}
func main() {
	flag.Parse()
	runApp(host)
}
func runApp(host string) {
	hs := http.NewServer(
		http.Address(host),
		http.Middleware(
			logging.Server(),
			validate.Server(),
			jwt.JWT(),
		),
	)
	//route := hs.Route("/")
	//route.GET("/", func(ctx http.Context) error {
	//	in := UserAddRequest{Name: "tom"}
	//	http.SetOperation(ctx, "/api/abc")
	//	h := ctx.Middleware(func(ctx context.Context, a any) (any, error) {
	//		return AddUser(ctx, a.(*UserAddRequest))
	//	})
	//	if tr, ok := transport.FromServerContext(ctx); ok {
	//		fmt.Println(tr.Operation())
	//	}
	//	out, err := h(ctx, &in)
	//	if err != nil {
	//		return err
	//	}
	//	reply, _ := out.(*UserAddResponse)
	//	reply.Id = host
	//	return ctx.Result(200, reply)
	//})

	user.RegisterUserHTTPServer(hs, &service.UserService{})
	client, err := clientv3.New(clientv3.Config{
		Endpoints: []string{"127.0.0.1:2379"},
	})
	if err != nil {
		panic(err)
	}
	app := kit.New(
		kit.Name("kit-server"),
		kit.Version("v1.0"),
		kit.ID(host),
		kit.Server(hs),
		kit.Registrar(etcd.New(client)),
	)
	fmt.Println(app.Run())
	fmt.Println(app.Stop())
}

type UserAddRequest struct {
	Name string
}
type UserAddResponse struct {
	Id string
}

func AddUser(ctx context.Context, request *UserAddRequest) (*UserAddResponse, error) {
	fmt.Println(jwt.FromUserContext(ctx))
	fmt.Println(jwt.FromAuthKeyContext(ctx))
	return &UserAddResponse{Id: request.Name}, nil
	//errors.New(500, "xx", "")
}