feat(context): close handle

solywsh · Dec 8, 2022 · 74479e0 · 74479e0
1 parent 8f27910
commit 74479e0
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/chatgpt.go b/chatgpt.go
@@ -13,6 +13,7 @@ type ChatGPT struct {
  maxToken int
  timeOut time.Duration
  timeOutChan chan struct{}
+ cancel func()
 }
 
 func New(ApiKey, UserId string, timeOut time.Duration) *ChatGPT {
@@ -21,7 +22,6 @@ func New(ApiKey, UserId string, timeOut time.Duration) *ChatGPT {
  go func() {
  <-ctx.Done()
  timeOutChan <- struct{}{}
- cancel()
  }()
  return &ChatGPT{
  client: gogpt.NewClient(ApiKey),
@@ -30,10 +30,13 @@ func New(ApiKey, UserId string, timeOut time.Duration) *ChatGPT {
  maxToken: 1024,
  timeOut: timeOut,
  timeOutChan: timeOutChan,
+ cancel: func() {
+ cancel()
+ },
  }
 }
 func (c *ChatGPT) Close() {
- c.ctx.Done()
+ c.cancel()
 }
 
 func (c *ChatGPT) GetTimeOutChan() chan struct{} {