Method Stats

internal/store/postgres/postgres.go:255–289

Stats 采集当前数据库大小、Ping 延迟及各表行数。 调用方负责在后台 goroutine 中调用，不要在 HTTP handler 中直接调用。

(_ string)

// Stats 采集当前数据库大小、Ping 延迟及各表行数。
// 调用方负责在后台 goroutine 中调用，不要在 HTTP handler 中直接调用。
func (db *DB) Stats(_ string) DBStats {
	var s DBStats
	ctx := context.Background()

	// 查询数据库大小
	db.conn.QueryRow(ctx, "SELECT pg_database_size(current_database())").Scan(&s.FileSizeBytes)
	// WALSizeBytes、PageCount、PageSize、FreePages 字段设为 0（PostgreSQL 无对应概念）

	start := time.Now()
	db.conn.QueryRow(ctx, "SELECT 1").Scan(new(int))
	s.PingLatencyMs = float64(time.Since(start).Microseconds()) / 1000.0

	// 枚举 public schema 下的用户表并统计行数
	// 表名来自 pg_tables.tablename（系统表，非用户输入），安全。
	rows, err := db.conn.Query(ctx, "SELECT tablename FROM pg_tables WHERE schemaname = 'public' ORDER BY tablename")
	if err == nil {
		defer rows.Close()
		s.Tables = make(map[string]int64)
		for rows.Next() {
			var tbl string
			if rows.Scan(&tbl) == nil {
				var count int64
				// tbl 来自 pg_tables，只含小写字母、数字和下划线，无注入风险。
				db.conn.QueryRow(ctx, "SELECT COUNT(*) FROM "+tbl).Scan(&count) //nolint:gosec
				s.Tables[tbl] = count
			}
		}
		if err := rows.Err(); err != nil {
			// 统计失败不致命，直接忽略
			_ = err
		}
	}

	return s
}