package com.njuzr.eaibackend.mapper;

import com.baomidou.mybatisplus.core.mapper.BaseMapper;
import com.njuzr.eaibackend.po.AssignmentClass;
import org.apache.ibatis.annotations.Delete;
import org.apache.ibatis.annotations.Insert;
import org.apache.ibatis.annotations.Mapper;
import org.apache.ibatis.annotations.Param;
import org.apache.ibatis.annotations.Select;

import java.util.List;

@Mapper
public interface AssignmentClassMapper extends BaseMapper<AssignmentClass> {

    @Insert("<script>" +
            "INSERT INTO assignment_class (assignment_id, class_id) VALUES " +
            "<foreach collection='classIds' item='classId' separator=','>" +
            "(#{assignmentId}, #{classId})" +
            "</foreach>" +
            "</script>")
    void insertBatch(@Param("assignmentId") Long assignmentId, @Param("classIds") List<Long> classIds);

    @Select("SELECT class_id FROM assignment_class WHERE assignment_id = #{assignmentId}")
    List<Long> selectClassIdsByAssignmentId(@Param("assignmentId") Long assignmentId);

    @Select("SELECT assignment_id FROM assignment_class WHERE class_id = #{classId}")
    List<Long> selectAssignmentIdsByClassId(@Param("classId") Long classId);

    @Delete("DELETE FROM assignment_class WHERE assignment_id = #{assignmentId}")
    void deleteByAssignmentId(@Param("assignmentId") Long assignmentId);
}