Haruka Yoshihara
null+****@clear*****
Fri Dec 14 16:53:04 JST 2012
Haruka Yoshihara 2012-12-14 16:53:04 +0900 (Fri, 14 Dec 2012) New Revision: 45567c2aa347eebbcca3e7007e8b64ca7613eb36 https://github.com/groonga/groonga-query-log/commit/45567c2aa347eebbcca3e7007e8b64ca7613eb36 Log: Add groonga-query-log-extract from rroonga repository Added files: bin/groonga-query-log-extract Added: bin/groonga-query-log-extract (+118 -0) 100755 =================================================================== --- /dev/null +++ bin/groonga-query-log-extract 2012-12-14 16:53:04 +0900 (ab247da) @@ -0,0 +1,118 @@ +#!/usr/bin/env ruby +# -*- coding: utf-8 -*- +# +# Copyright (C) 2011 Kouhei Sutou <kou �� clear-code.com> +# Copyright (C) 2012 Haruka Yoshihara <yoshihara �� clear-code.com> +# +# This library is free software; you can redistribute it and/or +# modify it under the terms of the GNU Lesser General Public +# License version 2.1 as published by the Free Software Foundation. +# +# This library is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU +# Lesser General Public License for more details. +# +# You should have received a copy of the GNU Lesser General Public +# License along with this library; if not, write to the Free Software +# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA + +require "ostruct" +require "optparse" +require "pathname" + +require "groonga/query-log" + +options = OpenStruct.new +options.unify_format = nil +options.commands = [] +options.exclude_commands = [] +options.output_path = nil +option_parser = OptionParser.new do |parser| + parser.banner += " QUERY_LOG1 ..." + + available_formats = ["uri", "command"] + parser.on("--unify-format=FORMAT", + available_formats, + "Unify command format to FORMAT.", + "(#{available_formats.join(', ')})", + "[not unify]") do |format| + options.unify_format = format + end + + parser.on("--command=COMMAND", + "Extract only COMMAND.", + "To extract one or more commands,", + "specify this command a number of times.", + "Use /.../ as COMMAND to match command with regular expression.", + "[all commands]") do |command| + case command + when /\A\/(.*)\/(i)?\z/ + options.commands << Regexp.new($1, $2 == "i") + when + options.commands << command + end + end + + parser.on("--exclude-command=COMMAND", + "Don't extract COMMAND.", + "To ignore one or more commands,", + "specify this command a number of times.", + "Use /.../ as COMMAND to match command with regular expression.", + "[no commands]") do |command| + case command + when /\A\/(.*)\/(i)?\z/ + options.exclude_commands << Regexp.new($1, $2 == "i") + when + options.exclude_commands << command + end + end + + parser.on("--output=PATH", + "Output to PATH.", + "[standard output]") do |path| + options.output_path = path + end +end +args = option_parser.parse!(ARGV) + +if args.empty? + puts(option_parser) + exit(false) +end + +def target?(command, options) + name = command.name + if options.commands.any? {|target_command| target_command === name} + true + elsif options.exclude_commands.any? {|exclude_command| exclude_command == name} + false + else + true + end +end + +def extract(output, options) + parser = Groonga::QueryLog::Parser.new + parser.parse(ARGF) do |statistic| + command = statistic.command + next unless target?(command, options) + command_text = nil + case options.unify_format + when "uri" + command_text = command.to_uri_format unless command.uri_format? + when "command" + command_text = command.to_command_format unless command.command_format? + end + command_text ||= statistic.raw_command + output.puts(command_text) + end +end + +if options.output_path + File.open(options.output_path, "w") do |output| + extract(output, options) + end +else + extract($stdout, options) +end -------------- next part -------------- HTML����������������������������...ダウンロード
