PHP -- Codeigniter框架数据库insert函数的改写

本文主要介绍了PHP -- Codeigniter框架数据库insert函数的改写方法，欢迎大家的学习。

对数据表新增记录时，有时有一些唯一索引的字段已经有值了，插入数据的时候会提示错误。

解决办法是先查询这个值是否存在，存在就跟新update，不存在就insert。

下面重写了Codeigniter框架模型中的insert方法

/**
 * 插入单条记录(Insert)
 *
 * @param array $params 插入字段及值数组
 * @param string $table 插入表名，未指定的场景插入主表
 * @param array $columns 插入表中所属字段名数组，若指定则按此进行过滤
 * @param array $updateColumns 若唯一键冲突时需要更新的列数组
 * @return mixed FALSE:没有插入的字段 int:插入记录id
 *
 * @throws \C_DB_Exception
 */
public function insert($params, $table = NULL, $columns = array(), $updateColumns = array())
{
    // 如果未指定表名，则认为对本Model层的主表进行操作
    if (empty($table)) {
        $table = $this->table;
        $columns = $this->columns;
    }
    // 过滤掉不属于该表中的字段
    if (!empty($columns)) {
        foreach ($params as $column => $val) {
            if (in_array($column, $columns)) {
                $row[$column] = $val;
            }
        }
    } // 不过滤
    else {
        $row = $params;
    }
    // 没有插入字段，直接返回
    if (!isset($row)) {
        return FALSE;
    }
    // 保证创建时间、更新时间必须设置
    if (!isset($row['create_at']) && in_array('create_at', $columns)) {
        $this->load->helper('common_function');
        $datetime = get_datetime();
        $row['create_at'] = $datetime;
        $row['update_at'] = $datetime;
    }
    // 封装SQL文，过滤不安全因素
    $sql = $this->db->insert_string($table, $row);
    // 若唯一键冲突时需要更新的列数组，则再次封装SQL文
    if (!empty($updateColumns)) {
        $updateValues = '';
        foreach ($updateColumns as $updateColumn) {
            if (isset($updateValues[0])) {
                $updateValues .= ',';
            }
            $updateValues .= $updateColumn . '=VALUES(' . $updateColumn . ')';
        }
        $sql = $sql . ' ON DUPLICATE KEY UPDATE ' . $updateValues;
    }
    // SQL执行
    $query = $this->db->query($sql);
    // SQL执行失败，抛出异常
    if (FALSE === $query) {
        $code = $this->db->_error_number();
        $msg = $this->db->_error_message();
        C_log::Error($code . ':' . $msg, $this->user_id, $sql, self::STATUS_DB_LOG);
        throw new C_DB_Exception('', C_DB_Exception::DB_ERR_CODE);
    }
    // SQL文写入DB日志
    if (LOG_UPDATE_SQL) {
        C_log::Info('', $this->user_id, $sql, self::STATUS_DB_LOG);
    }
    // 取得插入ID返回
    $id = $this->db->insert_id();
    return $id;
}

SQL语句的一种写法

语法： INSERT INTO table (a) VALUES (a) ON DUPLICATE KEY UPDATE row='$row'

例子：

INSERT INTO it_user (uid,name,qq,create_at) VALUES 
('$userid','$name','$qq',NOW()) ON DUPLICATE KEY UPDATE name='$name',qq='$qq',phone='$phone'";

以上这篇使用PHP -- Codeigniter框架数据库insert函数的改写就是小编分享给大家的全部内容了，希望能给大家一个参考。