updates for server files size limits

This commit is contained in:
onysd 2026-09-06 07:14:50 +03:00
parent b65ad60fe8
commit aeaf3f4596
8 changed files with 175 additions and 32 deletions

View file

@ -2573,17 +2573,35 @@ FROM photos p
// than physical when the same blob is attributed to more than one // than physical when the same blob is attributed to more than one
// document/photo, but a purged file with no file_blobs rows left correctly // document/photo, but a purged file with no file_blobs rows left correctly
// contributes 0 to both, never a stale non-zero "ghost" size). // contributes 0 to both, never a stale non-zero "ghost" size).
//
// Every field here deliberately EXCLUDES system/bundled content (owner_user_id
// = 0 on the documents/photos row -- the built-in sticker packs, emoji sets,
// default wallpapers, GIF catalog, and system-bot avatars this server seeds
// at every boot; see internal/app/files's Seed* functions). None of that is
// something an operator manages through storage retention/purge -- it isn't
// one of the Photo/Video/GIF/Music/Voice/File/Avatar categories those
// controls target, it's permanent server furniture -- so counting it here
// alongside real user uploads made every number on this page mean "user
// storage plus an unpredictable pile of bundled assets" instead of just
// answering "how much space are my users actually using". SystemBytes below
// is the one exception: it reports that excluded total separately, purely
// for an operator's own curiosity/disk-accounting, never folded into the
// other totals.
type StorageStatsRow struct { type StorageStatsRow struct {
PhysicalBytes int64 `json:"PhysicalBytes,string"` PhysicalBytes int64 `json:"PhysicalBytes,string"`
LogicalBytes int64 `json:"LogicalBytes,string"` LogicalBytes int64 `json:"LogicalBytes,string"`
// UnattributedBytes, DocumentCount, PhotoCount and AccountCount all count // SystemBytes is the physical size of excluded system/bundled content
// only items that still own real file_blobs bytes -- documents/photos // (owner_user_id = 0) -- shown separately so the gap between this and
// rows themselves are kept forever after a hard-retention purge (so a // what `docker exec ... mc du` or the MinIO console reports isn't a
// message can still render "here was a file"), so counting rows instead // mystery, but never added into PhysicalBytes/LogicalBytes/DocumentCount/
// of live bytes would keep growing even as the actual content becomes // PhotoCount/AccountCount above.
// physically empty, diverging further and further from PhysicalBytes SystemBytes int64 `json:"SystemBytes,string"`
// above. // DocumentCount, PhotoCount and AccountCount all count only items that
UnattributedBytes int64 `json:"UnattributedBytes,string"` // still own real file_blobs bytes -- documents/photos rows are
// deliberately kept forever after a hard-retention purge (so a message
// can still render "here was a file"), so counting rows instead of live
// bytes would keep growing even as the actual content becomes physically
// empty, diverging further and further from PhysicalBytes above.
DocumentCount int64 `json:"DocumentCount,string"` DocumentCount int64 `json:"DocumentCount,string"`
PhotoCount int64 `json:"PhotoCount,string"` PhotoCount int64 `json:"PhotoCount,string"`
AccountCount int64 `json:"AccountCount,string"` AccountCount int64 `json:"AccountCount,string"`
@ -2593,16 +2611,35 @@ type StorageStatsRow struct {
// StorageStats returns the admin panel's storage overview. // StorageStats returns the admin panel's storage overview.
func (s *readStore) StorageStats(ctx context.Context) (StorageStatsRow, error) { func (s *readStore) StorageStats(ctx context.Context) (StorageStatsRow, error) {
var stats StorageStatsRow var stats StorageStatsRow
if err := s.pool.QueryRow(ctx, `SELECT COALESCE(SUM(size), 0)::bigint FROM file_blobs`).Scan(&stats.PhysicalBytes); err != nil { // Physical usage dedups by (backend, object_key) like the unfiltered
// version used to, but only counts an object if at least one real user's
// (owner_user_id <> 0) document/photo still references it -- content
// shared between a system asset and a real upload (content-addressed
// storage, so only possible via a byte-for-byte coincidental duplicate)
// still counts, since a real user genuinely has that data stored.
if err := s.pool.QueryRow(ctx, `
SELECT COALESCE(SUM(size), 0)::bigint FROM (
SELECT DISTINCT ON (fb.backend, fb.object_key) fb.backend, fb.object_key, fb.size
FROM file_blobs fb
WHERE EXISTS (
SELECT 1 FROM documents d
WHERE d.owner_user_id <> 0
AND (fb.location_key = 'doc:' || d.id::text OR fb.location_key LIKE 'doc:' || d.id::text || ':%')
) OR EXISTS (
SELECT 1 FROM photos p
WHERE p.owner_user_id <> 0
AND (fb.location_key = 'photo:' || p.id::text OR fb.location_key LIKE 'photo:' || p.id::text || ':%')
)
) x`).Scan(&stats.PhysicalBytes); err != nil {
return StorageStatsRow{}, fmt.Errorf("sum physical blob bytes: %w", err) return StorageStatsRow{}, fmt.Errorf("sum physical blob bytes: %w", err)
} }
if err := s.pool.QueryRow(ctx, ` if err := s.pool.QueryRow(ctx, `
SELECT COALESCE(SUM(size), 0)::bigint FROM (`+perOwnerMediaSizeSQL+`) x`).Scan(&stats.LogicalBytes); err != nil { SELECT COALESCE(SUM(size), 0)::bigint FROM (`+perOwnerMediaSizeSQL+`) x WHERE owner_user_id <> 0`).Scan(&stats.LogicalBytes); err != nil {
return StorageStatsRow{}, fmt.Errorf("sum logical media bytes: %w", err) return StorageStatsRow{}, fmt.Errorf("sum logical media bytes: %w", err)
} }
if err := s.pool.QueryRow(ctx, ` if err := s.pool.QueryRow(ctx, `
SELECT COALESCE(SUM(size), 0)::bigint FROM (`+perOwnerMediaSizeSQL+`) x WHERE owner_user_id = 0`).Scan(&stats.UnattributedBytes); err != nil { SELECT COALESCE(SUM(size), 0)::bigint FROM (`+perOwnerMediaSizeSQL+`) x WHERE owner_user_id = 0`).Scan(&stats.SystemBytes); err != nil {
return StorageStatsRow{}, fmt.Errorf("sum unattributed media bytes: %w", err) return StorageStatsRow{}, fmt.Errorf("sum system media bytes: %w", err)
} }
// Documents/Photos/AccountCount all count only items that still own real // Documents/Photos/AccountCount all count only items that still own real
// file_blobs bytes -- documents/photos rows are deliberately kept forever // file_blobs bytes -- documents/photos rows are deliberately kept forever
@ -2610,9 +2647,10 @@ SELECT COALESCE(SUM(size), 0)::bigint FROM (`+perOwnerMediaSizeSQL+`) x WHERE ow
// a file"), so a plain count(*) would keep growing even as everything it // a file"), so a plain count(*) would keep growing even as everything it
// counts becomes physically empty, wildly diverging from PhysicalBytes // counts becomes physically empty, wildly diverging from PhysicalBytes
// above and making the overview page look broken/confusing rather than // above and making the overview page look broken/confusing rather than
// informative. // informative. owner_user_id <> 0 excludes system/bundled content -- see
// StorageStatsRow's doc comment.
if err := s.pool.QueryRow(ctx, ` if err := s.pool.QueryRow(ctx, `
SELECT count(*)::bigint FROM documents d WHERE EXISTS ( SELECT count(*)::bigint FROM documents d WHERE d.owner_user_id <> 0 AND EXISTS (
SELECT 1 FROM file_blobs fb SELECT 1 FROM file_blobs fb
WHERE fb.location_key = 'doc:' || d.id::text WHERE fb.location_key = 'doc:' || d.id::text
OR fb.location_key LIKE 'doc:' || d.id::text || ':%' OR fb.location_key LIKE 'doc:' || d.id::text || ':%'
@ -2620,7 +2658,7 @@ SELECT count(*)::bigint FROM documents d WHERE EXISTS (
return StorageStatsRow{}, fmt.Errorf("count documents: %w", err) return StorageStatsRow{}, fmt.Errorf("count documents: %w", err)
} }
if err := s.pool.QueryRow(ctx, ` if err := s.pool.QueryRow(ctx, `
SELECT count(*)::bigint FROM photos p WHERE EXISTS ( SELECT count(*)::bigint FROM photos p WHERE p.owner_user_id <> 0 AND EXISTS (
SELECT 1 FROM file_blobs fb SELECT 1 FROM file_blobs fb
WHERE fb.location_key = 'photo:' || p.id::text WHERE fb.location_key = 'photo:' || p.id::text
OR fb.location_key LIKE 'photo:' || p.id::text || ':%' OR fb.location_key LIKE 'photo:' || p.id::text || ':%'

View file

@ -23,7 +23,7 @@
})(); })();
</script> </script>
<script type="module" crossorigin src="/assets/index-DCHv80n3.js"></script> <script type="module" crossorigin src="/assets/index-DwjnA8dD.js"></script>
<link rel="stylesheet" crossorigin href="/assets/index-B7j2PW0q.css"> <link rel="stylesheet" crossorigin href="/assets/index-B7j2PW0q.css">
</head> </head>
<body> <body>

View file

@ -119,9 +119,8 @@ function StorageOverviewTab({ navigate }: { navigate: Navigate }) {
<Metric label={"Photos"} value={stats ? formatQuantity(stats.PhotoCount) : "-"} /> <Metric label={"Photos"} value={stats ? formatQuantity(stats.PhotoCount) : "-"} />
<Metric label={"Accounts with media"} value={stats ? formatQuantity(stats.AccountCount) : "-"} /> <Metric label={"Accounts with media"} value={stats ? formatQuantity(stats.AccountCount) : "-"} />
<Metric <Metric
label={"Unattributed"} label={"System/bundled content"}
value={stats ? formatBytes(stats.UnattributedBytes) : "-"} value={stats ? formatBytes(stats.SystemBytes) : "-"}
tone={stats && Number(stats.UnattributedBytes) > 0 ? "warn" : "neutral"}
/> />
</div> </div>

View file

@ -727,7 +727,7 @@ export type SharedDeviceGroupListResponse = {
export type StorageStatsResponse = { export type StorageStatsResponse = {
PhysicalBytes: string; PhysicalBytes: string;
LogicalBytes: string; LogicalBytes: string;
UnattributedBytes: string; SystemBytes: string;
DocumentCount: string; DocumentCount: string;
PhotoCount: string; PhotoCount: string;
AccountCount: string; AccountCount: string;

View file

@ -1611,6 +1611,7 @@ func run(logger *zap.Logger) error {
help.WithMapboxToken(cfg.MapboxToken), help.WithMapboxToken(cfg.MapboxToken),
help.WithEmailSignupEnable(cfg.EmailSignupEnable), help.WithEmailSignupEnable(cfg.EmailSignupEnable),
help.WithEmailSignupPhonePrefixes(cfg.EmailSignupPhonePrefixes), help.WithEmailSignupPhonePrefixes(cfg.EmailSignupPhonePrefixes),
help.WithMaxUploadFileBytes(cfg.StorageMaxUploadFileBytes),
help.WithAccountFreezeProvider(adminService), help.WithAccountFreezeProvider(adminService),
), ),
AppUpdates: appUpdateResolver, AppUpdates: appUpdateResolver,

View file

@ -82,6 +82,7 @@ type Service struct {
mapboxToken string mapboxToken string
emailSignupEnable bool emailSignupEnable bool
emailSignupPhonePrefixes []string emailSignupPhonePrefixes []string
maxUploadFileBytes int64
appConfigOnce sync.Once appConfigOnce sync.Once
appConfigCache domain.AppConfig appConfigCache domain.AppConfig
@ -129,6 +130,42 @@ func WithEmailSignupPhonePrefixes(prefixes []string) Option {
} }
} }
// maxUploadFilePartBytes mirrors internal/app/files.MaxUploadPartBytes (not
// imported to avoid pulling the whole files package into help just for one
// constant): the wire size of one upload chunk, used to convert
// TELESRV_STORAGE_MAX_UPLOAD_FILE_BYTES into the part-count unit
// upload_max_fileparts_default/_premium are declared in.
const maxUploadFilePartBytes = 524288
// WithMaxUploadFileBytes overrides the stock upload_max_fileparts_default/
// _premium app config keys (already in tdesktopDefaultAppConfigBase, values
// 4000/8000 -- 2000/4000 MB at 512KB/part) with TELESRV_STORAGE_MAX_UPLOAD_FILE_BYTES's
// part-count equivalent, when configured smaller than the protocol default.
// tdesktop's Data::PremiumLimits::uploadMaxDefault/Premium already read
// these two keys (data_premium_limits.cpp) and feed both the pre-upload size
// check and its warning dialog (localimageloader.cpp's FileSizeLimit/
// FileSizePremiumLimit, storage_media_prepare.cpp) -- so overriding the
// number here is enough to make an adapted client warn about (and reject)
// files above this self-hosted server's real ceiling, instead of only its
// own hardcoded ~2/4GB assumption. 0 (unlimited, the default) leaves the
// stock 4000/8000 values untouched.
func WithMaxUploadFileBytes(bytes int64) Option {
return func(s *Service) {
s.maxUploadFileBytes = bytes
}
}
// maxUploadFileParts converts a configured byte ceiling to the number of
// 512KB parts it takes to hold it, rounding up (a partial final part still
// needs a whole slot) and never below 1.
func maxUploadFileParts(bytes int64) int64 {
parts := (bytes + maxUploadFilePartBytes - 1) / maxUploadFilePartBytes
if parts < 1 {
return 1
}
return parts
}
// NewService 创建 help 服务。 // NewService 创建 help 服务。
func NewService(appConfigs store.AppConfigStore, countries store.CountryStore, opts ...Option) *Service { func NewService(appConfigs store.AppConfigStore, countries store.CountryStore, opts ...Option) *Service {
s := &Service{ s := &Service{
@ -143,12 +180,12 @@ func NewService(appConfigs store.AppConfigStore, countries store.CountryStore, o
return s return s
} }
func defaultAppConfig(mapboxToken string, emailSignupEnable bool, emailSignupPhonePrefixes []string) domain.AppConfig { func defaultAppConfig(mapboxToken string, emailSignupEnable bool, emailSignupPhonePrefixes []string, maxUploadFileBytes int64) domain.AppConfig {
jsonBytes := defaultAppConfigJSON(mapboxToken, emailSignupEnable, emailSignupPhonePrefixes) jsonBytes := defaultAppConfigJSON(mapboxToken, emailSignupEnable, emailSignupPhonePrefixes, maxUploadFileBytes)
return domain.AppConfig{Client: tdesktopClient, Hash: defaultAppConfigHashFor(mapboxToken, emailSignupEnable, emailSignupPhonePrefixes), JSON: jsonBytes} return domain.AppConfig{Client: tdesktopClient, Hash: defaultAppConfigHashFor(mapboxToken, emailSignupEnable, emailSignupPhonePrefixes, maxUploadFileBytes), JSON: jsonBytes}
} }
func defaultAppConfigJSON(mapboxToken string, emailSignupEnable bool, emailSignupPhonePrefixes []string) []byte { func defaultAppConfigJSON(mapboxToken string, emailSignupEnable bool, emailSignupPhonePrefixes []string, maxUploadFileBytes int64) []byte {
androidInvoiceBilling := `,"premium_playmarket_direct_currency_list":` + compatandroid.DirectInvoiceCurrenciesJSON() androidInvoiceBilling := `,"premium_playmarket_direct_currency_list":` + compatandroid.DirectInvoiceCurrenciesJSON()
base := tdesktopDefaultAppConfigBase + tdesktopNoForwardsAppConfig + androidInvoiceBilling base := tdesktopDefaultAppConfigBase + tdesktopNoForwardsAppConfig + androidInvoiceBilling
if emailSignupEnable { if emailSignupEnable {
@ -159,6 +196,11 @@ func defaultAppConfigJSON(mapboxToken string, emailSignupEnable bool, emailSignu
} }
} }
} }
if maxUploadFileBytes > 0 {
parts := strconv.FormatInt(maxUploadFileParts(maxUploadFileBytes), 10)
base = strings.Replace(base, `"upload_max_fileparts_default":4000`, `"upload_max_fileparts_default":`+parts, 1)
base = strings.Replace(base, `"upload_max_fileparts_premium":8000`, `"upload_max_fileparts_premium":`+parts, 1)
}
if mapboxToken == "" { if mapboxToken == "" {
return []byte(base + `}`) return []byte(base + `}`)
} }
@ -170,7 +212,7 @@ func defaultAppConfigJSON(mapboxToken string, emailSignupEnable bool, emailSignu
return []byte(base + `,"tdesktop_config_map":{"maps":` + tokenJSON + `,"geo":` + tokenJSON + `,"bmaps":` + tokenJSON + `,"bgeo":` + tokenJSON + `}}`) return []byte(base + `,"tdesktop_config_map":{"maps":` + tokenJSON + `,"geo":` + tokenJSON + `,"bmaps":` + tokenJSON + `,"bgeo":` + tokenJSON + `}}`)
} }
func defaultAppConfigHashFor(mapboxToken string, emailSignupEnable bool, emailSignupPhonePrefixes []string) int { func defaultAppConfigHashFor(mapboxToken string, emailSignupEnable bool, emailSignupPhonePrefixes []string, maxUploadFileBytes int64) int {
h := defaultAppConfigHash h := defaultAppConfigHash
if emailSignupEnable { if emailSignupEnable {
h += 1000003 // large odd offset so toggling the flag always changes the hash h += 1000003 // large odd offset so toggling the flag always changes the hash
@ -178,6 +220,9 @@ func defaultAppConfigHashFor(mapboxToken string, emailSignupEnable bool, emailSi
h += 1 + int(crc32.ChecksumIEEE([]byte(strings.Join(emailSignupPhonePrefixes, ",")))&0x3fffffff) h += 1 + int(crc32.ChecksumIEEE([]byte(strings.Join(emailSignupPhonePrefixes, ",")))&0x3fffffff)
} }
} }
if maxUploadFileBytes > 0 {
h += 1 + int(crc32.ChecksumIEEE([]byte(strconv.FormatInt(maxUploadFileBytes, 10)))&0x3fffffff)
}
if mapboxToken == "" { if mapboxToken == "" {
return h return h
} }
@ -243,9 +288,9 @@ func (s *Service) accountAppConfig(ctx context.Context, userID int64, base domai
func (s *Service) loadAppConfig(ctx context.Context) domain.AppConfig { func (s *Service) loadAppConfig(ctx context.Context) domain.AppConfig {
if s == nil { if s == nil {
return defaultAppConfig("", false, nil) return defaultAppConfig("", false, nil, 0)
} }
defaultCfg := defaultAppConfig(s.mapboxToken, s.emailSignupEnable, s.emailSignupPhonePrefixes) defaultCfg := defaultAppConfig(s.mapboxToken, s.emailSignupEnable, s.emailSignupPhonePrefixes, s.maxUploadFileBytes)
s.appConfigOnce.Do(func() { s.appConfigOnce.Do(func() {
if s.appConfigs == nil { if s.appConfigs == nil {
s.appConfigCache = defaultCfg s.appConfigCache = defaultCfg

View file

@ -119,6 +119,66 @@ func containsJSONCurrency(values []any, want string) bool {
return false return false
} }
func TestAppConfigKeepsStockUploadPartsLimitsByDefault(t *testing.T) {
cfg, notModified, err := (*Service)(nil).GetAppConfig(context.Background(), 0, 0)
if err != nil || notModified {
t.Fatalf("GetAppConfig = notModified %v err %v", notModified, err)
}
var decoded map[string]any
if err := json.Unmarshal(cfg.JSON, &decoded); err != nil {
t.Fatalf("app config json invalid: %v", err)
}
if got := decoded["upload_max_fileparts_default"]; got != float64(4000) {
t.Fatalf("upload_max_fileparts_default = %v, want 4000 (unconfigured limit)", got)
}
if got := decoded["upload_max_fileparts_premium"]; got != float64(8000) {
t.Fatalf("upload_max_fileparts_premium = %v, want 8000 (unconfigured limit)", got)
}
}
func TestAppConfigUsesConfiguredMaxUploadFileBytesAndHash(t *testing.T) {
// 500MiB / 512KiB-per-part = 1000 parts exactly.
svc := NewService(nil, nil, WithMaxUploadFileBytes(500*1024*1024))
cfg, notModified, err := svc.GetAppConfig(context.Background(), 0, 0)
if err != nil || notModified {
t.Fatalf("GetAppConfig = notModified %v err %v", notModified, err)
}
if cfg.Hash == defaultAppConfigHash {
t.Fatalf("hash = %d, want limit-specific hash", cfg.Hash)
}
if _, notModified, err := svc.GetAppConfig(context.Background(), 0, cfg.Hash); err != nil || !notModified {
t.Fatalf("GetAppConfig(hash) = notModified %v err %v, want notModified", notModified, err)
}
var decoded map[string]any
if err := json.Unmarshal(cfg.JSON, &decoded); err != nil {
t.Fatalf("app config json invalid: %v", err)
}
if got := decoded["upload_max_fileparts_default"]; got != float64(1000) {
t.Fatalf("upload_max_fileparts_default = %v, want 1000", got)
}
if got := decoded["upload_max_fileparts_premium"]; got != float64(1000) {
t.Fatalf("upload_max_fileparts_premium = %v, want 1000", got)
}
}
func TestMaxUploadFilePartsRoundsUpAndFloorsAtOne(t *testing.T) {
cases := []struct {
bytes int64
want int64
}{
{0, 1},
{1, 1},
{maxUploadFilePartBytes, 1},
{maxUploadFilePartBytes + 1, 2},
{500 * 1024 * 1024, 1000},
}
for _, c := range cases {
if got := maxUploadFileParts(c.bytes); got != c.want {
t.Errorf("maxUploadFileParts(%d) = %d, want %d", c.bytes, got, c.want)
}
}
}
func TestAppConfigOmitsMapboxTokenByDefault(t *testing.T) { func TestAppConfigOmitsMapboxTokenByDefault(t *testing.T) {
cfg, notModified, err := (*Service)(nil).GetAppConfig(context.Background(), 0, 0) cfg, notModified, err := (*Service)(nil).GetAppConfig(context.Background(), 0, 0)
if err != nil || notModified { if err != nil || notModified {