把golang程序注册成windows服务

共计 2362 个字符,预计需要花费 6 分钟才能阅读完成。

一、golang程序编写

1、下载第三方包github.com/kardianos/servicegithub.com/shirou/gopsutil/process

go get github.com/kardianos/service     //主要用于windows和linux作为服务运行的应用程序
go get github.com/shirou/gopsutil/process   //主要用于获取进程信息

2、可用代码

package main

import (
    "fmt"
    "github.com/kardianos/service"
    "github.com/shirou/gopsutil/process"
    "os"
    "time"
)

func main() {
    srvConfig := &service.Config{
        Name:        "MyUserExperienceService",
        DisplayName: "MyUserExperienceService数据提取",
        Description: "MyUserExperience数据提取服务",
    }
    prg := &program{}
    s, err := service.New(prg, srvConfig)
    if err != nil {
        fmt.Println(err)
    }
    if len(os.Args) > 1 {
        serviceAction := os.Args[1]
        switch serviceAction {
        case "install":
            err := s.Install()
            if err != nil {
                fmt.Println("安装服务失败: ", err.Error())
            } else {
                fmt.Println("安装服务成功")
            }
            return
        case "uninstall":
            err := s.Uninstall()
            if err != nil {
                fmt.Println("卸载服务失败: ", err.Error())
            } else {
                fmt.Println("卸载服务成功")
            }
            return
        case "start":
            err := s.Start()
            if err != nil {
                fmt.Println("运行服务失败: ", err.Error())
            } else {
                fmt.Println("运行服务成功")
            }
            return
        case "stop":
            err := s.Stop()
            if err != nil {
                fmt.Println("停止服务失败: ", err.Error())
            } else {
                fmt.Println("停止服务成功")
            }
            return
        }
    }

    err = s.Run()
    if err != nil {
        fmt.Println(err)
    }
}

type program struct{}

func (p *program) Start(s service.Service) error {
    fmt.Println("服务运行...")
    go p.run()
    return nil
}
func (p *program) run() {
    // 具体的服务实现
    t := time.Now()
    fileName := fmt.Sprintf("process-info-%s.txt", t.Format("2006-01-02-15-04-05"))
    file, err := os.Create(fileName)
    //file, err := os.OpenFile(fileName, os.O_CREATE|os.O_WRONLY, os.ModePerm)
    if err != nil {
        fmt.Println("创建文件失败:", err)
        return
    }
    defer file.Close()

    tick := time.Tick(3 * time.Second)

    for {
        select {
        case <-tick:
            now := t.Format("2006-01-02 15:04:05")
            processes, _ := process.Processes()
            for _, p := range processes {
                name, _ := p.Name()
                pid := p.Pid
                memory, _ := p.MemoryInfo()
                cpuPercent, _ := p.CPUPercent()
                //fmt.Printf("Process name: %s,PID:%d,Memory: Usage:%v,CPU uage:%f\n", name, pid, memory, cpuPercent)
                _, err := file.WriteString(fmt.Sprintf("time:%s,Process name: %s,PID:%d,Memory: Usage:%v,CPU uage:%f\n", now, name, pid, memory, cpuPercent))
                if err != nil {
                    fmt.Println("写入信息失败:", err)
                    return
                }
            }
            fmt.Printf("%s 写入成功 \n", now)
        }
    }
}
func (p *program) Stop(s service.Service) error {
    return nil
}

二、在windows中注册服务

1、在windows中编译成可执行文件,也就.exe文件
go 代码解读复制代码

go env                          #查看go环境的相关信息
set GOHOSTOS=windows            #发现是windows就可以了

go build 02_windows_daemon.go   #生成可执行文件

2、安装可执行文件即可:02_windows_daemon install,一定要用管理员身份远程cmd,然后切换到文件目录运行此条命令,此时我们会发现windows服务中有你注册的对应名称:
把golang程序注册成windows服务
启动就可以了,程序会生成一个.txt的文件夹,在C:\Windows\System32中可以看到

三、转载来源

经过验证,代码和教程无误,文章转载于:https://juejin.cn/post/7223305855923470393

正文完
 0
评论(没有评论)