Add SetLogger for setting global logger

[govpp.git] / adapter / socketclient / socketclient.go

diff --git a/adapter/socketclient/socketclient.go b/adapter/socketclient/socketclient.go
index b2c5d47..637bd69 100644 (file)
--- a/adapter/socketclient/socketclient.go
+++ b/adapter/socketclient/socketclient.go
@@ -19,7 +19,6 @@ import (
        "encoding/binary"
        "errors"
        "fmt"
-       "git.fd.io/govpp.git/adapter/socketclient/binapi/memclnt"
        "io"
        "net"
        "os"
@@ -32,6 +31,7 @@ import (
        "github.com/sirupsen/logrus"
 
        "git.fd.io/govpp.git/adapter"
+       "git.fd.io/govpp.git/binapi/memclnt"
        "git.fd.io/govpp.git/codec"
 )
 
@@ -56,16 +56,21 @@ var (
        debug       = strings.Contains(os.Getenv("DEBUG_GOVPP"), "socketclient")
        debugMsgIds = strings.Contains(os.Getenv("DEBUG_GOVPP"), "msgtable")
 
-       logger = logrus.New()
-       log    = logger.WithField("logger", "govpp/socketclient")
+       log logrus.FieldLogger
 )
 
-// init initializes global logger
+// SetLogger sets global logger.
+func SetLogger(logger logrus.FieldLogger) {
+       log = logger
+}
+
 func init() {
+       logger := logrus.New()
        if debug {
                logger.Level = logrus.DebugLevel
-               log.Debug("govpp: debug level enabled for socketclient")
+               logger.Debug("govpp: debug level enabled for socketclient")
        }
+       log = logger.WithField("logger", "govpp/socketclient")
 }
 
 const socketMissing = `
@@ -266,9 +271,8 @@ func (c *socketClient) Disconnect() error {
        // wait for readerLoop to return
        c.wg.Wait()
 
-       if err := c.close(); err != nil {
-               log.Debugf("closing failed: %v", err)
-       }
+       // Don't bother sending a vl_api_sockclnt_delete_t message,
+       // just close the socket.
 
        if err := c.disconnect(); err != nil {
                return err
@@ -375,48 +379,6 @@ func (c *socketClient) open() error {
        return nil
 }
 
-func (c *socketClient) close() error {
-       var msgCodec = codec.DefaultCodec
-
-       req := &memclnt.SockclntDelete{
-               Index: c.clientIndex,
-       }
-       msg, err := msgCodec.EncodeMsg(req, c.sockDelMsgId)
-       if err != nil {
-               log.Debugln("Encode error:", err)
-               return err
-       }
-       // set non-0 context
-       msg[5] = deleteMsgContext
-
-       log.Debugf("sending socklntDel (%d bytes): % 0X", len(msg), msg)
-
-       if err := c.writeMsg(msg); err != nil {
-               log.Debugln("Write error: ", err)
-               return err
-       }
-
-       msgReply, err := c.readMsgTimeout(nil, c.disconnectTimeout)
-       if err != nil {
-               if nerr, ok := err.(net.Error); ok && nerr.Timeout() {
-                       // we accept timeout for reply
-                       return nil
-               }
-               log.Debugln("Read error:", err)
-               return err
-       }
-
-       reply := new(memclnt.SockclntDeleteReply)
-       if err := msgCodec.DecodeMsg(msgReply, reply); err != nil {
-               log.Debugln("Decoding sockclnt_delete_reply failed:", err)
-               return err
-       } else if reply.Response != 0 {
-               return fmt.Errorf("sockclnt_delete_reply: response error (%d)", reply.Response)
-       }
-
-       return nil
-}
-
 func (c *socketClient) GetMsgID(msgName string, msgCrc string) (uint16, error) {
        if msgID, ok := c.msgTable[msgName+"_"+msgCrc]; ok {
                return msgID, nil