Вывод типа TypeScript / сужение задачи


В настоящее время я пытаюсь улучшить типы в некотором существующем коде. Мой код выглядит примерно так:

/* dispatcher.ts */
interface Message {
    messageType: string;
}

class Dispatcher<M extends Message> {    
    on<
        MessageType extends M["messageType"],
        SubMessage extends M & { messageType: MessageType }
    >(
        messageType: MessageType,
        handler: (message: SubMessage) => void
    ): void { }
}

/* messages.ts */
interface AddCommentMessage {
    messageType: "ADD_COMMENT";
    commentId: number;
    comment: string;
    userId: number;
}

interface PostPictureMessage {
    messageType: "POST_PICTURE";
    pictureId: number;
    userId: number;
}

type AppMessage = AddCommentMessage | PostPictureMessage;

/* app.ts */
const dispatcher = new Dispatcher<AppMessage>();

dispatcher.on("ADD_COMMENT", (message: AddCommentMessage ) => {
                                    /* ^^ REMOVE THIS TYPE HINT!*/
    console.log(message.comment);
});

Я хотел бы убрать необходимость явно сузить тип сообщения, передаваемого обработчику сообщений (где /*REMOVE THIS TYPE HINT!*/), таким образом, чтобы он правильно сужался до типа, который имеет соответствующий тип messageType (например, если messageType является "ADD_COMMENT", то message должен быть AddCommentMessage).

Если это невозможно прямо сейчас, пожалуйста, дайте мне знать. У меня такое впечатление, что это не так, но я не совсем конечно.
1 3

1 ответ:

Это невозможно, если только вы не хотите немного изменить код.

Ваш базовый интерфейс

interface Message {
    messageType: string;
}

Слишком общий, Я думаю, что messageType: string исключает любой вывод, основанный на значении messageType, и похоже, что его невозможно достаточно сузить в интерфейсе Dispatcher.

Если вы ограничиваете код только AppMessage и его потомками, вот пример того, как сделать typescript для вывода типов, которые вам нужны, руководствуясь строковыми литеральными типами (keyof AppMessageMap на самом деле является объединение типов строковых литералов "ADD_COMMENT" | "POST_PICTURE"):

/* dispatcher.ts */

class Dispatcher {    
    on<
        MessageType extends keyof AppMessageMap
    >(
        messageType: MessageType,
        handler: (message: AppMessageMap[MessageType] & {messageType: MessageType}) => void
    ): void { }
}

/* messages.ts */
interface AddCommentMessage {
    commentId: number;
    comment: string;
    userId: number;
}

interface PostPictureMessage {
    pictureId: number;
    userId: number;
}

interface AppMessageMap {
    "ADD_COMMENT": AddCommentMessage,
    "POST_PICTURE": PostPictureMessage
}
type AppMessage = AppMessageMap[keyof AppMessageMap];

/* app.ts */
const dispatcher = new Dispatcher();


dispatcher.on("ADD_COMMENT", (message) => {
    console.log(message.comment);
});

Я также удалил свойство messageType Из интерфейсов, чтобы избежать дублирования, я думаю, что тип пересечения в параметре handler достигает того же эффекта.