日志监控程序2（实现）

原创

梁十八 2019-03-10 16:45:49 博主文章分类：go基础/应用 ©著作权

©著作权归作者所有：来自51CTO博客作者梁十八的原创作品，请联系作者获取转载授权，否则将追究法律责任

package main

import (
	"fmt"
	"time"
	"os"
	"bufio"
	"io"
)

//读接口做抽象优化
type Reader interface {
	Read(rc chan []byte)
}

//写接口做抽象优化
type Writer interface {
	Write(wc chan []byte)
}

type LogProcess struct {
	rc chan []byte //读chan
	wc chan []byte //写chan
	read Reader
	write Writer
}

type ReadFromFile struct {
	path string //读取文件的路径

}

type WriteToInfluxDB struct {
	influxDBDsn string //influx data source

}

func (r *ReadFromFile) Read(rc chan []byte) {
	//读取模块
	//打开文件
	f, err := os.Open(r.path)
	if err != nil {
		panic(fmt.Sprintf("open file error:%s", err.Error()))
	}

	//从文件末尾开始逐行读取文件内容
	f.Seek(0, 2)//将文件字符指针移动到最后(0偏移量，2移动字符指针到末尾)
	rd := bufio.NewReader(f)
	for {
		line, err := rd.ReadBytes('\n') //读取文件内容直到遇到换行符（即读取一行内容）
		if err == io.EOF{  //读取到文件末尾也会返回err
			time.Sleep(time.Second)
			continue //当前循环不让继续向下执行
		}else if err != nil {
			panic(fmt.Sprintf("ReadBytes error:%s", err.Error()))
		}
		rc <- line[:len(line)-1]
	}

}

func (l *LogProcess) Process() {
	//解析模块

	for v := range l.rc {
		l.wc <- v
	}


}

func (w *WriteToInfluxDB) Write(wc chan []byte) {
	//写入模块
	for v := range wc {
		fmt.Println(string(v))
	}
}

func main() {
	r := &ReadFromFile{
		path: "D:/go_work_dir/logs/access.log",
	}
	w := &WriteToInfluxDB{
		influxDBDsn: "username&password..",
	}
	lp := &LogProcess{
		rc: make(chan []byte),
		wc: make(chan []byte),
		read: r,
		write: w,
	}

	go lp.read.Read(lp.rc)
	go lp.Process()
	go lp.write.Write(lp.wc)

	time.Sleep(10000 * time.Second)
}