메뉴 건너뛰기

검색창 닫기

Cloudera, BigData, Semantic IoT, Hadoop, NoSQL

Cloudera CDH/CDP 및 Hadoop EcoSystem, Semantic IoT등의 개발/운영 기술을 정리합니다. gooper@gooper.com로 문의 주세요.


mariadb ResultSet에서 데이타를 List<Map<String,String>>형태로 만들어서 리턴하는 소스(Collections.sort를 이용한 정렬 가능)

총관리자 2016.12.15 10:04 조회 수 : 4358

Mariadb에 접근하여 쿼리를 수행하고 필요시 정렬하여 List<Map<String, String>>에 담아서 리턴하는 메세드 예제


private final List<Map<String, String>> getResult (String query, String[] idxVals) throws Exception {
String db_server = ;
String db_port = ;
String db_name = ;
String db_user = ;
String db_pass = ;

Connection conn = null;
PreparedStatement pstmt = null;
ResultSet rs = null;

List<Map<String, String>> list = new ArrayList<Map<String, String>>();

try {
Class.forName("org.mariadb.jdbc.Driver");
conn = DriverManager.getConnection("jdbc:mariadb://" + db_server + ":" + db_port + "/" + db_name, db_user,  db_pass);

pstmt = conn.prepareStatement(query);
rs = pstmt.executeQuery();

int cnt = 0;

ResultSetMetaData md = rs.getMetaData();
int columns = md.getColumnCount();
while (rs.next()){
   HashMap<String,String> row = new HashMap<String, String>(columns);
   for(int i=1; i<=columns; i++){           
   row.put(md.getColumnName(i), rs.getObject(i).toString());
   }
   //log.debug("row("+(cnt++)+")  ===========>"+row.toString());
    list.add(row);
}

// 정렬(test)
        // Collections.sort(list, new MapStringComparator("rest_type"));
        //Collections.sort(list, new MapStringComparator("corner_id"));
        //Collections.sort(list, new MapFloatComparator("cnt"));
        
return list;
} catch (Exception e) {
throw e;
} finally {
if (rs != null)
try {
rs.close();
} catch (SQLException sqle) {
}
if (pstmt != null)
try {
pstmt.close();
} catch (SQLException sqle) {
}
if (conn != null) {
try {
conn.close();
} catch (SQLException sqle) {
}
}
}
}


*정렬을 위해서 사용되는 클래스(문자열)

class MapStringComparator implements Comparator<Map<String, String>> {
private final String key;

public MapStringComparator(String key) {
this.key = key;
}

@Override
public int compare(Map<String, String> first, Map<String, String> second) {
String firstValue =first.get(key);
        String secondValue = second.get(key);
        
         // 내림차순 정렬
             return firstValue.compareTo(secondValue);
}
}


*정렬을 위해서 사용되는 클래스(숫자)

class MapFloatComparator implements Comparator<Map<String, String>> {
private final String key;

public MapFloatComparator(String key) {
this.key = key;
}

@Override
public int compare(Map<String, String> first, Map<String, String> second) {
float firstValue = Float.valueOf(first.get(key));
         float secondValue = Float.valueOf(second.get(key));
         
// 내림차순 정렬
         if (firstValue > secondValue) {
             return -1;
         } else if (firstValue < secondValue) {
             return 1;
         } else /* if (firstValue == secondValue) */ {
             return 0;
         }
}
}


이 게시물을
이 글의 추천인 목록 목록
쓰기 태그
위로