🌐 AI搜索 & 代理 主页
Skip to content
Merged
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
33 changes: 18 additions & 15 deletions src/main/java/graphql/execution/DataFetcherResult.java
Original file line number Diff line number Diff line change
Expand Up @@ -6,6 +6,8 @@
import graphql.Internal;
import graphql.PublicApi;
import graphql.schema.DataFetcher;
import org.jspecify.annotations.NullMarked;
import org.jspecify.annotations.Nullable;

import java.util.ArrayList;
import java.util.List;
Expand All @@ -32,14 +34,15 @@
* @param <T> The type of the data fetched
*/
@PublicApi
@NullMarked
public class DataFetcherResult<T> {

private final T data;
private final @Nullable T data;
private final List<GraphQLError> errors;
private final Object localContext;
private final Map<Object, Object> extensions;
private final @Nullable Object localContext;
private final @Nullable Map<Object, Object> extensions;

private DataFetcherResult(T data, List<GraphQLError> errors, Object localContext, Map<Object, Object> extensions) {
private DataFetcherResult(@Nullable T data, List<GraphQLError> errors, @Nullable Object localContext, @Nullable Map<Object, Object> extensions) {
this.data = data;
this.errors = ImmutableList.copyOf(assertNotNull(errors));
this.localContext = localContext;
Expand All @@ -49,7 +52,7 @@ private DataFetcherResult(T data, List<GraphQLError> errors, Object localContext
/**
* @return The data fetched. May be null.
*/
public T getData() {
public @Nullable T getData() {
return data;
}

Expand All @@ -72,7 +75,7 @@ public boolean hasErrors() {
*
* @return a local context object
*/
public Object getLocalContext() {
public @Nullable Object getLocalContext() {
return localContext;
}

Expand All @@ -88,7 +91,7 @@ public Object getLocalContext() {
* @see graphql.extensions.ExtensionsBuilder
* @see graphql.extensions.ExtensionsMerger
*/
public Map<Object, Object> getExtensions() {
public @Nullable Map<Object, Object> getExtensions() {
return extensions;
}

Expand All @@ -115,7 +118,7 @@ public DataFetcherResult<T> transform(Consumer<Builder<T>> builderConsumer) {
*
* @return a new instance with where the data value has been transformed
*/
public <R> DataFetcherResult<R> map(Function<T, R> transformation) {
public <R> DataFetcherResult<R> map(Function<@Nullable T, @Nullable R> transformation) {
return new Builder<>(transformation.apply(this.data))
.errors(this.errors)
.extensions(this.extensions)
Expand All @@ -135,10 +138,10 @@ public static <T> Builder<T> newResult() {
}

public static class Builder<T> {
private T data;
private Object localContext;
private @Nullable T data;
private @Nullable Object localContext;
private final List<GraphQLError> errors = new ArrayList<>();
private Map<Object, Object> extensions;
private @Nullable Map<Object, Object> extensions;

public Builder(DataFetcherResult<T> existing) {
data = existing.getData();
Expand All @@ -147,14 +150,14 @@ public Builder(DataFetcherResult<T> existing) {
extensions = existing.extensions;
}

public Builder(T data) {
public Builder(@Nullable T data) {
this.data = data;
}

public Builder() {
}

public Builder<T> data(T data) {
public Builder<T> data(@Nullable T data) {
this.data = data;
return this;
}
Expand All @@ -181,12 +184,12 @@ public boolean hasErrors() {
return !errors.isEmpty();
}

public Builder<T> localContext(Object localContext) {
public Builder<T> localContext(@Nullable Object localContext) {
this.localContext = localContext;
return this;
}

public Builder<T> extensions(Map<Object, Object> extensions) {
public Builder<T> extensions(@Nullable Map<Object, Object> extensions) {
this.extensions = extensions;
return this;
}
Expand Down
Loading