mirror of
https://github.com/OpenListTeam/OpenList.git
synced 2025-09-20 12:46:17 +08:00
38 lines
828 B
Go
38 lines
828 B
Go
|
package driver
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
|
||
|
|
"github.com/OpenListTeam/OpenList/v5/proto/driver"
|
||
|
|
"github.com/hashicorp/go-plugin"
|
||
|
|
"google.golang.org/grpc"
|
||
|
|
)
|
||
|
|
|
||
|
|
func GetHandshakeConfig() plugin.HandshakeConfig {
|
||
|
|
return plugin.HandshakeConfig{
|
||
|
|
ProtocolVersion: 2,
|
||
|
|
MagicCookieKey: "OpenListTeam",
|
||
|
|
MagicCookieValue: "openlist",
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
type PluginIF interface {
|
||
|
|
Config(ctx context.Context) (Config, error)
|
||
|
|
}
|
||
|
|
|
||
|
|
type Plugin struct {
|
||
|
|
plugin.Plugin
|
||
|
|
Impl PluginIF
|
||
|
|
}
|
||
|
|
|
||
|
|
func (p *Plugin) GRPCServer(broker *plugin.GRPCBroker, s *grpc.Server) error {
|
||
|
|
driver.RegisterDriverServer(s, &GRPCServer{Impl: p.Impl})
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (p *Plugin) GRPCClient(ctx context.Context, broker *plugin.GRPCBroker, c *grpc.ClientConn) (any, error) {
|
||
|
|
return &GRPCClient{client: driver.NewDriverClient(c)}, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
var _ plugin.GRPCPlugin = (*Plugin)(nil)
|