专栏首页陶士涵的菜地[Go] imap收信非并发

[Go] imap收信非并发

待修正

package main

import (
    "flag"
    "fmt"
    "io/ioutil"
    "log"
    "os"

    "github.com/emersion/go-imap"
    "github.com/emersion/go-imap/client"
)

var (
    server, email, password string
    workernum               int
    imapClient              *client.Client
    mailDirs                []string
    mailSumNums             uint32
)

func main() {
    //获取命令行参数
    flag.StringVar(&server, "server", "", "imap服务地址(包含端口)")
    flag.StringVar(&email, "email", "", "邮箱名")
    flag.StringVar(&password, "password", "", "密码")
    flag.IntVar(&workernum, "workernum", 10, "并发数:")
    flag.Parse()
    if flag.NFlag() < 3 {
        flag.PrintDefaults()
        os.Exit(1)
    }
    if server == "" || email == "" || password == "" || workernum == 0 {
        log.Fatal("服务器地址,用户名,密码,参数错误")
    }
    //连接imap服务
    imapClient, err := client.Dial(server)
    if err != nil {
        log.Fatal(err)
    }
    //登陆
    if err := imapClient.Login(email, password); err != nil {
        log.Fatal(err)
    }
    //创建邮件夹目录
    os.Mkdir(email, os.ModePerm)
    mailboxes := make(chan *imap.MailboxInfo, 20)
    go func() {
        imapClient.List("", "*", mailboxes)
    }()
    //列取邮件夹
    for m := range mailboxes {
        mailDirs = append(mailDirs, m.Name)
    }

    for _, mailDir := range mailDirs {
        //选中每个邮件夹
        mbox, err := imapClient.Select(mailDir, false)
        if err != nil {
            log.Fatal(err)
        }
        mailDirNums := mbox.Messages
        log.Printf("%s : %d", mailDir, mailDirNums)
        fileDir := fmt.Sprintf("%s/%s_%d", email, mailDir, mailDirNums)
        //创建邮件夹目录
        os.Mkdir(fileDir, os.ModePerm)
        mailSumNums += mailDirNums
    }
    log.Printf("总邮件数 : %d", mailSumNums)
    for _, mailDir := range mailDirs {
        //选中每个邮件夹
        mbox, err := imapClient.Select(mailDir, false)
        if err != nil {
            log.Fatal(err)
        }
        //循环该邮件夹中的邮件
        seqset := new(imap.SeqSet)
        seqset.AddRange(1, mbox.Messages)
        section := &imap.BodySectionName{}
        items := []imap.FetchItem{section.FetchItem()}

        messages := make(chan *imap.Message, mbox.Messages)
        go func() {
            imapClient.Fetch(seqset, items, messages)
        }()

        for msg := range messages {
            mailFile := fmt.Sprintf("%s/%s_%d/%d.eml", email, mailDir, mbox.Messages, msg.SeqNum)
            r := msg.GetBody(section)
            if r == nil {
                log.Printf("%s-%dServer didn't returned message body", mailDir, msg.SeqNum)
            }
            if r != nil {
                body, err := ioutil.ReadAll(r)
                if err != nil {
                    log.Printf("%s:%d ioutil.ReadAll error", mailDir, msg.SeqNum)
                }

                file6, err := os.OpenFile(mailFile, os.O_RDWR|os.O_CREATE, 0766)
                if err != nil {
                    log.Printf("%s:%d os.OpenFile error %s", mailDir, msg.SeqNum, mailFile)
                }
                file6.Write(body)
                file6.Close()
                log.Printf("%s :第 %d ", mailDir, msg.SeqNum)
            }
        }
    }
}

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

我来说两句

0 条评论
登录 后参与评论

相关文章

  • [日常] nginx记录post数据

    server { access_log /var/log/nginx/default.access.log 日志名; }

    陶士涵
  • [Go] 并发imap收信

    陶士涵
  • [TCP/IP]TCP服务端accept发生在三次握手的哪一个阶段

    客户端 socket()==>connect()==>write()==>read() 服务端 socket()==>bind()==>listen()==>a...

    陶士涵
  • ELK学习笔记之ELK搜集OpenStack节点日志

    控制节点作为日志服务器,存储所有 OpenStack 及其相关日志。Logstash 部署于所有节点,收集本节点下所需收集的日志,然后以网络(node/http...

    Jetpropelledsnake21
  • JavaScript检测IE浏览器(最短代码)

    deepcc
  • ReactJS分析之入口函数render

    前言   在使用React进行构建应用时,我们总会有一个步骤将组建或者虚拟DOM元素渲染到真实的DOM上,将任务交给浏览器,进而进行layout和paint等...

    欲休
  • spring boot集成WebSocket实时输出日志到web页面

    前言碎语 今天来做个有趣的东西,就是实时将系统日志输出的前端web页面,因为是实时输出,所有第一时间就想到了使用webSocket,而且在spring boot...

    kl博主
  • js 判断当前操作系统是ios还是android还是电脑端

    lin_zone
  • spring boot集成WebSocket实时输出日志到web页面

    前言碎语 今天来做个有趣的东西,就是实时将系统日志输出的前端web页面,因为是实时输出,所有第一时间就想到了使用webSocket,而且在spring boot...

    kl博主
  • 代码中太多 if else 怎么办?

    前段时间,我将公司系统中的批量审单的功能进行了重构,用到了java的并发编程进行异步化处理,数据库的乐观锁机制处理多线程并发更新数据。其中批量审单的业务处理涉及...

    一个优秀的废人

扫码关注云+社区

领取腾讯云代金券