qunar.tc.qmq.task.store.impl.MessageClientStore Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of qmq-watchdog Show documentation

There is a newer version: 1.1.43

/*
 * Copyright 2018 Qunar, Inc.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package qunar.tc.qmq.task.store.impl;

import org.springframework.jdbc.core.JdbcTemplate;
import qunar.tc.qmq.task.database.DatasourceWrapper;
import qunar.tc.qmq.task.model.MsgQueue;
import qunar.tc.qmq.task.store.IMessageClientStore;

import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.Timestamp;
import java.util.Date;
import java.util.List;

/**
 * Created by zhaohui.yu
 * 16/3/28
 */
public class MessageClientStore implements IMessageClientStore {
    private static MsgQueue toMsgQueue(ResultSet rs, int rowNum) throws SQLException {
        return new MsgQueue(
                rs.getLong("id"),
                rs.getString("content"),
                rs.getInt("error"),
                rs.getTimestamp("update_time"));
    }

    @Override
    public List findErrorMsg(DatasourceWrapper dataSource, Date since) {
        return create(dataSource).query(dataSource.select_sql(), MessageClientStore::toMsgQueue, since);
    }

    @Override
    public void deleteByMessageId(DatasourceWrapper dataSource, long messageId) {
        create(dataSource).update(dataSource.delete_sql(), messageId);
    }

    @Override
    public void updateError(DatasourceWrapper dataSource, long messageId, int state) {
        create(dataSource).update(dataSource.error_sql(), state, new Timestamp(System.currentTimeMillis()), messageId);
    }

    protected JdbcTemplate create(DatasourceWrapper dataSource) {
        return new JdbcTemplate(dataSource.datasource());
    }
}