office

internal/api/office.go

@@ -0,0 +1,25 @@
+package api
+
+import (
+	"context"
+	"github.com/OpenIMSDK/Open-IM-Server/pkg/common/config"
+	"github.com/OpenIMSDK/Open-IM-Server/pkg/discoveryregistry"
+	"github.com/OpenIMSDK/Open-IM-Server/pkg/proto/office"
+	"google.golang.org/grpc"
+)
+
+func NewOffice(c discoveryregistry.SvcDiscoveryRegistry) *Office {
+	conn, err := c.GetConn(context.Background(), config.Config.RpcRegisterName.OpenImOfficeName)
+	if err != nil {
+		panic(err)
+	}
+	return &Office{conn: conn}
+}
+
+type Office struct {
+	conn *grpc.ClientConn
+}
+
+func (o *Office) client(ctx context.Context) (office.OfficeClient, error) {
+	return office.NewOfficeClient(o.conn), nil
+}