feat(input): support video metrics (#76)

ykzts · Oct 19, 2021 · a02b37a · a02b37a
1 parent ebada30
commit a02b37a
Show file tree

Hide file tree

Showing 3 changed files with 70 additions and 3 deletions.
diff --git a/README.md b/README.md
@@ -12,6 +12,12 @@ Gather channel information from [YouTube](https://www.youtube.com/) channels.
  "UCnrgOD6G0y0_rcubQuICpTQ"
  ]
 
+ ## List of videos to monitor.
+ videos = [
+ "gjoHHYnXdqs",
+ "OoCsY8odmpM"
+ ]
+
  ## YouTube API key.
  # api_key = ""
 ```
@@ -26,10 +32,22 @@ Gather channel information from [YouTube](https://www.youtube.com/) channels.
  - subscribers (int)
  - videos (int)
  - views (int)
+- youtube_video
+ - tags:
+ - id - The ID of the video
+ - title - The title name
+ - fields:
+ - comments (int)
+ - dislikes (int)
+ - favorites (int)
+ - likes (int)
+ - views (int)
 
 ### Example Output
 
 ```plain
-youtube_channel,id=UCBR8-60-B28hp2BmDPdntcQ,title=YouTube subscribers=30400000i,videos=318i,views=2256168113i 1591499999079340600
-youtube_channel,id=UCnrgOD6G0y0_rcubQuICpTQ,title=InfluxData subscribers=2740i,videos=150i,views=252193i 1591499999079340600
+youtube_channel,id=UCBR8-60-B28hp2BmDPdntcQ,title=YouTube subscribers=32000000i,videos=409i,views=2509852022i 1634655873951240300
+youtube_channel,id=UCnrgOD6G0y0_rcubQuICpTQ,title=InfluxData videos=261i,views=489944i,subscribers=4520i 1634655873951240300
+youtube_video,id=gjoHHYnXdqs,title=InfluxDB:\ The\ time\ series\ data\ platform\ built\ for\ developers comments=0i,dislikes=1i,favorites=0i,likes=8i,views=377i 1634655874031796300
+youtube_video,id=OoCsY8odmpM,title=Intro\ to\ Time\ Series\ Databases\ &\ Data\ |\ Getting\ Started\ [1\ of\ 7] comments=0i,dislikes=20i,favorites=0i,likes=514i,views=74740i 1634655874031796300
 ```
diff --git a/etc/youtube-telegraf-plugin.conf b/etc/youtube-telegraf-plugin.conf
@@ -7,5 +7,11 @@
  "UCnrgOD6G0y0_rcubQuICpTQ"
  ]
 
+ ## List of videos to monitor.
+ videos = [
+ "gjoHHYnXdqs",
+ "OoCsY8odmpM"
+ ]
+
  ## Google API key.
  api_key = "${GOOGLE_API_KEY}"
diff --git a/plugins/inputs/youtube/youtube.go b/plugins/inputs/youtube/youtube.go
@@ -14,6 +14,7 @@ import (
 // YouTube - plugin main structure
 type YouTube struct {
  Channels []string `toml:"channels"`
+ Videos []string `toml:"videos"`
  APIKey string `toml:"api_key"`
  youtubeService *youtube.Service
 }
@@ -25,6 +26,12 @@ const sampleConfig = `
  "UCnrgOD6G0y0_rcubQuICpTQ"
  ]
 
+ ## List of videos to monitor.
+ videos = [
+ "gjoHHYnXdqs",
+ "OoCsY8odmpM"
+ ]
+
  ## Google API key.
  # api_key = ""
 `
@@ -44,7 +51,7 @@ func (y *YouTube) createYouTubeService(ctx context.Context) (*youtube.Service, e
  return youtube.NewService(ctx, option.WithAPIKey(y.APIKey))
 }
 
-// Gather GitHub Metrics
+// Gather YouTube Metrics
 func (y *YouTube) Gather(acc telegraf.Accumulator) error {
  ctx := context.Background()
 
@@ -76,6 +83,25 @@ func (y *YouTube) Gather(acc telegraf.Accumulator) error {
  acc.AddFields("youtube_channel", fields, tags, now)
  }
 
+ call2 := y.youtubeService.Videos.
+ List([]string{"snippet", "statistics"}).
+ Id(strings.Join(y.Videos, ",")).
+ MaxResults(50)
+
+ resp2, err := call2.Do()
+ if err != nil {
+ return err
+ }
+
+ now2 := time.Now()
+
+ for _, item := range resp2.Items {
+ tags := getVideoTags(item)
+ fields := getVideoFields(item)
+
+ acc.AddFields("youtube_video", fields, tags, now2)
+ }
+
  return nil
 }
 
@@ -94,6 +120,23 @@ func getFields(channelInfo *youtube.Channel) map[string]interface{} {
  }
 }
 
+func getVideoTags(videoInfo *youtube.Video) map[string]string {
+ return map[string]string{
+ "id": videoInfo.Id,
+ "title": videoInfo.Snippet.Title,
+ }
+}
+
+func getVideoFields(videoInfo *youtube.Video) map[string]interface{} {
+ return map[string]interface{}{
+ "comments": videoInfo.Statistics.CommentCount,
+ "dislikes": videoInfo.Statistics.DislikeCount,
+ "favorites": videoInfo.Statistics.FavoriteCount,
+ "likes": videoInfo.Statistics.LikeCount,
+ "views": videoInfo.Statistics.ViewCount,
+ }
+}
+
 func init() {
  inputs.Add("youtube", func() telegraf.Input {
  return &YouTube{}