From 9935922176927ebb86434db4f264a400a4225581 Mon Sep 17 00:00:00 2001
From: Lunny Xiao <xiaolunwen@gmail.com>
Date: Sun, 16 Jun 2013 15:10:35 +0800
Subject: [PATCH] bug fixed

---
 engine.go  |  2 +-
 session.go | 18 ++++++++++++++----
 2 files changed, 15 insertions(+), 5 deletions(-)

diff --git a/engine.go b/engine.go
index 7f994c6f..87f7661b 100644
--- a/engine.go
+++ b/engine.go
@@ -132,7 +132,7 @@ func (engine *Engine) AutoMapType(t reflect.Type) *Table {
 	table, ok := engine.Tables[t]
 	if !ok {
 		table = engine.MapType(t)
-		//engine.Tables[t] = table
+		engine.Tables[t] = table
 	}
 	return table
 }
diff --git a/session.go b/session.go
index 7a518395..7f7d465f 100644
--- a/session.go
+++ b/session.go
@@ -673,9 +673,13 @@ func (session *Session) InsertOne(bean interface{}) (int64, error) {
 		if col.IsAutoIncrement && fieldValue.Int() == 0 {
 			continue
 		}
-		if table, ok := session.Engine.Tables[fieldValue.Type()]; ok {
-			pkField := reflect.Indirect(fieldValue).FieldByName(table.PKColumn().FieldName)
-			args = append(args, pkField.Interface())
+		if fieldTable, ok := session.Engine.Tables[fieldValue.Type()]; ok {
+			if fieldTable.PrimaryKey != "" {
+				pkField := reflect.Indirect(fieldValue).FieldByName(fieldTable.PKColumn().FieldName)
+				args = append(args, pkField.Interface())
+			} else {
+				continue
+			}
 		} else {
 			args = append(args, val)
 		}
@@ -696,10 +700,16 @@ func (session *Session) InsertOne(bean interface{}) (int64, error) {
 	}
 
 	id, err := res.LastInsertId()
-
 	if err != nil {
 		return -1, err
 	}
+	if id > 0 && table.PrimaryKey != "" {
+		pkValue := reflect.Indirect(reflect.ValueOf(bean)).FieldByName(table.PKColumn().FieldName)
+		if pkValue.CanSet() {
+			var v interface{} = id
+			pkValue.Set(reflect.ValueOf(v))
+		}
+	}
 
 	return id, nil
 }