SetMemoryOnly()

flusher.go

@@ -239,6 +239,9 @@ func (f *flusher) flush(root bool, lazy bool, transaction bool, entities ...Enti
 
 	for _, entity := range entities {
 		initIfNeeded(f.engine.registry, entity)
+		if entity.getORM().memory {
+			continue
+		}
 		schema := entity.getORM().tableSchema
 		if !transaction && schema.GetMysql(f.engine).inTransaction {
 			transaction = true
@@ -407,9 +410,7 @@ func (f *flusher) executeDeletes(lazy bool) {
 				if logEvent != nil {
 					logEvents = append(logEvents, logEvent)
 				}
-				if !schema.inMemory {
-					f.fillLazyQuery(db.GetPoolConfig().GetCode(), deleteSQLPrefix+strconv.FormatUint(id, 10)+")", false, id, logEvents)
-				}
+				f.fillLazyQuery(db.GetPoolConfig().GetCode(), deleteSQLPrefix+strconv.FormatUint(id, 10)+")", false, id, logEvents)
 			}
 			if hasLocalCache || hasRedis {
 				cacheKey := schema.getCacheKey(id)
@@ -489,9 +490,7 @@ func (f *flusher) executeInserts(flushPackage *flushPackage, lazy bool) {
 					logEvents = append(logEvents, logEvent)
 				}
 			}
-			if !schema.inMemory {
-				f.fillLazyQuery(db.GetPoolConfig().GetCode(), sql, true, 0, logEvents)
-			}
+			f.fillLazyQuery(db.GetPoolConfig().GetCode(), sql, true, 0, logEvents)
 		} else {
 			res := db.Exec(sql)
 			id := res.LastInsertId()
@@ -611,9 +610,7 @@ func (f *flusher) flushUpdate(entity Entity, bindBuilder *bindBuilder, currentID
 		if logEvent != nil {
 			logEvents = append(logEvents, logEvent)
 		}
-		if !schema.inMemory {
-			f.fillLazyQuery(db.GetPoolConfig().GetCode(), sql, false, currentID, logEvents)
-		}
+		f.fillLazyQuery(db.GetPoolConfig().GetCode(), sql, false, currentID, logEvents)
 	} else {
 		if f.updateSQLs == nil {
 			f.updateSQLs = make(map[string][]string)

orm.go

@@ -36,6 +36,7 @@ type Entity interface {
 	SetEntityLogMeta(key string, value interface{})
 	SetField(field string, value interface{}) error
 	Clone() Entity
+	SetMemoryOnly(memory bool)
 }
 
 type ORM struct {
@@ -51,6 +52,7 @@ type ORM struct {
 	elem                 reflect.Value
 	idElem               reflect.Value
 	logMeta              map[string]interface{}
+	memory               bool
 }
 
 func DisableCacheHashCheck() {
@@ -107,6 +109,10 @@ func (orm *ORM) SetOnDuplicateKeyUpdate(bind Bind) {
 	orm.onDuplicateKeyUpdate = bind
 }
 
+func (orm *ORM) SetMemoryOnly(memory bool) {
+	orm.memory = memory
+}
+
 func (orm *ORM) SetEntityLogMeta(key string, value interface{}) {
 	if orm.logMeta == nil {
 		orm.logMeta = make(map[string]interface{})

table_schema.go

@@ -132,7 +132,6 @@ type tableSchema struct {
 	logTableName            string
 	skipLogs                []string
 	hasUUID                 bool
-	inMemory                bool
 	mapBindToScanPointer    mapBindToScanPointer
 	mapPointerToValue       mapPointerToValue
 }
@@ -466,7 +465,6 @@ func (tableSchema *tableSchema) init(registry *Registry, entityType reflect.Type
 			return fmt.Errorf("entity %s with uuid enabled must be unit64", entityType.String())
 		}
 	}
-	inMemory := tableSchema.getTag("memory", "true", "false") == "true"
 	uniqueIndices := make(map[string]map[int]string)
 	uniqueIndicesSimple := make(map[string][]string)
 	uniqueIndicesSimpleGlobal := make(map[string][]string)
@@ -575,7 +573,6 @@ func (tableSchema *tableSchema) init(registry *Registry, entityType reflect.Type
 	tableSchema.uniqueIndicesGlobal = uniqueIndicesSimpleGlobal
 	tableSchema.hasLog = logPoolName != ""
 	tableSchema.hasUUID = hasUUID
-	tableSchema.inMemory = inMemory
 	tableSchema.logPoolName = logPoolName
 	tableSchema.logTableName = fmt.Sprintf("_log_%s_%s", tableSchema.mysqlPoolName, tableSchema.tableName)
 	tableSchema.skipLogs = skipLogs