コード例 #1
0
ファイル: middleware.go プロジェクト: go-ndn/health
// InfluxDB writes interest processing time to influxDB.
//
// The data collected can be viewed with: SELECT "value" FROM :name WHERE "name" = ':interest_name'.
func InfluxDB(client influxdb.Client, db, name string, tags map[string]string) mux.Middleware {
	return func(next mux.Handler) mux.Handler {
		return mux.HandlerFunc(func(w ndn.Sender, i *ndn.Interest) {
			before := time.Now()
			next.ServeNDN(w, i)
			t := make(map[string]string)
			for k, v := range tags {
				t[k] = v
			}
			t["name"] = i.Name.String()
			pt, _ := influxdb.NewPoint(name, t, map[string]interface{}{
				"value": float64(time.Since(before)) / float64(time.Millisecond),
			}, time.Now())
			bp, _ := influxdb.NewBatchPoints(influxdb.BatchPointsConfig{
				Database: db,
			})
			bp.AddPoint(pt)

			err := client.Write(bp)
			if err != nil {
				log.Println(err)
				return
			}
		})
	}
}
コード例 #2
0
ファイル: main.go プロジェクト: farkhana/go-ndnfs
func FileServer(from, to string) (string, mux.Handler) {
	return from, mux.HandlerFunc(func(w ndn.Sender, i *ndn.Interest) {
		//content, err := ioutil.ReadFile(to + filepath.Clean(strings.TrimPrefix(i.Name.String(), from)))
		//if err != nil {
		//	return
		//}
		file, err := os.Open(to + filepath.Clean(strings.TrimPrefix(i.Name.String(), from)))
		if err != nil {
			return
		}

		fileInfo, _ := file.Stat()
		var fileSize int64 = fileInfo.Size()
		bytes := make([]byte, fileSize)

		buffer := bufio.NewReader(file)
		_, err = buffer.Read(bytes)

		w.SendData(&ndn.Data{
			Name:    i.Name,
			Content: bytes,
		})
	})
}