grpc重写插件，首次提交

drivers/plugins/grpc-proxy-rewrite/proxy-rewrite.go

@@ -0,0 +1,77 @@
+package grpc_proxy_rewrite
+
+import (
+	"strings"
+
+	grpc_context "github.com/eolinker/eosc/eocontext/grpc-context"
+
+	"github.com/eolinker/apinto/drivers"
+	"github.com/eolinker/eosc"
+	"github.com/eolinker/eosc/eocontext"
+	http_service "github.com/eolinker/eosc/eocontext/http-context"
+)
+
+var _ eocontext.IFilter = (*ProxyRewrite)(nil)
+var _ grpc_context.GrpcFilter = (*ProxyRewrite)(nil)
+
+var (
+	regexpErrInfo   = `[plugin proxy-rewrite2 config err] Compile regexp fail. err regexp: %s `
+	notMatchErrInfo = `[plugin proxy-rewrite2 err] Proxy path rewrite fail. Request path can't match any rewrite-path. request path: %s `
+)
+
+type ProxyRewrite struct {
+	drivers.WorkerBase
+
+	host            string
+	headers         map[string]string
+	tls             bool
+	skipCertificate bool
+}
+
+func (p *ProxyRewrite) DoFilter(ctx eocontext.EoContext, next eocontext.IChain) (err error) {
+	return grpc_context.DoGrpcFilter(p, ctx, next)
+}
+
+func (p *ProxyRewrite) DoGrpcFilter(ctx grpc_context.IGrpcContext, next eocontext.IChain) (err error) {
+	if p.host != "" {
+		ctx.Proxy().SetHost(p.host)
+	}
+	ctx.EnableTls(p.tls)
+	ctx.InsecureCertificateVerify(p.skipCertificate)
+	for key, value := range p.headers {
+		ctx.Proxy().Headers().Set(key, value)
+	}
+	if next != nil {
+		return next.DoChain(ctx)
+	}
+	return nil
+}
+
+func (p *ProxyRewrite) Start() error {
+	return nil
+}
+
+func (p *ProxyRewrite) Reset(v interface{}, workers map[eosc.RequireId]eosc.IWorker) error {
+	conf, err := check(v)
+	if err != nil {
+		return err
+	}
+	p.skipCertificate = conf.SkipCertificate
+	p.headers = conf.Headers
+	p.tls = conf.Tls
+	p.host = strings.TrimSpace(conf.Authority)
+	return nil
+}
+
+func (p *ProxyRewrite) Stop() error {
+	return nil
+}
+
+func (p *ProxyRewrite) Destroy() {
+
+	p.headers = nil
+}
+
+func (p *ProxyRewrite) CheckSkill(skill string) bool {
+	return http_service.FilterSkillName == skill
+}