学习go语言并完成第一个作品

  1. 之前有使用C#写一个Windows下的发送邮件的命令行工具,方便一些脚本出现异常时向我的邮箱发送邮件提醒。但这并没有被我频繁使用,因为我的有些脚本还是在linux下面运行,因此我又有一篇文章用linux的C编写一个发送邮件的可执行程序,但是功能太简单了,中文字符很难处理。
  2. 因此我选择了Go语言,因为Go可以直接build成一个可执行程序,一套代码可以编译成linux和Windows的可执行程序,方便维护,且没有任何依赖,copy到其他机器上也可以直接运行,太方便了。因此我决定好好的研究Go,毕竟有Google这个大佬来带头还是比较可靠的。
  3. 这里我先说一下我的Go环境安装部署。
1.下载go:https://www.golangtc.com/download
2.安装很简单,完了记得配置GOROOT、GOPATH,等环境变量
3.这里推荐使用32位的Go,因为这样可以同时在32位和64位机器运行,好处多多
4.一个好的编辑器必不可少:https://www.jetbrains.com/go/
  1. 以上ok,那么就试一试hello word!吧,将一下代码保存为main.go
    执行go run main.go即可,如果需要可执行程序就go build main.go
package main
import (
  "fmt"
)
func main() {
  fmt.Println("hello word!")
}
  1. 经过以上步骤我们就验证了环境,下面就要想一想go的第三方发送邮件的包了
    这里是我找到一个包:http://gopkg.in/gomail.v2
    使用命令:go get gopkg.in/gomail.v2 即可下载这个包
    代码中用:import "gopkg.in/gomail.v2" 即可包含这个包
    具体使用方法看:http://godoc.org/gopkg.in/gomail.v2 里面有例子代码
  2. 以上只准备好了环境和第三方包,但是GO语言基础知识还是需要好好学习的
    相信这个网站:http://tour.studygolang.com/list 会让你一步一个脚印的学习
    基础语法等和大多数语言差不多,我觉得Go好用的就是defer、协程、chan、以及<-和->的操作,用的好了简直非常方便。
    !!!有一点我觉得非常好,就是Go禁止有二义性的操作,连三目运算符while、do while等等貌似都去掉了。传递参数也非常严谨,相比于我之前使用的ruby和Python更加死板,有点贴近C语言的感觉。这让代码维护和可读性很高。
  3. 我大概花了一个星期把上面的事情做完,且有一定的编码基础了,当然离Go打师还差的远了。看看我找的发送邮件的第三方包源码,简直和天书一样。革命尚未成功,同志仍需努力啊。下面是我完成的一整套代码,程序的输入由配置文件和命令行两个地方得到,命令行的部分会覆盖配置文件,及一个配置在配置文件和命令行同时包含则只使用命令行的配置
package main

import (
  "encoding/json"
  "flag"
  "fmt"
  "io/ioutil"
  "os"
  "path/filepath"
  "regexp"
  "strconv"
  "strings"
  
  "gopkg.in/gomail.v2"
)

/***
* json结构体
* 解析json配置时使用
***/
type Login struct {
  User string
  Pass string
}
type Email struct {
  Name string
  Mail string
}
type MyMail struct {
  Address string `json:"address"`
  From    Email `json:"from"`
  Login   Login `json:"login"`
  To      []Email `json:"to"`
  Cc      []Email `json:"cc"`
  Bcc     []Email `json:"bcc"`
  Subject string `json:"subject"`
  Msg     string `json:"msg"`
}
type Args struct {
  address *string
  from    *string
  login   *string
  to      *string
  cc      *string
  bcc     *string
  subject *string
  msg     *string
  msgf    *string
  files   *string
}

/***
*  最终可用参数
***/
type Msg struct {
  ip      string
  port    int
  from    string
  login   Login
  to      []string
  cc      []string
  bcc     []string
  subject string
  msg     string
}

