当前位置:首页 > 文章列表 > Golang > Go问答 > Golang的defer语句并不总是被执行

Golang的defer语句并不总是被执行

来源:stackoverflow 2024-02-06 15:45:22 0浏览 收藏

Golang小白一枚,正在不断学习积累知识,现将学习到的知识记录一下,也是将我的所得分享给大家!而今天这篇文章《Golang的defer语句并不总是被执行》带大家来了解一下##content_title##,希望对大家的知识积累有所帮助,从而弥补自己的不足,助力实战开发!


问题内容

我正在运行一个 golang tcp 服务器,它接收连接,生成一个 goroutine 来处理该连接,然后生成一个新的 goroutine 来从连接中读取数据并将数据写入通道。还有一个简单的函数可以从通道读取数据。

这个想法是,如果客户端关闭连接,readfromconn goroutine 将在读取时收到错误,然后返回。延迟代码将写入 done 通道并关闭 donequeue 通道。如果done通道有数据,消费者代码将停止处理并返回当时的任何结果。

另一种情况是消费者可以确定自己有足够的数据来做出决定并返回。这将导致执行返回到handle函数。当该函数返回时,handle 函数上的 defer 将关闭连接,导致 readfromconn goroutine 在从连接读取数据时收到错误并关闭所有挂起的通道。

这效果很好,但我正在做一些负载测试,并注意到测试完成后 golang 服务器的内存使用量并没有像负载停止时应用程序的所有其他部分那样减少。我拿了一些 id 并检查了日志。我看到有时(并非总是)显示error reading from conn消息,但没有read_from_conn defer日志,所以我认为延迟永远不会被调用。结果,processqueue 挂起从通道的读取,因为它从未关闭,并且来自 handle 函数的 closing... 日志为也不见了。

至少,这就是我认为正在发生的事情,并且我相信这就是为什么负载测试结束时内存消耗永远不会下降的原因,因为代码仍在运行一些从通道读取的 goroutine,这些通道应该由延迟代码关闭从conn读取。这种行为是不可预测的;并非所有连接都会发生这种情况,所以我不知道可能出了什么问题。

这是我的 golang 服务器的简化版本:

package main

import (
  "os"
  "net"
  "fmt"
  "io"
)

type CustomStruct struct {
  Type string
  Stop bool
}

func main() {
  // Creates server
  server, err := net.Listen("tcp", "0.0.0.0:80")
  if err != nil {
    fmt.Println("failed to bind listener to socket", err)
  }
  defer server.Close()
  fmt.Println("Listening new connections V2")

  // Starts reading from the server
  for {
    conn, err := server.Accept()
    if err != nil {
      fmt.Println("failed to accept new connection:", err)
      continue
    }

    go Handle(conn)
  }
}

func Handle(conn net.Conn) {
  defer conn.Close()

  id := "some uuid for each conn"

  // Creates channels
  queue := make(chan []byte, 512)
  done := make(chan bool)

  // Starts reading from the server
  go ReadFromConn(id, conn, queue, done)

  result := ProcessQueue(id, queue, done)

  fmt.Println(id, "CLOSING...")

  // Do stuffs with result...
  fmt.Println(id, result)
}

func ReadFromConn(
  id string,
  conn io.Reader,
  queue chan []byte,
  done chan bool,
) {
  defer func() {
    done <- true
    close(queue)
    close(done)
    fmt.Println(id, "READ_FROM_CONN DEFER")
  }()

  tmp := make([]byte, 256)
  for {
    _, err := conn.Read(tmp)
    if err != nil {
      fmt.Println(id, "ERROR READING FROM CONN " + err.Error())
      return
    }
    if (tmp[0] == 0x00) {
      return
    }
    queue <- tmp
  }
}

func ProcessQueue(
  id string,
  queue chan []byte,
  done chan bool,
) CustomStruct {
  defer fmt.Println(id, "GET_TRANSCRIPTION_RESULT ENDED")
  fmt.Println(id, "GET_TRANSCRIPTION_RESULT STARTED")

  result := CustomStruct{
    Type: "transcription",
    Stop: false,
  }

  for {
    select {
    case <-done:
      fmt.Println(id, "DONE DETECTED")
      return result
    default:
      fmt.Println(id, "DEFAULT")
      payload, open := <-queue
      if open == false {
        fmt.Println(id, "QUEUE IS CLOSED")
        return result
      } else {
        fmt.Println(id, "QUEUE IS OPEN")
      }
      // ... Do stuffs with payload, if certain condition is met, of the result of processing payload, return
      if (payload[0] == 0x01) {
        return result
      }
    }
  }

  return result
}

正确答案


不完全确定问题是什么,但是使用上下文对象可以更干净地控制读取器的生命周期。上下文避免了必须密切管理通道对象,并且在需要时使用 context.cause(ctx) 提供了一种从 goroutine 报告错误的干净方法。

示例设置可能如下所示:

ctx, cancel := context.withcancelcause(context.background())
queue := make(chan []byte, 512)
go readfromconn(id, conn, queue, ctx, cancel)
result := processqueue(id, queue, ctx, cancel)

