add restore api
This commit is contained in:
		
							parent
							
								
									8c146f8981
								
							
						
					
					
						commit
						a0d19fc26d
					
				@ -641,13 +641,13 @@ func (db *DB) Restore(values interface{}) (tx *DB) {
 | 
				
			|||||||
	tx.Statement.Unscoped = true
 | 
						tx.Statement.Unscoped = true
 | 
				
			||||||
	tx.Statement.Parse(values)
 | 
						tx.Statement.Parse(values)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	dest := make(map[string]interface{})
 | 
						tx.Statement.Dest = make(map[string]interface{})
 | 
				
			||||||
	for _, c := range tx.Statement.Schema.DeleteClauses {
 | 
					 | 
				
			||||||
		fieldName := c.(SoftDeleteDeleteClause).Field.DBName
 | 
					 | 
				
			||||||
		dest[fieldName] = nil
 | 
					 | 
				
			||||||
	}
 | 
					 | 
				
			||||||
 | 
					
 | 
				
			||||||
	tx.Statement.Dest = dest
 | 
						for _, c := range tx.Statement.Schema.DeleteClauses {
 | 
				
			||||||
 | 
							if optimizer, ok := c.(UnDeleteDester); ok {
 | 
				
			||||||
 | 
								optimizer.SetUnDeleteDest(tx.Statement)
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
	return tx.callbacks.Update().Execute(tx)
 | 
						return tx.callbacks.Update().Execute(tx)
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
@ -162,3 +162,13 @@ func (sd SoftDeleteDeleteClause) ModifyStatement(stmt *Statement) {
 | 
				
			|||||||
		stmt.Build("UPDATE", "SET", "WHERE")
 | 
							stmt.Build("UPDATE", "SET", "WHERE")
 | 
				
			||||||
	}
 | 
						}
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					func (sd SoftDeleteDeleteClause) SetUnDeleteDest(stmt *Statement) {
 | 
				
			||||||
 | 
						if stmt.Unscoped {
 | 
				
			||||||
 | 
							if sd.Field.DefaultValue != "" {
 | 
				
			||||||
 | 
								stmt.SetColumn(sd.Field.DBName, clause.Expr{SQL: sd.Field.DefaultValue}, true)
 | 
				
			||||||
 | 
							} else {
 | 
				
			||||||
 | 
								stmt.SetColumn(sd.Field.DBName, nil, true)
 | 
				
			||||||
 | 
							}
 | 
				
			||||||
 | 
						}
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
@ -57,6 +57,11 @@ type StatementModifier interface {
 | 
				
			|||||||
	ModifyStatement(*Statement)
 | 
						ModifyStatement(*Statement)
 | 
				
			||||||
}
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					// UnDeleteDester set undelete dest interface
 | 
				
			||||||
 | 
					type UnDeleteDester interface {
 | 
				
			||||||
 | 
						SetUnDeleteDest(*Statement)
 | 
				
			||||||
 | 
					}
 | 
				
			||||||
 | 
					
 | 
				
			||||||
// WriteString write string
 | 
					// WriteString write string
 | 
				
			||||||
func (stmt *Statement) WriteString(str string) (int, error) {
 | 
					func (stmt *Statement) WriteString(str string) (int, error) {
 | 
				
			||||||
	return stmt.SQL.WriteString(str)
 | 
						return stmt.SQL.WriteString(str)
 | 
				
			||||||
 | 
				
			|||||||
		Loading…
	
	
			
			x
			
			
		
	
		Reference in New Issue
	
	Block a user