package utils

import (
	. "gitee.com/wjzqx010/ear-picking/config"
	. "gitee.com/wjzqx010/ear-picking/pojo"
)

type ParameterMappingTokenHandler struct {
	parameterMappings []ParamMapping
}

func NewParameterMappingTokenHandler() *ParameterMappingTokenHandler {
	return &ParameterMappingTokenHandler{
		parameterMappings: make([]ParamMapping, 0, DataCapacity),
	}
}

/*
HandleToken
  - 通用处理器，保存替换的字段名，返回替换的值
  - 用于sql语句中条件字段的替换，返回“?”占位符
*/
func (p *ParameterMappingTokenHandler) HandleToken(content string) string {
	p.parameterMappings = append(p.parameterMappings, buildParameterMapping(content))
	return "?"
}

func buildParameterMapping(content string) ParamMapping {

	mapping := NewParamMapping(content)
	return mapping
}

func (p *ParameterMappingTokenHandler) SetParameterMappings(list []ParamMapping) {
	p.parameterMappings = list
}

func (p *ParameterMappingTokenHandler) GetParameterMappings() []ParamMapping {
	return p.parameterMappings
}