Skip to content

Refactor SDK index construction: removing mutable state. #8761

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 1 commit into from
May 15, 2025
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
13 changes: 3 additions & 10 deletions app/lib/search/dart_sdk_mem_index.dart
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,6 @@

import 'package:gcloud/service_scope.dart' as ss;
import 'package:logging/logging.dart';
import 'package:meta/meta.dart';

import 'backend.dart';
import 'sdk_mem_index.dart';
Expand All @@ -14,7 +13,6 @@ final _logger = Logger('search.dart_sdk_mem_index');
/// Results from these libraries are ranked with lower score and
/// will be displayed only if the query has the library name, or
/// there are not other results that could match the query.
@visibleForTesting
const dartSdkLibraryWeights = <String, double>{
'dart:html': 0.7,
};
Expand All @@ -37,14 +35,9 @@ SdkMemIndex? get dartSdkMemIndex =>
/// was an error parsing the file or building the index.
Future<SdkMemIndex?> createDartSdkMemIndex() async {
try {
final index = await SdkMemIndex.dart();
final content = await loadOrFetchSdkIndexJsonAsString(index.indexJsonUri);
await index.addDartdocIndex(DartdocIndex.parseJsonText(content));
index.updateWeights(
libraryWeights: dartSdkLibraryWeights,
apiPageDirWeights: {},
);
return index;
final content =
await loadOrFetchSdkIndexJsonAsString(SdkMemIndex.dartSdkIndexJsonUri);
return SdkMemIndex.dart(index: DartdocIndex.parseJsonText(content));
} catch (e, st) {
_logger.warning('Unable to load Dart SDK index.', e, st);
return null;
Expand Down
16 changes: 6 additions & 10 deletions app/lib/search/flutter_sdk_mem_index.dart
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,7 @@ import 'sdk_mem_index.dart';
/// regular packages. The selected libraries are unique to the index.json.
///
/// TODO: try to find a way to derive this list automatically.
const _allowedLibraries = <String>{
const flutterSdkAllowedLibraries = <String>{
'dart:ui',
'animation',
'cupertino',
Expand All @@ -32,7 +32,7 @@ const _allowedLibraries = <String>{
'flutter_web_plugins',
};

const _flutterApiPageDirWeights = <String, double>{
const flutterApiPageDirWeights = <String, double>{
'cupertino/CupertinoIcons': 0.25,
'material/Icons': 0.25,
};
Expand All @@ -54,14 +54,10 @@ SdkMemIndex? get flutterSdkMemIndex =>
/// api.flutter.dev and returns search results based on [SdkMemIndex].
Future<SdkMemIndex?> createFlutterSdkMemIndex() async {
try {
final index = SdkMemIndex.flutter();
final content = await loadOrFetchSdkIndexJsonAsString(index.indexJsonUri);
await index.addDartdocIndex(DartdocIndex.parseJsonText(content),
allowedLibraries: _allowedLibraries);
index.updateWeights(
libraryWeights: {},
apiPageDirWeights: _flutterApiPageDirWeights,
);
final content = await loadOrFetchSdkIndexJsonAsString(
SdkMemIndex.flutterSdkIndexJsonUri);
final index =
SdkMemIndex.flutter(index: DartdocIndex.parseJsonText(content));
return index;
} catch (e, st) {
_logger.warning('Unable to load Flutter SDK index.', e, st);
Expand Down
45 changes: 27 additions & 18 deletions app/lib/search/sdk_mem_index.dart
Original file line number Diff line number Diff line change
Expand Up @@ -8,6 +8,8 @@ import 'package:meta/meta.dart';
// ignore: implementation_imports
import 'package:pana/src/dartdoc/dartdoc_index.dart';
import 'package:path/path.dart' as p;
import 'package:pub_dev/search/dart_sdk_mem_index.dart';
import 'package:pub_dev/search/flutter_sdk_mem_index.dart';

import '../shared/versions.dart';
import 'search_service.dart';
Expand All @@ -23,39 +25,55 @@ class SdkMemIndex {
final _tokensPerLibrary = <String, TokenIndex<String>>{};
final _baseUriPerLibrary = <String, String>{};
final _descriptionPerLibrary = <String, String>{};
final _libraryWeights = <String, double>{};
final _apiPageDirWeights = <String, double>{};
final Map<String, double> _libraryWeights;
final Map<String, double> _apiPageDirWeights;

SdkMemIndex({
required String sdk,
required String? version,
required Uri baseUri,
required DartdocIndex index,
Set<String>? allowedLibraries,
Map<String, double>? libraryWeights,
Map<String, double>? apiPageDirWeights,
}) : _sdk = sdk,
_version = version,
_baseUri = baseUri;
_baseUri = baseUri,
_libraryWeights = libraryWeights ?? const {},
_apiPageDirWeights = apiPageDirWeights ?? const {} {
_addDartdocIndex(index, allowedLibraries);
}

static Future<SdkMemIndex> dart() async {
static SdkMemIndex dart({required DartdocIndex index}) {
return SdkMemIndex(
sdk: 'dart',
version: runtimeSdkVersion,
baseUri: Uri.parse('https://api.dart.dev/stable/latest/'),
index: index,
libraryWeights: dartSdkLibraryWeights,
);
}

factory SdkMemIndex.flutter() {
factory SdkMemIndex.flutter({required DartdocIndex index}) {
return SdkMemIndex(
sdk: 'flutter',
version: null,
baseUri: Uri.parse('https://api.flutter.dev/flutter/'),
index: index,
allowedLibraries: flutterSdkAllowedLibraries,
apiPageDirWeights: flutterApiPageDirWeights,
);
}

late final indexJsonUri = _baseUri.resolve('index.json');
static final dartSdkIndexJsonUri =
Uri.parse('https://api.dart.dev/stable/latest/index.json');
static final flutterSdkIndexJsonUri =
Uri.parse('https://api.flutter.dev/flutter/index.json');

Future<void> addDartdocIndex(
DartdocIndex index, {
void _addDartdocIndex(
DartdocIndex index,
Set<String>? allowedLibraries,
}) async {
) {
final textsPerLibrary = <String, Map<String, String>>{};
for (final f in index.entries) {
final library = f.qualifiedName?.split('.').first;
Expand Down Expand Up @@ -86,15 +104,6 @@ class SdkMemIndex {
}
}

/// Updates the non-default weight for libraries.
void updateWeights({
required Map<String, double> libraryWeights,
required Map<String, double> apiPageDirWeights,
}) {
_libraryWeights.addAll(libraryWeights);
_apiPageDirWeights.addAll(apiPageDirWeights);
}

List<SdkLibraryHit> search(
String query, {
int? limit,
Expand Down
7 changes: 4 additions & 3 deletions app/test/search/backend_test.dart
Original file line number Diff line number Diff line change
Expand Up @@ -13,9 +13,10 @@ import '../shared/test_services.dart';
void main() {
group('search backend', () {
testWithProfile('fetch SDK library description', fn: () async {
final index = await SdkMemIndex.dart();
final content = await loadOrFetchSdkIndexJsonAsString(index.indexJsonUri);
await index.addDartdocIndex(DartdocIndex.parseJsonText(content));
final content = await loadOrFetchSdkIndexJsonAsString(
SdkMemIndex.dartSdkIndexJsonUri);
final index =
SdkMemIndex.dart(index: DartdocIndex.parseJsonText(content));
expect(
index.getLibraryDescription('dart:async'),
'Support for asynchronous programming, with classes such as Future and Stream.',
Expand Down
3 changes: 1 addition & 2 deletions app/test/search/dartdoc_index_parsing_test.dart
Original file line number Diff line number Diff line change
Expand Up @@ -66,8 +66,7 @@ void main() {
expect(parserWithoutFirstEntry, originalWithoutFirstEntry);

// parsing into SDK index
final sdkMemIndex = SdkMemIndex.flutter();
await sdkMemIndex.addDartdocIndex(index);
final sdkMemIndex = SdkMemIndex.flutter(index: index);
final rs = sdkMemIndex.search('StatelessWidget');
expect(json.decode(json.encode(rs)), [
{
Expand Down
85 changes: 42 additions & 43 deletions app/test/search/result_combiner_test.dart
Original file line number Diff line number Diff line change
Expand Up @@ -31,49 +31,48 @@ void main() {
final combiner = SearchResultCombiner(
primaryIndex: primaryIndex,
dartSdkMemIndex: SdkMemIndex(
sdk: 'dart',
version: runtimeSdkVersion,
baseUri: Uri.parse('https://api.dart.dev/stable/$runtimeSdkVersion/'))
..addDartdocIndex(
DartdocIndex.fromJsonList([
{
'name': 'dart:core',
'qualifiedName': 'dart:core',
'href': 'dart-core/dart-core-library.html',
'kind': 8,
'overriddenDepth': 0,
'packageName': 'Dart'
},
{
'name': 'String',
'qualifiedName': 'dart:core.String',
'href': 'dart-core/String-class.html',
'kind': 3,
'overriddenDepth': 0,
'packageName': 'Dart',
'enclosedBy': {'name': 'dart:core', 'kind': 8}
},
{
'name': 'substring',
'qualifiedName': 'dart:core.String.substring',
'href': 'dart-core/String/substring.html',
'kind': 9,
'overriddenDepth': 0,
'packageName': 'Dart',
'enclosedBy': {'name': 'String', 'kind': 3}
},
{
// fake method for checking the package name matches
'name': 'stringutils',
'qualifiedName': 'dart:core.String.stringutils',
'href': 'dart-core/String/stringutils.html',
'kind': 9,
'overriddenDepth': 0,
'packageName': 'Dart',
'enclosedBy': {'name': 'String', 'kind': 3}
},
]),
),
sdk: 'dart',
version: runtimeSdkVersion,
baseUri: Uri.parse('https://api.dart.dev/stable/$runtimeSdkVersion/'),
index: DartdocIndex.fromJsonList([
{
'name': 'dart:core',
'qualifiedName': 'dart:core',
'href': 'dart-core/dart-core-library.html',
'kind': 8,
'overriddenDepth': 0,
'packageName': 'Dart'
},
{
'name': 'String',
'qualifiedName': 'dart:core.String',
'href': 'dart-core/String-class.html',
'kind': 3,
'overriddenDepth': 0,
'packageName': 'Dart',
'enclosedBy': {'name': 'dart:core', 'kind': 8}
},
{
'name': 'substring',
'qualifiedName': 'dart:core.String.substring',
'href': 'dart-core/String/substring.html',
'kind': 9,
'overriddenDepth': 0,
'packageName': 'Dart',
'enclosedBy': {'name': 'String', 'kind': 3}
},
{
// fake method for checking the package name matches
'name': 'stringutils',
'qualifiedName': 'dart:core.String.stringutils',
'href': 'dart-core/String/stringutils.html',
'kind': 9,
'overriddenDepth': 0,
'packageName': 'Dart',
'enclosedBy': {'name': 'String', 'kind': 3}
},
]),
),
flutterSdkMemIndex: null,
);

Expand Down
Loading