AutoCreateMealOrderTask.java 5.68 KB
package com.infoloop.tianting.logic.task;

import com.infoloop.tianting.clientinventoryservice.SingleClientRpcResponse;
import com.infoloop.tianting.context.LoginContextHolder;
import com.infoloop.tianting.enums.LoginSourceEnum;
import com.infoloop.tianting.mealorderservice.MealOrderCreation;
import com.infoloop.tianting.mealorderservice.MealOrderOrderMethodEnum;
import com.infoloop.tianting.mealorderservice.MenuSchedule;
import com.infoloop.tianting.mealorderservice.SingleDinerRpcResponse;
import com.infoloop.tianting.mealorderservice.SingleGradeClassRpcResponse;
import com.infoloop.tianting.mealorderservice.SingleMealMenuRecordRpcResponse;
import com.infoloop.tianting.mealorderservice.SingleMealOrderRpcResponse;
import com.infoloop.tianting.service.client.ClientInventoryServiceRpcClient;
import com.infoloop.tianting.service.client.MealOrderServiceRpcClient;
import com.infoloop.tianting.utils.DateUtil;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.scheduling.annotation.Scheduled;
import org.springframework.stereotype.Component;

import java.time.DayOfWeek;
import java.time.Instant;
import java.time.LocalDate;
import java.time.LocalTime;
import java.util.ArrayList;
import java.util.function.Function;
import java.util.stream.Collectors;

@Slf4j
@Component
@RequiredArgsConstructor
public class AutoCreateMealOrderTask {

    private final MealOrderServiceRpcClient mealOrderServiceRpcClient;

    private final ClientInventoryServiceRpcClient clientInventoryServiceRpcClient;

    @Scheduled(cron = "0 0 13 ? * WED")
    public void executeTask() {
        int enterpriseId = 449;
        final var today = LocalDate.now();
        final var zonedDateTime = today.with(DayOfWeek.MONDAY).atStartOfDay().atZone(DateUtil.CHINA_ZONE);
        final var startTimestamp = zonedDateTime.toInstant().toEpochMilli();
        final var endTimestamp = today.with(DayOfWeek.SUNDAY).atTime(LocalTime.MAX).atZone(DateUtil.CHINA_ZONE).toInstant().toEpochMilli();
        final var mealOrders = mealOrderServiceRpcClient.queryMealOrdersByCondition(449, null, startTimestamp, endTimestamp);
        final var dinerIds = mealOrders.stream().map(SingleMealOrderRpcResponse::getDinerId).toList();
        final var allDiners = mealOrderServiceRpcClient.queryDinersByCondition(enterpriseId, null, null, null);
        final var notReservedDiners = allDiners.stream().filter(diner -> !dinerIds.contains(diner.getId())).toList();
        final var clientIds = notReservedDiners.stream().map(SingleDinerRpcResponse::getClientId).distinct().toList();
        final var clientMap = clientInventoryServiceRpcClient.getClientsByIds(enterpriseId, clientIds).getClientsList().stream().collect(Collectors.toMap(SingleClientRpcResponse::getId, Function.identity()));
        final var gradeClassMap = mealOrderServiceRpcClient.getGradeClassesByIds(enterpriseId, notReservedDiners.stream().map(SingleDinerRpcResponse::getClassId).distinct().toList()).stream().collect(Collectors.toMap(SingleGradeClassRpcResponse::getId, Function.identity()));
        final var latestMealMenuRecordMap = mealOrderServiceRpcClient.queryLatestMealMenuRecordsByClientIds(enterpriseId, clientIds).stream().collect(Collectors.toMap(SingleMealMenuRecordRpcResponse::getClientId, Function.identity()));
        final var creations = new ArrayList<MealOrderCreation>();
        for (var diner : notReservedDiners) {
            final var mealMenuRecordRpcResponse = latestMealMenuRecordMap.get(diner.getClientId());
            if (mealMenuRecordRpcResponse == null) {
                continue;
            }
            final var singleClientRpcResponse = clientMap.get(diner.getClientId());
            if (singleClientRpcResponse == null) {
                continue;
            }
            final var singleGradeClassRpcResponse = gradeClassMap.get(diner.getClassId());
            if (singleGradeClassRpcResponse == null) {
                continue;
            }
            MealOrderCreation.newBuilder()
                    .setClientId(diner.getClientId())
                    .setMenuId(mealMenuRecordRpcResponse.getId())
                    .setAccountId(0)
                    .setOrderMethod(MealOrderOrderMethodEnum.MEAL_ORDER_METHOD_SYSTEM)
                    .setDinerId(diner.getId())
                    .setDinerName(diner.getName())
                    .setClientName(singleClientRpcResponse.getName())
                    .setGradeName(singleGradeClassRpcResponse.getGradeName())
                    .setClassName(singleGradeClassRpcResponse.getClassName())
                    .setStudentNo(diner.getStudentNo())
                    .setReserveDate(Instant.now().toEpochMilli())
                    .addMeals(MenuSchedule.newBuilder().setDate(zonedDateTime.format(DateUtil.YYYY_MM_DD_LEFT_SYMBOL)).setMenu("A").build())
                    .addMeals(MenuSchedule.newBuilder().setDate(zonedDateTime.plusDays(1).format(DateUtil.YYYY_MM_DD_LEFT_SYMBOL)).setMenu("A").build())
                    .addMeals(MenuSchedule.newBuilder().setDate(zonedDateTime.plusDays(2).format(DateUtil.YYYY_MM_DD_LEFT_SYMBOL)).setMenu("A").build())
                    .addMeals(MenuSchedule.newBuilder().setDate(zonedDateTime.plusDays(3).format(DateUtil.YYYY_MM_DD_LEFT_SYMBOL)).setMenu("A").build())
                    .addMeals(MenuSchedule.newBuilder().setDate(zonedDateTime.plusDays(4).format(DateUtil.YYYY_MM_DD_LEFT_SYMBOL)).setMenu("A").build())
                    .build();
        }
        final var loginInfo = LoginContextHolder.LoginInfo.builder()
                .id(0)
                .enterpriseId(enterpriseId)
                .loginSource(LoginSourceEnum.MINI_PROGRAM)
                .build();
        mealOrderServiceRpcClient.batchCreateMealOrders(loginInfo, creations);
    }
}