aboutsummaryrefslogtreecommitdiff
path: root/juick-server/src/main/java/com/juick/service/search/SphinxSearchServiceImpl.java
blob: de0a101f1b3fcb4f11aa6bdc48e2a65b2dd33fa1 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
package com.juick.service.search;

import org.apache.commons.lang3.StringUtils;
import org.springframework.jdbc.core.JdbcTemplate;
import org.springframework.jdbc.core.namedparam.MapSqlParameterSource;
import org.springframework.jdbc.core.namedparam.NamedParameterJdbcTemplate;
import org.springframework.util.Assert;

import javax.sql.DataSource;
import java.util.Collections;
import java.util.List;

/**
 * Created by aalexeev on 11/18/16.
 */

/* Note
 * Don't use any spring's component annotation (such as @Repository, @Service, @Component, etc).
 * This class directly used by spring's search configuration class
 */
public class SphinxSearchServiceImpl implements SearchService {
    private static final int DEFAULT_MAX_RESULT = 25;

    private final NamedParameterJdbcTemplate namedParameterSearchJdbcTemplate;

    private int maxResult = DEFAULT_MAX_RESULT;


    public SphinxSearchServiceImpl(JdbcTemplate searchJdbcTemplate) {
        Assert.notNull(searchJdbcTemplate);
        this.namedParameterSearchJdbcTemplate = new NamedParameterJdbcTemplate(searchJdbcTemplate);
    }

    public SphinxSearchServiceImpl(DataSource searchDataSource) {
        Assert.notNull(searchDataSource);
        this.namedParameterSearchJdbcTemplate = new NamedParameterJdbcTemplate(searchDataSource);
    }

    @Override
    public List<Integer> searchInAllMessages(final String searchString, final int messageIdBefore) {
        if (StringUtils.isBlank(searchString))
            return Collections.emptyList();

        MapSqlParameterSource sqlParameterSource = new MapSqlParameterSource()
                .addValue("search", searchString)
                .addValue("before", messageIdBefore)
                .addValue("limit", maxResult);

        return namedParameterSearchJdbcTemplate.queryForList(
                "SELECT id AS message_id FROM messages WHERE MATCH(:search) " +
                        (messageIdBefore > 0 ?
                                " AND id < :before " : "") +
                        " ORDER BY id DESC LIMIT :limit",
                sqlParameterSource,
                Integer.class);
    }

    @Override
    public List<Integer> searchByStringAndUser(final String searchString, final int userId, int messageIdBefore) {
        if (StringUtils.isBlank(searchString))
            return Collections.emptyList();

        MapSqlParameterSource sqlParameterSource = new MapSqlParameterSource()
                .addValue("search", searchString)
                .addValue("userId", userId)
                .addValue("before", messageIdBefore)
                .addValue("limit", maxResult);

        return namedParameterSearchJdbcTemplate.queryForList(
                "SELECT id AS message_id FROM messages WHERE user_id = :userId AND MATCH(:search) " +
                        (messageIdBefore > 0 ?
                                " AND id < :before " : "") +
                        " ORDER BY id DESC LIMIT :limit",
                sqlParameterSource,
                Integer.class);
    }

    @Override
    public void setMaxResult(int maxResult) {
        if (maxResult <= 0)
            throw new IllegalArgumentException("maxResult value (" + maxResult + ") must be greater then 0");

        this.maxResult = maxResult;
    }
}