Mybatis源碼分析:@Mapkey的使用

@Mapkey的使用

　　在多值查詢的時候,通常要把方法返回類型設置為List<Dto>類型,Mybatis為我們提供了另一種解決方式,通過K-V的形式將查詢結果保存在Map中,這種實現方式只需要在方法上標注為@Mapkey即可。

　　如下代碼:查詢一個學生表,包含id,name,age三個字段，使用@mapkey將id號作為K,查詢結果作為V.使用@Mapkey時最好將K設置為唯一的,否則后續的結果將會覆蓋已查詢到的結果,如果設置的@Mapkey設置的value為空或者一個跟查詢結果不匹配的字段,那么查到的記錄K成了null,查詢結果始終都是最后一條。這個可以通過源碼中的DefaultMapResultHandler<K, V>類得出結果,代碼見最后， final K key = (K) mo.getValue(mapKey);這句代碼就是將@Mapkey中指定的類型字段查詢出對應的value值,再將value值作為最終結果的key。

package com.zzz.mybatis.mapper;

import java.util.List;
import java.util.Map;

import org.apache.ibatis.annotations.Flush;
import org.apache.ibatis.annotations.MapKey;
import org.apache.ibatis.annotations.Select;

public interface SqlMapper {
    
    public List<Map<String, Object>> list();
    
    @Flush
    public void flush();
    
    @MapKey(value="id")
    @Select("select id,name,age from student")
    public Map<String,Map<String,String>> listByMapkey();
}

 

相關測試類

package com.zzz.mybatis.service;

import java.util.List;
import java.util.Map;

import org.apache.ibatis.session.SqlSession;
import org.apache.ibatis.session.SqlSessionFactory;
import org.apache.ibatis.session.SqlSessionFactoryBuilder;
import org.junit.Before;
import org.junit.Test;

import com.zzz.mybatis.mapper.SqlMapper;
import com.zzz.mybatis.mapper.StudentMapper;

public class StudentMapperTest {
    private SqlSession session;
    
    private  StudentMapper mapper;
    
    @Before
    public void before() {
        SqlSessionFactoryBuilder builder =new SqlSessionFactoryBuilder();
        SqlSessionFactory sessionFactory=builder.build(Class.class.getResourceAsStream("/mybatis-config.xml"));
        this.session=sessionFactory.openSession(false);
    }
    @Test
    public void getStuByMapkey() {
        mapper=session.getMapper(StudentMapper.class);
        mapper.flush();
        Map<String, Map<String, String>> rs=mapper.listByMapkey();
        System.out.println(rs.toString());
    }
}

測試結果
{1={name=zhangsan, id=1, age=20}, 2={name=lisi, id=2, age=30}, 3={name=wangwu, id=3, age=40}}

 

DefaultMapResultHandler<K, V>

  @Override
  public void handleResult(ResultContext<? extends V> context) {
    final V value = context.getResultObject();
    final MetaObject mo = MetaObject.forObject(value, objectFactory, objectWrapperFactory, reflectorFactory);
    // TODO is that assignment always true?
    final K key = (K) mo.getValue(mapKey);
    mappedResults.put(key, value);
  }