Netflix DGS 订阅

问题描述

我正在关注 DGS subscriptions 的文档,没有收到任何错误,但也没有取回任何数据。

设置非常简单。在 schema.graphqls 文件中,我定义了订阅

type Subscription {
    ratings: rating
}

type rating {
    stars: Int
}

Java 代码如下:

@DgsComponent
public class SubscriptionDataFetcher {
    @DgsData(parentType = "Subscription",field = "ratings")
    public Publisher<rating> ratings() {
        return Flux.interval(Duration.ofSeconds(1)).map(t -> new rating(4));
    }
}

如果我现在将一个 websocket 连接到我的后端,它连接得很好但没有按预期返回任何数据(不管我怎么做,也尝试使用 JavaScript,也连接得很好但没有获取任何数据)。

例如使用 curl 连接(但使用 JavaScript 结果相同,连接但没有数据):

curl -o - --http1.1 \
    --include \
    --no-buffer \
    --header "Connection: Upgrade" \
    --header "Upgrade: websocket" \
    --header "Host: localhost:8443" \
    --header "Origin: https://localhost:8443" \
    --header "Sec-WebSocket-Key: SGVsbG8sIHdvcmxkIQ==" \
    --header "Sec-WebSocket-Version: 13" \
    https://localhost:8443/subscriptions\?query\=ewoicxvlcnkiOiAic3Vic2NyaXB0aW9uIHsgiHN0b2NrcyB7bmFtZX0gfSIKfQ==

我也尝试通过 Graphiql 接口连接,但出现错误

subscription {
  ratings {
    stars
  }
}

错误信息:

{
  "message": "response is not defined","stack": "ReferenceError: response is not defined\n    at https://localhost:8443/graphiql:46:35\n    at async Object.onRun (https://unpkg.com/graphiql/graphiql.min.js:1:540500)"
}

链接中的示例中我还不清楚的另一件事是如何实际管理订阅。例如,假设我想在发生突变时发布通知。任何关于如何使用 Netflix DGS 框架完成的指示也将不胜感激。

解决方法

不幸的是,DGS 自带的 graphiql 接口似乎不能正确处理订阅 - 如果您将 playground-spring-boot-starter 添加到您的项目中,/playground 将提供一个更完善的工具,它完全支持订阅.如果您在那里尝试订阅它应该可以工作(假设您已经按照文档添加了 graphql-dgs-subscriptions-websockets-autoconfigure)。

关于您关于如何在发生突变时发布通知的第二个问题 - 不幸的是文档中缺少这一点,但 examples repo 中有一个示例。

我在这里稍微简化了这个例子。如果您想支持这样的订阅和变更:

@DgsSubscription
public Publisher<Review> reviewAdded() {
    return reviewsService.getReviewsPublisher();
}

@DgsMutation
public Review addReview(@InputArgument SubmittedReview review) {
    return reviewsService.saveReview(review);
}

在您的服务中,您将创建一个 Flux(返回给订阅者)并保留对其发射器的引用,以便在发生突变时调用 next。

@Service
public class ReviewsService {

    private FluxSink<Review> reviewsStream;
    private ConnectableFlux<Review> reviewsPublisher;

    @PostConstruct
    public void init() {
        Flux<Review> publisher = Flux.create(emitter -> {
            reviewsStream = emitter;
        });

        reviewsPublisher = publisher.publish();
        reviewsPublisher.connect();
    }

    public Review saveReview(SubmittedReview reviewInput) {
        Review review = Review.newBuilder()
                .username(reviewInput.getUsername())
                .starScore(reviewInput.getStarScore())
                .submittedDate(OffsetDateTime.now()).build();

        // Save to the database,etc.

        reviewsStream.next(review); // publishes the review to subscribers
        return review;
    }

    public Publisher<Review> getReviewsPublisher() {
        return reviewsPublisher;
    }
}