diff options
author | Christopher Speller <crspeller@gmail.com> | 2018-06-25 12:33:13 -0700 |
---|---|---|
committer | GitHub <noreply@github.com> | 2018-06-25 12:33:13 -0700 |
commit | 1e5c432e1029601a664454388ae366ef69618d62 (patch) | |
tree | cb9e8bfb66640ac3b29c934bb2c3202d25aeb368 /plugin/http.go | |
parent | ecefa6cdd1e7376046bbec82c1b47f7756fea646 (diff) | |
download | chat-1e5c432e1029601a664454388ae366ef69618d62.tar.gz chat-1e5c432e1029601a664454388ae366ef69618d62.tar.bz2 chat-1e5c432e1029601a664454388ae366ef69618d62.zip |
MM-10702 Moving plugins to use hashicorp go-plugin. (#8978)
* Moving plugins to use hashicorp go-plugin.
* Tweaks from feedback.
Diffstat (limited to 'plugin/http.go')
-rw-r--r-- | plugin/http.go | 91 |
1 files changed, 91 insertions, 0 deletions
diff --git a/plugin/http.go b/plugin/http.go new file mode 100644 index 000000000..5faf8f08a --- /dev/null +++ b/plugin/http.go @@ -0,0 +1,91 @@ +// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved. +// See LICENSE.txt for license information. + +package plugin + +import ( + "io" + "net/http" + "net/rpc" +) + +type HTTPResponseWriterRPCServer struct { + w http.ResponseWriter +} + +func (w *HTTPResponseWriterRPCServer) Header(args struct{}, reply *http.Header) error { + *reply = w.w.Header() + return nil +} + +func (w *HTTPResponseWriterRPCServer) Write(args []byte, reply *struct{}) error { + _, err := w.w.Write(args) + return err +} + +func (w *HTTPResponseWriterRPCServer) WriteHeader(args int, reply *struct{}) error { + w.w.WriteHeader(args) + return nil +} + +func (w *HTTPResponseWriterRPCServer) SyncHeader(args http.Header, reply *struct{}) error { + dest := w.w.Header() + for k := range dest { + if _, ok := args[k]; !ok { + delete(dest, k) + } + } + for k, v := range args { + dest[k] = v + } + return nil +} + +func ServeHTTPResponseWriter(w http.ResponseWriter, conn io.ReadWriteCloser) { + server := rpc.NewServer() + server.Register(&HTTPResponseWriterRPCServer{ + w: w, + }) + server.ServeConn(conn) +} + +type HTTPResponseWriterRPCClient struct { + client *rpc.Client + header http.Header +} + +var _ http.ResponseWriter = (*HTTPResponseWriterRPCClient)(nil) + +func (w *HTTPResponseWriterRPCClient) Header() http.Header { + if w.header == nil { + w.client.Call("Plugin.Header", struct{}{}, &w.header) + } + return w.header +} + +func (w *HTTPResponseWriterRPCClient) Write(b []byte) (int, error) { + if err := w.client.Call("Plugin.SyncHeader", w.header, nil); err != nil { + return 0, err + } + if err := w.client.Call("Plugin.Write", b, nil); err != nil { + return 0, err + } + return len(b), nil +} + +func (w *HTTPResponseWriterRPCClient) WriteHeader(statusCode int) { + if err := w.client.Call("Plugin.SyncHeader", w.header, nil); err != nil { + return + } + w.client.Call("Plugin.WriteHeader", statusCode, nil) +} + +func (h *HTTPResponseWriterRPCClient) Close() error { + return h.client.Close() +} + +func ConnectHTTPResponseWriter(conn io.ReadWriteCloser) *HTTPResponseWriterRPCClient { + return &HTTPResponseWriterRPCClient{ + client: rpc.NewClient(conn), + } +} |