fix(v1.0): fix replace

cuisongliu · Mar 24, 2020 · 6eac7d9 · 6eac7d9
1 parent bfb1f5a
commit 6eac7d9
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/pkg/sshutil/ssh.go b/pkg/sshutil/ssh.go
@@ -1,8 +1,11 @@
 package sshutil
 
-import "github.com/wonderivan/logger"
+import (
+	"github.com/wonderivan/logger"
+	"strings"
+)
 
-//Cmd is
+//Cmd is in host exec cmd
 func (ss *SSH) Cmd(host string, cmd string) []byte {
 	logger.Info("[%s]exec cmd is : %s", host, cmd)
 	session, err := ss.Connect(host)
@@ -29,9 +32,12 @@ func (ss *SSH) Cmd(host string, cmd string) []byte {
 	return b
 }
 
-func (ss *SSH) CmdToString(host, cmd string) string {
+//CmdToString is in host exec cmd and replace to spilt str
+func (ss *SSH) CmdToString(host, cmd, spilt string) string {
 	data := ss.Cmd(host, cmd)
 	if data != nil {
+		str := string(data)
+		str = strings.ReplaceAll(str, "\r\n", spilt)
 		return string(data)
 	}
 	return ""