日志监控程序1（抽象接口进行优化）

模拟写个日志监控程序：

package main

import (
	"fmt"
	"time"
	"strings"
)

type LogProcess struct {
	rc chan string //读chan
	wc chan string //写chan
	path string //读取文件的路径
	influxDBDsn string //influx data source
}

func (l *LogProcess) ReadFromFile() {
	//读取模块
	line := "message"
	l.rc <- line
}

func (l *LogProcess) Process() {
	//解析模块
	data := <- l.rc
	l.wc <- strings.ToUpper(data)
}

func (l *LogProcess) WriteToInfluxDB() {
	//写入模块
	fmt.Println("写入成功:", <-l.wc)
}

func main() {
	lp := &LogProcess{
		rc: make(chan string),
		wc: make(chan string),
		path: "D:/go_work_dir/logs/access.log",
		influxDBDsn: "username&password..",
	}

	go lp.ReadFromFile()
	go lp.Process()
	go lp.WriteToInfluxDB()

	time.Sleep(time.Second)
}

但是程序写死了，ReadFromFile只能从文件里读取；而WriteToInfluxDB只能写入InfluxDB... 如果是从队列，从标准输出读取呢？难道要定义更多模块？下面用到接口的设计方式解决

package main

import (
	"fmt"
	"time"
)

//接口做抽象优化
type Reader interface {
	Read(rc chan string)
}

//接口做抽象优化
type Writer interface {
	Write(wc chan string)
}



type LogProcess struct {
	rc chan string //读chan
	wc chan string //写chan
	read Reader
	write Writer
}

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

}

type WriteToInfluxDB struct {
	influxDBDsn string //influx data source

}

func (r *ReadFromFile) Read(rc chan string) {
	//读取模块
	line := "message"
	rc <- line
}

func (l *LogProcess) Process() {
	//解析模块
	data := <- l.rc

	l.wc <- data
}

func (w *WriteToInfluxDB) Write(wc chan string) {
	//写入模块
	fmt.Println("写入成功:", <-wc)
}

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

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

	time.Sleep(time.Second)
}