rtc_stats.go

package kohaku

import (
	"encoding/json"

	"github.com/labstack/echo/v4"
	zlog "github.com/rs/zerolog/log"
)

type RTCStats struct {
	Timestamp float64 `json:"timestamp"`
	ID        string  `json:"id"`
	Type      string  `json:"type"`
}

// type SoraRtcStats struct {
// 	Timestamp time.Time `json:"timestamp"`
//
// 	Label    string
// 	Version  string
// 	NodeName string
//
// 	Multistream bool
// 	Simulcast   bool
// 	Spotlight   bool
//
// 	Role         string
// 	ChannelID    string
// 	SessionID    string
// 	ClientID     string
// 	ConnectionID string
//
// 	RtcStatsTimestamp float64
// 	RtcStatsType      string
// 	RtcStatsID        string
// 	RtcStatsData      json.RawMessage
// }

func (s *Server) rtcStats(c echo.Context, stats soraConnectionStats) error {
	// TODO: Insert する場合は batch が良さそう
	for _, v := range stats.Stats {
		// timestamp と id と type のみを取り出す
		rtcStats := new(RTCStats)
		if err := json.Unmarshal(v, &rtcStats); err != nil {
			return err
		}
		// TODO: ここで timestamp と id と type が空の場合はエラーにする

		zlog.Debug().
			Time("timestamp", stats.Timestamp).
			Str("channel_id", stats.ChannelID).
			Str("connection_id", stats.ConnectionID).
			Float64("rtc_stats_timestamp", rtcStats.Timestamp).
			Str("rtc_stats_type", rtcStats.Type).Send()

		// マイクロ秒まで入れたいので ClickHouse DateTime64 にあわせて {秒}.{マイクロ秒} 形式に変換する
		timestamp := float64(stats.Timestamp.UTC().UnixMicro()) / 1e6
		rtcStatsTimestamp := (rtcStats.Timestamp / 1000)

		// INSERT する
		if err := s.conn.Exec(c.Request().Context(), `
			INSERT INTO sora_rtc_stats (
				timestamp,
				version, label, node_name,
				multistream, simulcast, spotlight,
				role,
				channel_id, session_id, client_id, connection_id,
				rtc_stats_timestamp, rtc_stats_type, rtc_stats_id,
				rtc_stats_data
			) VALUES (
				?,
				?, ?, ?,
				?, ?, ?,
				?,
				?, ?, ?, ?,
				?, ?, ?,
				?
			)`,
			timestamp,
			stats.Version, stats.Label, stats.NodeName,
			stats.Multistream, stats.Simulcast, stats.Spotlight,
			stats.Role,
			stats.ChannelID, stats.SessionID, stats.ClientID, stats.ConnectionID,
			rtcStatsTimestamp, rtcStats.Type, rtcStats.ID,
			// stats は json.RawMessage なので string() で囲むだけ
			string(v),
		); err != nil {
			return err
		}
	}

	return nil
}