延迟调用可能会在上下文原因中放置错误,这会导致 ctx.done 方法现在返回具有存在值的通道。此方法也是可重复的,这意味着在使用简单通道方法时,多个 goroutine 都可以接收一个值,而不是仅接收一个值。

发送到队列时,您还可以选择 ctx.done,以防止永远等待不再读取的队列。同样,您也可以从队列中读取数据。

func readfromconn(...) {
  defer cancel(fmt.errorf("reader defer"))
  ...
  for {
    ...
    select {
    case queue <- tmp:
    case <-ctx.done():
        return
    }
  }
}
func ProcessQueue(...) {
  defer cancel(fmt.Errorf("queue defer"))
  ...
  for {
    select {
    case <-ctx.Done():
      ...
    case payload := <-queue:
      ...
    }
  }

理论要掌握,实操不能落!以上关于《Golang的defer语句并不总是被执行》的详细介绍,大家都掌握了吧!如果想要继续提升自己的能力,那么就来关注golang学习网公众号吧!

版本声明
本文转载于:stackoverflow 如有侵犯,请联系study_golang@163.com删除
Go 中 DynamoDB 转换失败的错误 ResourceNotFoundExceptionGo 中 DynamoDB 转换失败的错误 ResourceNotFoundException
上一篇
Go 中 DynamoDB 转换失败的错误 ResourceNotFoundException
Go Build 与其他 Go 命令遇到 Git 异常
下一篇
Go Build 与其他 Go 命令遇到 Git 异常
查看更多
最新文章
查看更多
课程推荐
  • 前端进阶之JavaScript设计模式
    前端进阶之JavaScript设计模式
    设计模式是开发人员在软件开发过程中面临一般问题时的解决方案,代表了最佳的实践。本课程的主打内容包括JS常见设计模式以及具体应用场景,打造一站式知识长龙服务,适合有JS基础的同学学习。
    542次学习
  • GO语言核心编程课程
    GO语言核心编程课程
    本课程采用真实案例,全面具体可落地,从理论到实践,一步一步将GO核心编程技术、编程思想、底层实现融会贯通,使学习者贴近时代脉搏,做IT互联网时代的弄潮儿。
    511次学习
  • 简单聊聊mysql8与网络通信
    简单聊聊mysql8与网络通信
    如有问题加微信:Le-studyg;在课程中,我们将首先介绍MySQL8的新特性,包括性能优化、安全增强、新数据类型等,帮助学生快速熟悉MySQL8的最新功能。接着,我们将深入解析MySQL的网络通信机制,包括协议、连接管理、数据传输等,让
    498次学习
  • JavaScript正则表达式基础与实战
    JavaScript正则表达式基础与实战
    在任何一门编程语言中,正则表达式,都是一项重要的知识,它提供了高效的字符串匹配与捕获机制,可以极大的简化程序设计。
    487次学习
  • 从零制作响应式网站—Grid布局
    从零制作响应式网站—Grid布局
    本系列教程将展示从零制作一个假想的网络科技公司官网,分为导航,轮播,关于我们,成功案例,服务流程,团队介绍,数据部分,公司动态,底部信息等内容区块。网站整体采用CSSGrid布局,支持响应式,有流畅过渡和展现动画。
    484次学习
查看更多
AI推荐
  • 千音漫语:智能声音创作助手,AI配音、音视频翻译一站搞定!
    千音漫语
    千音漫语,北京熠声科技倾力打造的智能声音创作助手,提供AI配音、音视频翻译、语音识别、声音克隆等强大功能,助力有声书制作、视频创作、教育培训等领域,官网:https://qianyin123.com
    96次使用
  • MiniWork:智能高效AI工具平台,一站式工作学习效率解决方案
    MiniWork
    MiniWork是一款智能高效的AI工具平台,专为提升工作与学习效率而设计。整合文本处理、图像生成、营销策划及运营管理等多元AI工具,提供精准智能解决方案,让复杂工作简单高效。
    89次使用
  • NoCode (nocode.cn):零代码构建应用、网站、管理系统,降低开发门槛
    NoCode
    NoCode (nocode.cn)是领先的无代码开发平台,通过拖放、AI对话等简单操作,助您快速创建各类应用、网站与管理系统。无需编程知识,轻松实现个人生活、商业经营、企业管理多场景需求,大幅降低开发门槛,高效低成本。
    107次使用
  • 达医智影:阿里巴巴达摩院医疗AI影像早筛平台,CT一扫多筛癌症急慢病
    达医智影
    达医智影,阿里巴巴达摩院医疗AI创新力作。全球率先利用平扫CT实现“一扫多筛”,仅一次CT扫描即可高效识别多种癌症、急症及慢病,为疾病早期发现提供智能、精准的AI影像早筛解决方案。
    98次使用
  • 智慧芽Eureka:更懂技术创新的AI Agent平台,助力研发效率飞跃
    智慧芽Eureka
    智慧芽Eureka,专为技术创新打造的AI Agent平台。深度理解专利、研发、生物医药、材料、科创等复杂场景,通过专家级AI Agent精准执行任务,智能化工作流解放70%生产力,让您专注核心创新。
    100次使用
微信登录更方便
  • 密码登录
  • 注册账号
登录即同意 用户协议隐私政策
返回登录
  • 重置密码