func main() {
  var input_args Args // 输入各个参数
  conf_name := flag.String("c", "", "-c=/etc/mail.json\n\tConfig file!")
  input_args.address = flag.String("addr", "", "-addr=192.200.4.13:25\n\tServer address!")
  input_args.from = flag.String("from", "", "-from=xxx@yyy.com\n\tMail From!")
  input_args.login = flag.String("login", "", "-login=\"xxx@yyy.com,******\"\n\tLogin user and pass!")
  input_args.to = flag.String("to", "", "-to=\"to1@yyy.com,to2@yyy.com\"\n\tMail To!")
  input_args.cc = flag.String("cc", "", "-cc=\"cc1@yyy.com,cc2@yyy.com\"\n\tMail Cc!")
  input_args.bcc = flag.String("bcc", "", "-bcc=\"bcc1@yyy.com,bcc2@yyy.com\"\n\tMail Bcc!")
  input_args.subject = flag.String("subject", "", "-subject=\"This is default mail title!\"\n\tMail Subject!")
  input_args.msg = flag.String("msg", "", "-msg=\"This is default mail body!\"\n\tMsg Body!")
  input_args.msgf = flag.String("msgf", "", "-msgf=/home/msg_body.txt\n\tMsg Body From File!")
  input_args.files = flag.String("files", "", "-files=\"a.txt,b.txt\"\n\tMail add files!")
  expt_conf := flag.String("export", "", "-export=mail.json\n\tExport example config!")
  flag.Parse()
  
  if *expt_conf != "" {
    name := filepath.Base(*expt_conf)
    file, err := os.Create(name)
    if err != nil {
      fmt.Println("open file failed :", err.Error())
      return
    }
    file.WriteString(`{
  "address": "192.200.4.13:25",
  "from": {
    "name": "Send",
    "mail": "xxx@yyy.com"
  },
  "login": {
    "user": "user name",
    "pass": "******"
  },
  "to": [
    {
      "name": "To1",
      "mail": "xxx@yyy.com"
    },
    {
      "name": "To2",
      "mail": "xxx@yyy.com"
    }
  ],
  "cc": [
    {
      "name": "Cc1",
      "mail": "xxx@yyy.com"
    },
    {
      "name": "Cc2",
      "mail": "xxx@yyy.com"
    }
  ],
  "bcc": [
    {
      "name": "Bcc1",
      "mail": "xxx@yyy.com"
    },
    {
      "name": "Bcc2",
      "mail": "xxx@yyy.com"
    }
  ],
  "subject": "测试邮件标题",
  "msg": "测试邮件内容"
}`)
    fmt.Println("  export file:", name, "ok!")
    file.Close()
    return
  }
  
  m := gomail.NewMessage() // mail变量赋值
  stat, msg := check_config(*conf_name, m)
  if stat == false {
    return // 从json中读配置错误
  }
  
  stat, msg = merge_config(msg, input_args)
  if stat == false {
    return // 合并json文件和命令行错误
  }
  
  m.SetHeader("From", msg.from)
  if len(msg.to) != 0 {
    m.SetHeader("To", msg.to ...)
  }
  if len(msg.cc) != 0 {
    m.SetHeader("Cc", msg.cc ...)
  }
  if len(msg.bcc) != 0 {
    m.SetHeader("Bcc", msg.bcc ...)
  }
  
  m.SetHeader("Subject", msg.subject) // 设置主题
  
  if *input_args.msgf != "" {
    b, err := ioutil.ReadFile(*input_args.msgf)
    if err != nil {
      fmt.Println("  -msgf=" + *input_args.msgf + ", readfile " + err.Error())
      return
    }
    m.SetBody("text/plain", string(b))
  } else {
    m.SetBody("text/plain", msg.msg)
  }
  
  for _, i := range strings.Split(*input_args.files, ",") {
    if FilePathExist(i, true) {
      m.Attach(i)
    }
  }
  
  var send_err error // 下面包含使用密码和不使用密码的发送方式
  if msg.login.User == "" || msg.login.Pass == "" {
    d := gomail.Dialer{Host: msg.ip, Port: msg.port}
    send_err = d.DialAndSend(m)
  } else {
    d := gomail.NewDialer(msg.ip, msg.port, msg.login.User, msg.login.Pass)
    send_err = d.DialAndSend(m)
  }
  
  if send_err != nil {
    panic(send_err)
  }
  
  fmt.Println("send mail ok...")
}

/**
* 从json文件中读取配置
***/
func read_config(filename string) (error, MyMail) {
  var r MyMail
  
  bytes, err := ioutil.ReadFile(filename)
  if err != nil {
    //fmt.Println("ReadFile: ", err.Error())
    return err, r
  }
  
  if err := json.Unmarshal(bytes, &r); err != nil {
    fmt.Println("Unmarshal: ", err.Error())
    return err, r
  }
  
  return nil, r
}

/**
*  检查文件传入的参数,并返回正确和对应的值
**/
func check_config(filename string, m *gomail.Message) (bool, Msg) {
  var result Msg // 返回数据
  
  err, config := read_config(filename)
  if err != nil { // 读取json文件失败,则可从命令行输入
    return true, result
  }
  
  tmp := strings.Split(config.Address, ":")
  if len(tmp) == 2 && tmp[0] != "" && tmp[1] != "" {
    result.ip = tmp[0] // 以上得到服务器ip和端口
    if result.port, err = strconv.Atoi(tmp[1]); err != nil {
      fmt.Println("strconv.Atoi: ", err.Error())
      return false, result
    }
  }
  
  if !IsEmail(config.From.Mail) {
    fmt.Println("From mail <" + config.From.Mail + "> is not mailbox!")
    return false, result
  }
  result.from = m.FormatAddress(config.From.Mail, config.From.Name)
  result.login = config.Login // 登录用户名和密码
  
  for _, i := range config.To {
    if IsEmail(i.Mail) {
      result.to = append(result.to, m.FormatAddress(i.Mail, i.Name))
    }
  }
  
  for _, i := range config.Cc {
    if IsEmail(i.Mail) {
      result.cc = append(result.cc, m.FormatAddress(i.Mail, i.Name))
    }
  }
  
  for _, i := range config.Bcc {
    if IsEmail(i.Mail) {
      result.bcc = append(result.bcc, m.FormatAddress(i.Mail, i.Name))
    }
  }
  
  result.subject = config.Subject
  result.msg = config.Msg
  return true, result
}

