package gudusoft.gsqlparser.dlineage.statistics;

import java.util.*;

/**
 * 文件统计信息类
 */
public class FileStatistics {
    private String file;
    private long file_size;
    private int line_count;
    private int total_statements;
    private int select_count;
    private int insert_count;
    private int update_count;
    private int delete_count;
    private int merge_count;
    private int create_table_count;
    private int create_temp_table_count;
    private int ctas_count;
    private int create_view_count;
    private int create_temp_view_count;
    private int drop_count;
    private int truncate_count;
    private int join_count;
    private int parse_error_count;
    private Map<String, Integer> join_types;
    private int subquery_depth;
    private int cte_count;
    private int case_count;
    private int union_count;
    private int window_function_count;
    private int aggregate_function_count;
    private Set<String> distinct_table_references;
    private int where_predicate_count;
    private List<String> objects_created;
    private List<String> objects_read;

    public FileStatistics(String file) {
        this.file = file;
        this.file_size = 0;
        this.line_count = 0;
        this.join_types = new HashMap<>();
        this.distinct_table_references = new HashSet<>();
        this.objects_created = new ArrayList<>();
        this.objects_read = new ArrayList<>();
        this.parse_error_count = 0;
    }

    // Getters and setters
    public String getFile() {
        return file;
    }

    public void setFile(String file) {
        this.file = file;
    }

    public int getTotal_statements() {
        return total_statements;
    }

    public void setTotal_statements(int total_statements) {
        this.total_statements = total_statements;
    }

    public void incrementTotal_statements() {
        this.total_statements++;
    }

    public int getSelect_count() {
        return select_count;
    }

    public void setSelect_count(int select_count) {
        this.select_count = select_count;
    }

    public void incrementSelect_count() {
        this.select_count++;
    }

    public int getInsert_count() {
        return insert_count;
    }

    public void setInsert_count(int insert_count) {
        this.insert_count = insert_count;
    }

    public void incrementInsert_count() {
        this.insert_count++;
    }

    public int getUpdate_count() {
        return update_count;
    }

    public void setUpdate_count(int update_count) {
        this.update_count = update_count;
    }

    public void incrementUpdate_count() {
        this.update_count++;
    }

    public int getDelete_count() {
        return delete_count;
    }

    public void setDelete_count(int delete_count) {
        this.delete_count = delete_count;
    }

    public void incrementDelete_count() {
        this.delete_count++;
    }

    public int getMerge_count() {
        return merge_count;
    }

    public void setMerge_count(int merge_count) {
        this.merge_count = merge_count;
    }

    public void incrementMerge_count() {
        this.merge_count++;
    }

    public int getCreate_table_count() {
        return create_table_count;
    }

    public void setCreate_table_count(int create_table_count) {
        this.create_table_count = create_table_count;
    }

    public void incrementCreate_table_count() {
        this.create_table_count++;
    }

    public int getCreate_temp_table_count() {
        return create_temp_table_count;
    }

    public void setCreate_temp_table_count(int create_temp_table_count) {
        this.create_temp_table_count = create_temp_table_count;
    }

    public void incrementCreate_temp_table_count() {
        this.create_temp_table_count++;
    }

    public int getCtas_count() {
        return ctas_count;
    }

    public void setCtas_count(int ctas_count) {
        this.ctas_count = ctas_count;
    }

    public void incrementCtas_count() {
        this.ctas_count++;
    }

    public int getCreate_view_count() {
        return create_view_count;
    }

    public void setCreate_view_count(int create_view_count) {
        this.create_view_count = create_view_count;
    }

    public void incrementCreate_view_count() {
        this.create_view_count++;
    }

    public int getCreate_temp_view_count() {
        return create_temp_view_count;
    }

    public void setCreate_temp_view_count(int create_temp_view_count) {
        this.create_temp_view_count = create_temp_view_count;
    }

    public void incrementCreate_temp_view_count() {
        this.create_temp_view_count++;
    }

    public int getDrop_count() {
        return drop_count;
    }

    public void setDrop_count(int drop_count) {
        this.drop_count = drop_count;
    }

    public void incrementDrop_count() {
        this.drop_count++;
    }

    public int getTruncate_count() {
        return truncate_count;
    }

    public void setTruncate_count(int truncate_count) {
        this.truncate_count = truncate_count;
    }

    public void incrementTruncate_count() {
        this.truncate_count++;
    }

    public int getParse_error_count() {
        return parse_error_count;
    }

    public void setParse_error_count(int parse_error_count) {
        this.parse_error_count = parse_error_count;
    }

    public void incrementParse_error_count() {
        this.parse_error_count++;
    }

    public int getJoin_count() {
        return join_count;
    }

    public void setJoin_count(int join_count) {
        this.join_count = join_count;
    }

    public void incrementJoin_count() {
        this.join_count++;
    }


