Merge pull request #133 from jumpserver/dev_bugfix

Dev bugfix

pkg/common/client.go

@@ -28,7 +28,7 @@ type Client struct {
 	BaseHost   string
 	basicAuth  []string
 	cookie     map[string]string
-	http       *http.Client
+	http       http.Client
 	UrlParsers []UrlParser
 }
 
@@ -36,11 +36,12 @@ type UrlParser interface {
 	parse(url string, params ...map[string]string) string
 }
 
-func NewClient(timeout time.Duration, baseHost string) *Client {
+func NewClient(timeout time.Duration, baseHost string) Client {
 	headers := make(map[string]string, 0)
-	client := new(http.Client)
-	client.Timeout = timeout * time.Second
-	return &Client{
+	client := http.Client{
+		Timeout: timeout * time.Second,
+	}
+	return Client{
 		BaseHost: baseHost,
 		Timeout:  timeout * time.Second,
 		Headers:  headers,

pkg/common/table.go

@@ -148,14 +148,14 @@ func (t *WrapperTable) convertDataToSlice() [][]string {
 				switch t.TruncPolicy {
 				case TruncSuffix:
 					row[m] = fmt.Sprintf("%s...",
-						GetValidString(j[n],columSize-3, true))
+						GetValidString(j[n], columSize-3, true))
 				case TruncPrefix:
 					row[m] = fmt.Sprintf("...%s",
-						GetValidString(j[n],len(j[n])-columSize-3, false))
+						GetValidString(j[n], len(j[n])-columSize-3, false))
 				case TruncMiddle:
 					midValue := (columSize - 3) / 2
-					row[m] = fmt.Sprintf("%s...%s",GetValidString(j[n],midValue, true),
-						GetValidString(j[n],len(j[n])-midValue, false))
+					row[m] = fmt.Sprintf("%s...%s", GetValidString(j[n], midValue, true),
+						GetValidString(j[n], len(j[n])-midValue, false))
 				}
 			}
 
@@ -168,8 +168,8 @@ func (t *WrapperTable) convertDataToSlice() [][]string {
 func (t *WrapperTable) Display() string {
 	t.CalculateColumnsSize()
 
-	tableString := &strings.Builder{}
-	table := tablewriter.NewWriter(tableString)
+	tableString := strings.Builder{}
+	table := tablewriter.NewWriter(&tableString)
 	table.SetBorder(false)
 	table.SetHeader(t.Labels)
 	colors := make([]tablewriter.Colors, len(t.Fields))
@@ -193,23 +193,23 @@ func (t *WrapperTable) Display() string {
 	return tableString.String()
 }
 
-func GetValidString(s string, position int, positive bool) string{
+func GetValidString(s string, position int, positive bool) string {
 	step := 1
 	if positive {
 		step = -1
 	}
-	for position >=0 && position <= len(s) {
+	for position >= 0 && position <= len(s) {
 		switch positive {
 		case true:
-			if utf8.ValidString(s[:position]){
+			if utf8.ValidString(s[:position]) {
 				return s[:position]
 			}
 		case false:
-			if utf8.ValidString(s[position:]){
+			if utf8.ValidString(s[position:]) {
 				return s[position:]
 			}
 		}
 		position += step
 	}
 	return ""
-}
+}
\ No newline at end of file

pkg/common/utils.go

@@ -28,10 +28,12 @@ func GzipCompressFile(srcPath, dstPath string) error {
 	if err != nil {
 		return err
 	}
+	defer sf.Close()
 	df, err := os.Create(dstPath)
 	if err != nil {
 		return err
 	}
+	defer df.Close()
 	writer := gzip.NewWriter(df)
 	writer.Name = dstPath
 	writer.ModTime = time.Now().UTC()
@@ -60,6 +62,6 @@ func Abs(x int) int {
 	return x
 }
 
-func CurrentUTCTime()string{
+func CurrentUTCTime() string {
 	return time.Now().UTC().Format("2006-01-02 15:04:05 +0000")
-}
+}
\ No newline at end of file

pkg/config/config.go

@@ -172,18 +172,14 @@ var Conf = &Config{
 	ZipTmpPath:         "/tmp",
 }
 
-func SetConf(conf *Config) {
+func SetConf(conf Config) {
 	lock.Lock()
 	defer lock.Unlock()
-	Conf = conf
+	Conf = &conf
 }
 
-func GetConf() *Config {
+func GetConf() Config {
 	lock.RLock()
 	defer lock.RUnlock()
-	var conf Config
-	if confBytes, err := json.Marshal(Conf); err == nil {
-		_ = json.Unmarshal(confBytes, &conf)
-	}
-	return &conf
+	return *Conf
 }

pkg/handler/pagination.go

@@ -13,8 +13,8 @@ import (
 	"github.com/jumpserver/koko/pkg/utils"
 )
 
-func NewAssetPagination(term *utils.Terminal, assets []model.Asset) *AssetPagination {
-	assetPage := &AssetPagination{term: term, assets: assets}
+func NewAssetPagination(term *utils.Terminal, assets []model.Asset) AssetPagination {
+	assetPage := AssetPagination{term: term, assets: assets}
 	assetPage.Initial()
 	return assetPage
 }
@@ -181,8 +181,8 @@ func (p *AssetPagination) displayTipsInfo() {
 
 }
 
-func NewUserPagination(term *utils.Terminal, uid, search string, policy bool) *UserAssetPagination {
-	return &UserAssetPagination{
+func NewUserPagination(term *utils.Terminal, uid, search string, policy bool) UserAssetPagination {
+	return UserAssetPagination{
 		UserID:        uid,
 		offset:        0,
 		limit:         0,
@@ -269,7 +269,7 @@ func (p *UserAssetPagination) Start() []model.Asset {
 
 func (p *UserAssetPagination) displayPageAssets() {
 	if len(p.Data.Data) == 0 {
-		_, _ = p.term.Write([]byte(getI18nFromMap("NoAssets")+"\n\r"))
+		_, _ = p.term.Write([]byte(getI18nFromMap("NoAssets") + "\n\r"))
 		return
 	}
 

pkg/handler/wrappersession.go

@@ -43,6 +43,7 @@ func (w *WrapperSession) readLoop() {
 	}
 	w.mux.RLock()
 	_ = w.inWriter.Close()
+	_ = w.outReader.Close()
 	w.mux.RUnlock()
 	close(w.closed)
 	logger.Infof("Request %s: Read loop break", w.Uuid)

pkg/proxy/parser.go

@@ -31,8 +31,8 @@ const (
 	CommandOutputParserName = "Command Output parser"
 )
 
-func newParser(sid string) *Parser {
-	parser := &Parser{id: sid}
+func newParser(sid string) Parser {
+	parser := Parser{id: sid}
 	parser.initial()
 	return parser
 }
@@ -55,8 +55,8 @@ type Parser struct {
 
 	command         string
 	output          string
-	cmdInputParser  *CmdParser
-	cmdOutputParser *CmdParser
+	cmdInputParser  CmdParser
+	cmdOutputParser CmdParser
 
 	cmdFilterRules []model.SystemUserFilterRule
 	closed         chan struct{}

pkg/proxy/parsercmd.go

@@ -12,8 +12,8 @@ import (
 
 var ps1Pattern = regexp.MustCompile(`^\[?.*@.*\]?[\\$#]\s|mysql>\s`)
 
-func NewCmdParser(sid, name string) *CmdParser {
-	parser := &CmdParser{id: sid, name:name}
+func NewCmdParser(sid, name string) CmdParser {
+	parser := CmdParser{id: sid, name:name}
 	parser.initial()
 	return parser
 }
@@ -67,16 +67,16 @@ func (cp *CmdParser) initial() {
 	go func() {
 		logger.Infof("Session %s: %s start", cp.id, cp.name)
 		defer logger.Infof("Session %s: %s parser close", cp.id, cp.name)
-	inloop:
+	loop:
 		for {
 			line, err := cp.term.ReadLine()
 			if err != nil {
 				select {
 				case <-cp.closed:
-					goto outloop
+					break loop
 				default:
 				}
-				goto inloop
+				goto loop
 			}
 			cp.lock.Lock()
 			cp.currentLength += len(line)
@@ -85,7 +85,6 @@ func (cp *CmdParser) initial() {
 			}
 			cp.lock.Unlock()
 		}
-	outloop:
 	}()
 }
 

pkg/proxy/recorder.go

@@ -21,8 +21,8 @@ func NewCommandRecorder(sid string) (recorder *CommandRecorder) {
 	return recorder
 }
 
-func NewReplyRecord(sid string) (recorder *ReplyRecorder) {
-	recorder = &ReplyRecorder{SessionID: sid}
+func NewReplyRecord(sid string) (recorder ReplyRecorder) {
+	recorder = ReplyRecorder{SessionID: sid}
 	recorder.initial()
 	return recorder
 }

pkg/proxy/recorderstorage/azure.go

@@ -18,7 +18,7 @@ type AzureReplayStorage struct {
 	EndpointSuffix string
 }
 
-func (a *AzureReplayStorage) Upload(gZipFilePath, target string) (err error) {
+func (a AzureReplayStorage) Upload(gZipFilePath, target string) (err error) {
 	file, err := os.Open(gZipFilePath)
 	if err != nil {
 		return

pkg/proxy/recorderstorage/es.go

@@ -17,7 +17,7 @@ type ESCommandStorage struct {
 	DocType string
 }
 
-func (es *ESCommandStorage) BulkSave(commands []*model.Command) (err error) {
+func (es ESCommandStorage) BulkSave(commands []*model.Command) (err error) {
 	var buf bytes.Buffer
 	esClinet, err := elasticsearch.NewClient(elasticsearch.Config{
 		Addresses: es.Hosts,

pkg/proxy/recorderstorage/oss.go

@@ -13,7 +13,7 @@ type OSSReplayStorage struct {
 	SecretKey string
 }
 
-func (o *OSSReplayStorage) Upload(gZipFilePath, target string) (err error) {
+func (o OSSReplayStorage) Upload(gZipFilePath, target string) (err error) {
 	client, err := oss.New(o.Endpoint, o.AccessKey, o.SecretKey)
 	if err != nil {
 		return

pkg/proxy/recorderstorage/s3.go

@@ -19,7 +19,7 @@ type S3ReplayStorage struct {
 	Endpoint  string
 }
 
-func (s *S3ReplayStorage) Upload(gZipFilePath, target string) (err error) {
+func (s S3ReplayStorage) Upload(gZipFilePath, target string) (err error) {
 
 	file, err := os.Open(gZipFilePath)
 	if err != nil {

pkg/proxy/recorderstorage/server.go

@@ -11,7 +11,7 @@ import (
 type ServerCommandStorage struct {
 }
 
-func (s *ServerCommandStorage) BulkSave(commands []*model.Command) (err error) {
+func (s ServerCommandStorage) BulkSave(commands []*model.Command) (err error) {
 	return service.PushSessionCommand(commands)
 }
 
@@ -19,7 +19,7 @@ type ServerReplayStorage struct {
 	StorageType string
 }
 
-func (s *ServerReplayStorage) Upload(gZipFilePath, target string) (err error) {
+func (s ServerReplayStorage) Upload(gZipFilePath, target string) (err error) {
 	sessionID := strings.Split(filepath.Base(gZipFilePath), ".")[0]
 	return service.PushSessionReplay(sessionID, gZipFilePath)
 }

pkg/proxy/switch.go

@@ -117,8 +117,8 @@ func (s *SwitchSession) SetFilterRules(cmdRules []model.SystemUserFilterRule) {
 // Bridge 桥接两个链接
 func (s *SwitchSession) Bridge(userConn UserConnection, srvConn srvconn.ServerConnection) (err error) {
 	var (
-		parser         *Parser
-		replayRecorder *ReplyRecorder
+		parser         Parser
+		replayRecorder ReplyRecorder
 
 		userInChan chan []byte
 		srvInChan  chan []byte

pkg/proxy/util.go

@@ -16,8 +16,8 @@ type CommandStorage interface {
 	BulkSave(commands []*model.Command) error
 }
 
-var defaultCommandStorage = &storage.ServerCommandStorage{}
-var defaultReplayStorage = &storage.ServerReplayStorage{StorageType: "server"}
+var defaultCommandStorage = storage.ServerCommandStorage{}
+var defaultReplayStorage = storage.ServerReplayStorage{StorageType: "server"}
 
 func NewReplayStorage() ReplayStorage {
 	cf := config.GetConf().ReplayStorage
@@ -31,14 +31,14 @@ func NewReplayStorage() ReplayStorage {
 		if endpointSuffix == "" {
 			endpointSuffix = "core.chinacloudapi.cn"
 		}
-		return &storage.AzureReplayStorage{
+		return storage.AzureReplayStorage{
 			AccountName:    cf["ACCOUNT_NAME"].(string),
 			AccountKey:     cf["ACCOUNT_KEY"].(string),
 			ContainerName:  cf["CONTAINER_NAME"].(string),
 			EndpointSuffix: endpointSuffix,
 		}
 	case "oss":
-		return &storage.OSSReplayStorage{
+		return storage.OSSReplayStorage{
 			Endpoint:  cf["ENDPOINT"].(string),
 			Bucket:    cf["BUCKET"].(string),
 			AccessKey: cf["ACCESS_KEY"].(string),
@@ -58,7 +58,7 @@ func NewReplayStorage() ReplayStorage {
 			region = strings.Split(endpoint, ".")[1]
 		}
 
-		return &storage.S3ReplayStorage{
+		return storage.S3ReplayStorage{
 			Bucket:    bucket,
 			Region:    region,
 			AccessKey: cf["ACCESS_KEY"].(string),
@@ -90,7 +90,7 @@ func NewCommandStorage() CommandStorage {
 		if docType == "" {
 			docType = "command_store"
 		}
-		return &storage.ESCommandStorage{Hosts: hosts, Index: index, DocType: docType}
+		return storage.ESCommandStorage{Hosts: hosts, Index: index, DocType: docType}
 	default:
 		return defaultCommandStorage
 	}

pkg/service/init.go

@@ -35,10 +35,9 @@ func Initial(ctx context.Context) {
 	go KeepSyncConfigWithServer(ctx)
 }
 
-func newClient() *common.Client {
+func newClient() common.Client {
 	cf := config.GetConf()
-	cli := common.NewClient(30, "")
-	cli.BaseHost = cf.CoreHost
+	cli := common.NewClient(30, cf.CoreHost)
 	return cli
 }
 
@@ -87,7 +86,7 @@ func MustLoadServerConfigOnce() {
 
 func LoadConfigFromServer() (err error) {
 	conf := config.GetConf()
-	_, err = authClient.Get(TerminalConfigURL, conf)
+	_, err = authClient.Get(TerminalConfigURL, &conf)
 	if err != nil {
 		return err
 	}