/**
*  合并配置文件的配置和前台输入的配置
**/
func merge_config(msg Msg, input Args) (bool, Msg) {
  var (
    result Msg
    cnt    int
    err    error
    tmp    []string
  )
  
  result.ip, result.port = msg.ip, msg.port
  tmp = strings.Split(*input.address, ":")
  if len(tmp) == 2 && tmp[0] != "" && tmp[1] != "" {
    result.ip = tmp[0] // 以上得到服务器ip和端口
    if result.port, err = strconv.Atoi(tmp[1]); err != nil {
      fmt.Println("strconv.Atoi: ", err.Error())
      return false, result
    }
  }
  if result.ip == "" || result.port <= 0 {
    fmt.Println("ip or port not input!")
    return false, result
  } // 以上步骤获取ip和port,命令行输入覆盖配置文件
  
  result.from = msg.from
  if *input.from != "" && IsEmail(*input.from) {
    result.from = *input.from // 命令行发件人优先
  }
  if result.from == "" {
    fmt.Println("mail from not input!")
    return false, result
  }
  
  result.login.User, result.login.Pass = msg.login.User, msg.login.Pass
  tmp = strings.Split(*input.login, ",")
  if len(tmp) == 2 && tmp[0] != "" && tmp[1] != "" {
    result.login.User, result.login.Pass = tmp[0], tmp[1]
  } // 登录用户名和密码,可以为空
  
  if cnt = len(msg.to); cnt > 0 {
    result.to = make([]string, cnt)
    copy(result.to, msg.to)
  }
  tmp = strings.Split(*input.to, ",")
  for _, i := range tmp {
    if IsEmail(i) {
      result.to = append(result.to, i)
    }
  }
  
  if cnt = len(msg.cc); cnt > 0 {
    result.cc = make([]string, cnt)
    copy(result.cc, msg.cc)
  }
  tmp = strings.Split(*input.cc, ",")
  for _, i := range tmp {
    if IsEmail(i) {
      result.cc = append(result.cc, i)
    }
  }
  
  if cnt = len(msg.bcc); cnt > 0 {
    result.bcc = make([]string, cnt)
    copy(result.bcc, msg.bcc)
  }
  tmp = strings.Split(*input.bcc, ",")
  for _, i := range tmp {
    if IsEmail(i) {
      result.bcc = append(result.bcc, i)
    }
  }
  
  if len(result.to) == 0 && len(result.cc) == 0 && len(result.bcc) == 0 {
    fmt.Println("must set To or Cc or Bcc mailbox !")
    return false, result
  } // 至少要有一个收件人或抄送人或密送人
  
  result.subject = msg.subject
  if *input.subject != "" {
    result.subject = *input.subject
  }
  if result.subject == "" {
    result.subject = "This is default mail title!"
  }
  
  result.msg = msg.msg
  if *input.msg != "" {
    result.msg = *input.msg
  }
  if result.msg == "" {
    result.msg = "This is default mail body!"
  }
  
  return true, result
}

// 判断是否为邮箱
func IsEmail(email string) bool {
  if email != "" { // ^(\\w)+(\\.\\w+)*@(\\w)+((\\.\\w+)+)$
    if isOk, _ := regexp.MatchString("^[_a-z0-9-]+(\\.[_a-z0-9-]+)*@[a-z0-9-]+(\\.[a-z0-9-]+)*(\\.[a-z]{2,4})$", email); isOk {
      return true
    }
  }
  return false
}

// 判断文件夹或文件存在
func FilePathExist(path string, isfile bool) bool {
  f, err := os.Stat(path)
  if err == nil {
    return isfile || f.IsDir()
  }
  return os.IsExist(err)
}
  1. 以上是学习Go的整个过程,以及成果代码展示。期间遇到很多问题均自己通过度娘解决。更加深入理解了Go语言的一些特性。下面要讲一些Go的小技巧
    编译时去掉调试信息:go build -ldflags "-s -w" mail.go(减小程序体积)
    另外有个非常好的软件Windows和linux都有:upx -9 mail.exe(减小程序体积)
    Go的一个hello word程序都是好几M,因此在使用时有必要这么搞一搞
  2. 本次通过系统的给自己定目标并认真学习Go语言,让自己了解了很多可以方便工作的方式方法。以后要是再写一些小程序,肯定会优先考虑Go语言的,毕竟这年头编译成可执行程序的语言不多了,脚本语言换个平台还要搭建开发环境真的很烦人啊!
posted @ 2020-09-20 10:08  janbar  阅读(344)  评论(0编辑  收藏  举报