perf: add comment for appletHost session

jumpserver · Aug 1, 2024 · eab1177 · eab1177
1 parent a1dbd2e
commit eab1177
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/pkg/jms-sdk-go/model/session.go b/pkg/jms-sdk-go/model/session.go
@@ -20,6 +20,7 @@ type Session struct {
  UserID string `json:"user_id"`
  AssetID string `json:"asset_id"`
  AccountID string `json:"account_id"`
+ Comment string `json:"comment,omitempty"`
 }
 
 type ReplayVersion string

diff --git a/pkg/session/server.go b/pkg/session/server.go
@@ -252,9 +252,15 @@ func (s *Server) Create(ctx *gin.Context, opts ...TunnelOption) (sess TunnelSess
  sess.AppletOpts = opt.appletOpt
  sess.VirtualAppOpts = opt.virtualAppOPt
  sess.AuthInfo = opt.authInfo
+ comment := ""
  if opt.appletOpt != nil {
  sess.RemoteApp = &opt.appletOpt.Applet
+ comment = fmt.Sprintf(appletCommentTmpl,
+ opt.appletOpt.Host.String(),
+ opt.appletOpt.Account.String(),
+ opt.appletOpt.Applet.Name)
  }
+
  sess.User = opt.User
  sess.ExpireInfo = opt.ExpireInfo
  sess.Permission = &perm
@@ -273,6 +279,7 @@ func (s *Server) Create(ctx *gin.Context, opts ...TunnelOption) (sess TunnelSess
  UserID: sess.User.ID,
  AssetID: sess.Asset.ID,
  AccountID: opt.Account.ID,
+ Comment: comment,
  }
  sess.ConnectedCallback = s.RegisterConnectedCallback(jmsSession)
  sess.ConnectedSuccessCallback = s.RegisterConnectedSuccessCallback(jmsSession)
@@ -496,3 +503,9 @@ func ValidReplayDirname(dirname string) bool {
  _, err := time.Parse(recordDirTimeFormat, dirname)
  return err == nil
 }
+
+const appletCommentTmpl = `
+AppletHost: %s
+Account: %s
+Applet：%s
+`