io.trino.gateway.ha.persistence.dao.QueryHistoryDao Maven / Gradle / Ivy

Go to download
/*
 * 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 io.trino.gateway.ha.persistence.dao;

import org.jdbi.v3.core.mapper.MapMapper;
import org.jdbi.v3.sqlobject.customizer.Bind;
import org.jdbi.v3.sqlobject.customizer.Define;
import org.jdbi.v3.sqlobject.statement.SqlQuery;
import org.jdbi.v3.sqlobject.statement.SqlUpdate;
import org.jdbi.v3.sqlobject.statement.UseRowMapper;

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

public interface QueryHistoryDao
{
    @SqlQuery("""
            SELECT * FROM query_history
            ORDER BY created DESC
            LIMIT 2000
            """)
    List findRecentQueries();

    @SqlQuery("""
            SELECT * FROM query_history
            WHERE user_name = :userName
            ORDER BY created DESC
            LIMIT 2000
            """)
    List findRecentQueriesByUserName(String userName);

    @SqlQuery("""
            SELECT backend_url FROM query_history
            WHERE query_id = :queryId
            """)
    String findBackendUrlByQueryId(String queryId);

    @SqlQuery("""
            SELECT * FROM query_history
            WHERE 1 = 1 
            ORDER BY created DESC
            LIMIT :limit
            OFFSET :offset
            """)
    List pageQueryHistory(@Define("condition") String condition, @Bind("limit") int limit, @Bind("offset") int offset);

    @SqlQuery("""
            SELECT count(1) FROM query_history
            WHERE 1 = 1 
            """)
    Long count(@Define("condition") String condition);

    @SqlQuery("""
            SELECT FLOOR(created / 1000 / 60) AS minute,
                   backend_url AS backend_url,
                   COUNT(1) AS query_count
            FROM query_history
            WHERE created > :created
            GROUP BY minute, backend_url
            """)
    @UseRowMapper(MapMapper.class)
    List> findDistribution(long created);

    @SqlUpdate("""
            INSERT INTO query_history (query_id, query_text, backend_url, user_name, source, created)
            VALUES (:queryId, :queryText, :backendUrl, :userName, :source, :created)
            """)
    void insertHistory(String queryId, String queryText, String backendUrl, String userName, String source, long created);

    @SqlUpdate("""
            DELETE FROM query_history
            WHERE created < :created
            """)
    void deleteOldHistory(long created);
}