com.jfirer.jsql.model.UpdateModel Maven / Gradle / Ivy

Go to download
package com.jfirer.jsql.model;

import com.jfirer.jsql.annotation.TableDef;
import com.jfirer.jsql.metadata.TableEntityInfo;

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

public class UpdateModel extends Model
{
    private List setEntries;

    class UpdateEntry
    {
        final String propertyName;
        final Object value;

        UpdateEntry(String propertyName, Object value)
        {
            this.propertyName = propertyName;
            this.value = value;
        }
    }

    public UpdateModel set(String property, Object value)
    {
        if (setEntries == null)
        {
            setEntries = new LinkedList();
        }
        setEntries.add(new UpdateEntry(property, value));
        return this;
    }

    @Override
    public String _getSql()
    {
        StringBuilder cache = new StringBuilder();
        cache.append("update ").append(entityClass.getAnnotation(TableDef.class).name()).append(" ");
        if (setEntries == null)
        {
            throw new IllegalArgumentException("没有设置需要更新的字段");
        }
        cache.append("set ");
        Map columnInfoMap = TableEntityInfo.parse(entityClass).getPropertyNameKeyMap();
        for (UpdateEntry each : setEntries)
        {
            cache.append(columnInfoMap.get(each.propertyName).getColumnName()).append("=?,");
        }
        cache.setLength(cache.length() - 1);
        cache.append(' ');
        setWhereColumns(cache);
        return cache.toString();
    }

    @Override
    public List