1 // Copyright 2020 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 // See the License for the specific language governing permissions and 12 // limitations under the License. 13 14 package client 15 16 import ( 17 "google.golang.org/grpc" 18 19 chaosdaemon "github.com/chaos-mesh/chaos-mesh/pkg/chaosdaemon/pb" 20 ) 21 22 // ChaosDaemonClientInterface represents the ChaosDaemonClient, it's used to simply unit test 23 type ChaosDaemonClientInterface interface { 24 chaosdaemon.ChaosDaemonClient 25 Close() error 26 } 27 28 // GrpcChaosDaemonClient would act like chaosdaemon.ChaosDaemonClient with a Close method 29 type GrpcChaosDaemonClient struct { 30 chaosdaemon.ChaosDaemonClient 31 conn *grpc.ClientConn 32 } 33 34 func (c *GrpcChaosDaemonClient) Close() error { 35 return c.conn.Close() 36 } 37 38 func New(conn *grpc.ClientConn) ChaosDaemonClientInterface { 39 return &GrpcChaosDaemonClient{ 40 ChaosDaemonClient: chaosdaemon.NewChaosDaemonClient(conn), 41 conn: conn, 42 } 43 } 44