1
package main
2

3
import (
4
  "bufio"
5
  "flag"
6
  "fmt"
7
  "github.com/rehtt/gogpio"
8
  "github.com/kardianos/service"
9
  "log"
10
  "os"
11
  "strconv"
12
  "time"
13
)
14

15
var (
16
  max, min    int
17
  showTemp, h bool
18
)
19

20
func main() {
21

22
  flag.BoolVar(&h, "h", false, "help")
23
  flag.IntVar(&max, "m", 55, "开启风扇温度")
24
  flag.IntVar(&min, "n", 50, "关闭风扇温度")
25
  flag.BoolVar(&showTemp, "s", false, "查看当前温度")
26
  flag.Usage = help
27
  flag.Parse()
28
  if h {
29
    flag.Usage()
30
  } else {
31
    if !showTemp {
32
      //服务
33
      cfg := &service.Config{
34
        Name:        "temp",
35
        DisplayName: "RPI CPU Temp",
36
        Description: "RPI CPU Temp",
37
      }
38
      prg := &program{}
39
      s, err := service.New(prg, cfg)
40
      if err != nil {
41
        log.Fatal(err)
42
      }
43
      // logger 用于记录系统日志
44
      logger, err := s.Logger(nil)
45
      if err != nil {
46
        log.Fatal(err)
47
      }
48
      if len(os.Args) == 2 { //如果有命令则执行
49
        err = service.Control(s, os.Args[1])
50
        if err != nil {
51
          log.Fatal(err)
52
        }
53
      } else { //否则说明是方法启动了
54
        err = s.Run()
55
        if err != nil {
56
          logger.Error(err)
57
        }
58
      }
59
      if err != nil {
60
        logger.Error(err)
61
      }
62
    } else {
63
      fmt.Println("当前温度：", t(), "°")
64
    }
65
  }
66

67
}
68

69
func t() int {
70
  txt, _ := os.Open("/sys/class/thermal/thermal_zone0/temp")
71
  defer txt.Close()
72
  buf := bufio.NewReader(txt)
73
  t, _, _ := buf.ReadLine()
74
  temp, _ := strconv.Atoi(string(t))
75
  temp = temp / 1000
76
  return temp
77
}
78

79
func help() {
80
  fmt.Fprintln(os.Stderr, `使用：temp [选项] [服务操作]
81
选项：`)
82
  flag.PrintDefaults()
83
  fmt.Fprintln(os.Stderr,`
84
服务操作：
85
  install    安装服务（开启服务必做）
86
  start    运行服务
87
  stop    停止服务
88
  uninstall  卸载服务
89

90
示例：
91
  temp -s      显示温度
92
  temp install && temp start    安装服务并启动
93
  temp stop    停止服务
94
  temp -m 50  start  带参数启动服务
95
启动服务需要root权限`)
96
}
97

98
//服务
99
type program struct{}
100

101
func (p *program) Start(s service.Service) error {
102
  log.Println("开始服务")
103
  go p.run()
104
  return nil
105
}
106
func (p *program) Stop(s service.Service) error {
107
  log.Println("停止服务")
108
  return nil
109
}
110
func (p *program) run() {
111
  // 这里放置程序要执行的代码……
112
  pin, _ := gogpio.Open(14, gogpio.OUT)
113
  for ; ; {
114
    if t() >= max {
115
      pin.High()
116
    } else if t() <= min {
117
      pin.Low()
118
    }
119
    time.Sleep(30 * time.Second)
120
  }
121
}