    public Map<String, Integer> getJoin_types() {
        return join_types;
    }

    public void setJoin_types(Map<String, Integer> join_types) {
        this.join_types = join_types;
    }

    public void addJoin_type(String join_type) {
        this.join_types.put(join_type, this.join_types.getOrDefault(join_type, 0) + 1);
    }

    public int getSubquery_depth() {
        return subquery_depth;
    }

    public void setSubquery_depth(int subquery_depth) {
        this.subquery_depth = subquery_depth;
    }

    public int getCte_count() {
        return cte_count;
    }

    public void setCte_count(int cte_count) {
        this.cte_count = cte_count;
    }

    public void incrementCte_count() {
        this.cte_count++;
    }

    public void addCte_count(int count) {
        this.cte_count += count;
    }

    public int getCase_count() {
        return case_count;
    }

    public void setCase_count(int case_count) {
        this.case_count = case_count;
    }

    public void incrementCase_count() {
        this.case_count++;
    }

    public void addCase_count(int count) {
        this.case_count += count;
    }

    public int getUnion_count() {
        return union_count;
    }

    public void setUnion_count(int union_count) {
        this.union_count = union_count;
    }

    public void incrementUnion_count() {
        this.union_count++;
    }

    public int getWindow_function_count() {
        return window_function_count;
    }

    public void setWindow_function_count(int window_function_count) {
        this.window_function_count = window_function_count;
    }

    public void incrementWindow_function_count() {
        this.window_function_count++;
    }

    public void addWindow_function_count(int count) {
        this.window_function_count += count;
    }

    public int getAggregate_function_count() {
        return aggregate_function_count;
    }

    public void setAggregate_function_count(int aggregate_function_count) {
        this.aggregate_function_count = aggregate_function_count;
    }

    public void incrementAggregate_function_count() {
        this.aggregate_function_count++;
    }

    public void addAggregate_function_count(int count) {
        this.aggregate_function_count += count;
    }

    public Set<String> getDistinct_table_references() {
        return distinct_table_references;
    }

    public void setDistinct_table_references(Set<String> distinct_table_references) {
        this.distinct_table_references = distinct_table_references;
    }

    public void addTableReference(String tableReference) {
        this.distinct_table_references.add(tableReference);
    }

    public void addTableReferences(Set<String> tableReferences) {
        this.distinct_table_references.addAll(tableReferences);
    }

    public int getWhere_predicate_count() {
        return where_predicate_count;
    }

    public void setWhere_predicate_count(int where_predicate_count) {
        this.where_predicate_count = where_predicate_count;
    }

    public void incrementWhere_predicate_count() {
        this.where_predicate_count++;
    }

    public void addWhere_predicate_count(int count) {
        this.where_predicate_count += count;
    }

    public List<String> getObjects_created() {
        return objects_created;
    }

    public void setObjects_created(List<String> objects_created) {
        this.objects_created = objects_created;
    }

    public void addObject_created(String object) {
        this.objects_created.add(object);
    }

    public List<String> getObjects_read() {
        return objects_read;
    }

    public void setObjects_read(List<String> objects_read) {
        this.objects_read = objects_read;
    }

    public long getFile_size() {
        return file_size;
    }

    public void setFile_size(long file_size) {
        this.file_size = file_size;
    }

    public int getLine_count() {
        return line_count;
    }

    public void setLine_count(int line_count) {
        this.line_count = line_count;
    }

    public void addObject_read(String object) {
        if (!this.objects_read.contains(object)) {
            this.objects_read.add(object);
        }
    }

    public Map toJSON() {
        Map json = new LinkedHashMap();
        json.put("file", file);
        json.put("file_size", file_size);
        json.put("line_count", line_count);
        json.put("total_statements", total_statements);
        json.put("parse_error_count", parse_error_count);
        json.put("select_count", select_count);
        json.put("insert_count", insert_count);
        json.put("update_count", update_count);
        json.put("delete_count", delete_count);
        json.put("merge_count", merge_count);
        json.put("create_table_count", create_table_count);
        json.put("create_temp_table_count", create_temp_table_count);
        json.put("ctas_count", ctas_count);
        json.put("create_view_count", create_view_count);
        json.put("create_temp_view_count", create_temp_view_count);
        json.put("drop_count", drop_count);
        json.put("truncate_count", truncate_count);
        json.put("join_count", join_count);
        json.put("join_types", join_types);
        json.put("subquery_depth", subquery_depth);
        json.put("cte_count", cte_count);
        json.put("case_count", case_count);
        json.put("union_count", union_count);
        json.put("window_function_count", window_function_count);
        json.put("aggregate_function_count", aggregate_function_count);
        json.put("distinct_table_references", distinct_table_references.size());
        json.put("where_predicate_count", where_predicate_count);
        json.put("objects_created", objects_created);
        json.put("objects_read", objects_read);
        return json;
    }
}