1 // Copyright 2021 Chaos Mesh Authors. 2 // 3 // Licensed under the Apache License, Version 2.0 (the "License"); 4 // you may not use this file except in compliance with the License. 5 // You may obtain a copy of the License at 6 // 7 // http://www.apache.org/licenses/LICENSE-2.0 8 // 9 // Unless required by applicable law or agreed to in writing, software 10 // distributed under the License is distributed on an "AS IS" BASIS, 11 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 12 // See the License for the specific language governing permissions and 13 // limitations under the License. 14 // 15 16 package client 17 18 import ( 19 "google.golang.org/grpc" 20 21 chaosdaemon "github.com/chaos-mesh/chaos-mesh/pkg/chaosdaemon/pb" 22 ) 23 24 // ChaosDaemonClientInterface represents the ChaosDaemonClient, it's used to simply unit test 25 type ChaosDaemonClientInterface interface { 26 chaosdaemon.ChaosDaemonClient 27 Close() error 28 } 29 30 // GrpcChaosDaemonClient would act like chaosdaemon.ChaosDaemonClient with a Close method 31 type GrpcChaosDaemonClient struct { 32 chaosdaemon.ChaosDaemonClient 33 conn *grpc.ClientConn 34 } 35 36 func (c *GrpcChaosDaemonClient) Close() error { 37 return c.conn.Close() 38 } 39 40 func New(conn *grpc.ClientConn) ChaosDaemonClientInterface { 41 return &GrpcChaosDaemonClient{ 42 ChaosDaemonClient: chaosdaemon.NewChaosDaemonClient(conn), 43 conn: conn, 44 } 45 } 46