示例#1
0
文件: main.go 项目: bcandrea/gotail
func main() {
	flag.StringVar(&fname, "file", "testing.log", "File to tail")
	flag.Parse()

	log.Println("Running Benchmarks")
	var concurrency int = runtime.NumCPU()
	var rowcount = 5000000          // number of rows to write
	runtime.GOMAXPROCS(concurrency) // number of writer processes
	var row int

	_ = os.Remove(fname)
	createFile("")

	f, err := os.OpenFile(fname, os.O_APPEND|os.O_WRONLY, 0600)
	if err != nil {
		log.Fatalln(err)
	}

	tail, err := gotail.NewTail(fname, gotail.Config{Timeout: 10})
	time.Sleep(10 * time.Millisecond)

	for i := 0; i < concurrency; i++ {
		go func(i int) {
			log.Printf("Spawned %d Workers\n", i)

			for row <= rowcount {
				body := fmt.Sprintf("%d Worker doing the write.  %d iteration is being written.\n", i, row)
				writeContents(f, body)
				row++
			}
		}(i)
	}

	count := 0
	startTime := time.Now()

	go func() {
		for {
			startTime := time.Now()
			countNow := count
			time.Sleep(5 * time.Second)
			duration := time.Since(startTime).Seconds()
			newCount := count

			log.Printf("%d processed at %0.2f rows/sec\n", count, float64(newCount-countNow)/duration)
		}
	}()

	for _ = range tail.Lines {
		count++
		if count == rowcount {
			break
		}
	}

	duration := time.Since(startTime).Seconds()
	fmt.Printf("%d rows processed in %0.4f seconds, at the rate of %0.4f rows/s\n", count, duration, float64(count)/duration)
	removeFile()
}
示例#2
0
func main() {
	// Limit to 1 proc to avoid running intensively for no reason
	runtime.GOMAXPROCS(1)

	flag.StringVar(&fname, "file", "", "File to tail. Ex: ./tailsocket -file=\"/var/log/my_awesome_log\"")
	flag.Parse()

	http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
		index, err := Asset("index.html")
		check(err)

		// Prints the bundled index.html directly to the connection
		fmt.Fprintf(w, "%s", index)
	})

	// The handler for the websocket
	http.HandleFunc("/v1/ws", func(w http.ResponseWriter, r *http.Request) {
		conn, _ := upgrader.Upgrade(w, r, nil)

		// Listens to messages from the client and closes the connection when necessary
		go func(conn *websocket.Conn) {
			for {
				// Prevent using the CPU massively just to listen to close request
				time.Sleep(1 * time.Second)
				_, _, err := conn.ReadMessage()
				if err != nil {
					conn.Close()
				}
			}
		}(conn)

		// Sends data back to the client
		go func(conn *websocket.Conn) {
			tail, err := gotail.NewTail(fname, gotail.Config{Timeout: 2})
			check(err)
			for line := range tail.Lines {
				conn.WriteJSON(Log{
					LogLine: html.EscapeString(line),
				})
				// Delay sending the message back, again to provide CPU relief
				time.Sleep(500 * time.Millisecond)
			}
		}(conn)
	})

	http.ListenAndServe(":3000", nil)
}