Beego/orm/models_test.go

package orm

import (
	"fmt"
	"os"
	"strings"
	"time"

	// _ "github.com/bylevel/pq"
	_ "github.com/go-sql-driver/mysql"
	_ "github.com/lib/pq"
	_ "github.com/mattn/go-sqlite3"
)

type Data struct {
	Id       int
	Boolean  bool
	Char     string    `orm:"size(50)"`
	Text     string    `orm:"type(text)"`
	Date     time.Time `orm:"type(date)"`
	DateTime time.Time
	Byte     byte
	Rune     rune
	Int      int
	Int8     int8
	Int16    int16
	Int32    int32
	Int64    int64
	Uint     uint
	Uint8    uint8
	Uint16   uint16
	Uint32   uint32
	Uint64   uint64
	Float32  float32
	Float64  float64
	Decimal  float64 `orm:"digits(8);decimals(4)"`
}

type DataNull struct {
	Id       int
	Boolean  bool      `orm:"null"`
	Char     string    `orm:"size(50);null"`
	Text     string    `orm:"type(text);null"`
	Date     time.Time `orm:"type(date);null"`
	DateTime time.Time `orm:"null"`
	Byte     byte      `orm:"null"`
	Rune     rune      `orm:"null"`
	Int      int       `orm:"null"`
	Int8     int8      `orm:"null"`
	Int16    int16     `orm:"null"`
	Int32    int32     `orm:"null"`
	Int64    int64     `orm:"null"`
	Uint     uint      `orm:"null"`
	Uint8    uint8     `orm:"null"`
	Uint16   uint16    `orm:"null"`
	Uint32   uint32    `orm:"null"`
	Uint64   uint64    `orm:"null"`
	Float32  float32   `orm:"null"`
	Float64  float64   `orm:"null"`
	Decimal  float64   `orm:"digits(8);decimals(4);null"`
}

type User struct {
	Id         int
	UserName   string `orm:"size(30);unique"`
	Email      string `orm:"size(100)"`
	Password   string `orm:"size(100)"`
	Status     int16
	IsStaff    bool
	IsActive   bool      `orm:"default(1)"`
	Created    time.Time `orm:"auto_now_add;type(date)"`
	Updated    time.Time `orm:"auto_now"`
	Profile    *Profile  `orm:"null;rel(one);on_delete(set_null)"`
	Posts      []*Post   `orm:"reverse(many)" json:"-"`
	ShouldSkip string    `orm:"-"`
}

func NewUser() *User {
	obj := new(User)
	return obj
}

type Profile struct {
	Id    int
	Age   int16
	Money float64
	User  *User `orm:"reverse(one)" json:"-"`
}

func (u *Profile) TableName() string {
	return "user_profile"
}

func NewProfile() *Profile {
	obj := new(Profile)
	return obj
}

type Post struct {
	Id      int
	User    *User     `orm:"rel(fk)"`
	Title   string    `orm:"size(60)"`
	Content string    `orm:"type(text)"`
	Created time.Time `orm:"auto_now_add"`
	Updated time.Time `orm:"auto_now"`
	Tags    []*Tag    `orm:"rel(m2m)"`
}

func NewPost() *Post {
	obj := new(Post)
	return obj
}

type Tag struct {
	Id    int
	Name  string  `orm:"size(30)"`
	Posts []*Post `orm:"reverse(many)" json:"-"`
}

func NewTag() *Tag {
	obj := new(Tag)
	return obj
}

type Comment struct {
	Id      int
	Post    *Post     `orm:"rel(fk)"`
	Content string    `orm:"type(text)"`
	Parent  *Comment  `orm:"null;rel(fk)"`
	Created time.Time `orm:"auto_now_add"`
}

func NewComment() *Comment {
	obj := new(Comment)
	return obj
}

var DBARGS = struct {
	Driver string
	Source string
	Debug  string
}{
	os.Getenv("ORM_DRIVER"),
	os.Getenv("ORM_SOURCE"),
	os.Getenv("ORM_DEBUG"),
}

var (
	IsMysql    = DBARGS.Driver == "mysql"
	IsSqlite   = DBARGS.Driver == "sqlite3"
	IsPostgres = DBARGS.Driver == "postgres"
)

var dORM Ormer

var initSQLs = map[string]string{
	"mysql": "DROP TABLE IF EXISTS `user_profile`;\n" +
		"DROP TABLE IF EXISTS `user`;\n" +
		"DROP TABLE IF EXISTS `post`;\n" +
		"DROP TABLE IF EXISTS `tag`;\n" +
		"DROP TABLE IF EXISTS `post_tags`;\n" +
		"DROP TABLE IF EXISTS `comment`;\n" +
		"DROP TABLE IF EXISTS `data`;\n" +
		"DROP TABLE IF EXISTS `data_null`;\n" +
		"CREATE TABLE `user_profile` (\n" +
		"    `id` integer AUTO_INCREMENT NOT NULL PRIMARY KEY,\n" +
		"    `age` smallint NOT NULL,\n" +
		"    `money` double precision NOT NULL\n" +
		") ENGINE=INNODB;\n" +
		"CREATE TABLE `user` (\n" +
		"    `id` integer AUTO_INCREMENT NOT NULL PRIMARY KEY,\n" +
		"    `user_name` varchar(30) NOT NULL UNIQUE,\n" +
		"    `email` varchar(100) NOT NULL,\n" +
		"    `password` varchar(100) NOT NULL,\n" +
		"    `status` smallint NOT NULL,\n" +
		"    `is_staff` bool NOT NULL,\n" +
		"    `is_active` bool NOT NULL,\n" +
		"    `created` date NOT NULL,\n" +
		"    `updated` datetime NOT NULL,\n" +
		"    `profile_id` integer\n" +
		") ENGINE=INNODB;\n" +
		"CREATE TABLE `post` (\n" +
		"    `id` integer AUTO_INCREMENT NOT NULL PRIMARY KEY,\n" +
		"    `user_id` integer NOT NULL,\n" +
		"    `title` varchar(60) NOT NULL,\n" +
		"    `content` longtext NOT NULL,\n" +
		"    `created` datetime NOT NULL,\n" +
		"    `updated` datetime NOT NULL\n" +
		") ENGINE=INNODB;\n" +
		"CREATE TABLE `tag` (\n" +
		"    `id` integer AUTO_INCREMENT NOT NULL PRIMARY KEY,\n" +
		"    `name` varchar(30) NOT NULL\n" +
		") ENGINE=INNODB;\n" +
		"CREATE TABLE `post_tags` (\n" +
		"    `id` integer AUTO_INCREMENT NOT NULL PRIMARY KEY,\n" +
		"    `post_id` integer NOT NULL,\n" +
		"    `tag_id` integer NOT NULL,\n" +
		"    UNIQUE (`post_id`, `tag_id`)\n" +
		") ENGINE=INNODB;\n" +
		"CREATE TABLE `comment` (\n" +
		"    `id` integer AUTO_INCREMENT NOT NULL PRIMARY KEY,\n" +
		"    `post_id` integer NOT NULL,\n" +
		"    `content` longtext NOT NULL,\n" +
		"    `parent_id` integer,\n" +
		"    `created` datetime NOT NULL\n" +
		") ENGINE=INNODB;\n" +
		"CREATE TABLE `data` (\n" +
		"    `id` integer AUTO_INCREMENT NOT NULL PRIMARY KEY,\n" +
		"    `boolean` bool NOT NULL,\n" +
		"    `char` varchar(50) NOT NULL,\n" +
		"    `text` longtext NOT NULL,\n" +
		"    `date` date NOT NULL,\n" +
		"    `date_time` datetime NOT NULL,\n" +
		"    `byte` tinyint unsigned NOT NULL,\n" +
		"    `rune` integer NOT NULL,\n" +
		"    `int` integer NOT NULL,\n" +
		"    `int8` tinyint NOT NULL,\n" +
		"    `int16` smallint NOT NULL,\n" +
		"    `int32` integer NOT NULL,\n" +
		"    `int64` bigint NOT NULL,\n" +
		"    `uint` integer unsigned NOT NULL,\n" +
		"    `uint8` tinyint unsigned NULL,\n" +
		"    `uint16` smallint unsigned NOT NULL,\n" +
		"    `uint32` integer unsigned NOT NULL,\n" +
		"    `uint64` bigint unsigned NOT NULL,\n" +
		"    `float32` double precision NOT NULL,\n" +
		"    `float64` double precision NOT NULL,\n" +
		"    `decimal` numeric(8,4) NOT NULL\n" +
		") ENGINE=INNODB;\n" +
		"CREATE TABLE `data_null` (\n" +
		"    `id` integer AUTO_INCREMENT NOT NULL PRIMARY KEY,\n" +
		"    `boolean` bool,\n" +
		"    `char` varchar(50),\n" +
		"    `text` longtext,\n" +
		"    `date` date,\n" +
		"    `date_time` datetime,\n" +
		"    `byte` tinyint unsigned,\n" +
		"    `rune` integer,\n" +
		"    `int` integer,\n" +
		"    `int8` tinyint,\n" +
		"    `int16` smallint,\n" +
		"    `int32` integer,\n" +
		"    `int64` bigint,\n" +
		"    `uint` integer unsigned,\n" +
		"    `uint8` tinyint unsigned,\n" +
		"    `uint16` smallint unsigned,\n" +
		"    `uint32` integer unsigned,\n" +
		"    `uint64` bigint unsigned,\n" +
		"    `float32` double precision,\n" +
		"    `float64` double precision,\n" +
		"    `decimal` numeric(8,4)\n" +
		") ENGINE=INNODB;\n" +
		"CREATE INDEX `user_141c6eec` ON `user` (`profile_id`);\n" +
		"CREATE INDEX `post_fbfc09f1` ON `post` (`user_id`);\n" +
		"CREATE INDEX `comment_699ae8ca` ON `comment` (`post_id`);\n" +
		"CREATE INDEX `comment_63f17a16` ON `comment` (`parent_id`);",

	"sqlite3": `
DROP TABLE IF EXISTS "user_profile";
DROP TABLE IF EXISTS "user";
DROP TABLE IF EXISTS "post";
DROP TABLE IF EXISTS "tag";
DROP TABLE IF EXISTS "post_tags";
DROP TABLE IF EXISTS "comment";
DROP TABLE IF EXISTS "data";
DROP TABLE IF EXISTS "data_null";
CREATE TABLE "user_profile" (
    "id" integer NOT NULL PRIMARY KEY AUTOINCREMENT,
    "age" smallint NOT NULL,
    "money" real NOT NULL
);
CREATE TABLE "user" (
    "id" integer NOT NULL PRIMARY KEY AUTOINCREMENT,
    "user_name" varchar(30) NOT NULL UNIQUE,
    "email" varchar(100) NOT NULL,
    "password" varchar(100) NOT NULL,
    "status" smallint NOT NULL,
    "is_staff" bool NOT NULL,
    "is_active" bool NOT NULL,
    "created" date NOT NULL,
    "updated" datetime NOT NULL,
    "profile_id" integer
);
CREATE TABLE "post" (
    "id" integer NOT NULL PRIMARY KEY AUTOINCREMENT,
    "user_id" integer NOT NULL,
    "title" varchar(60) NOT NULL,
    "content" text NOT NULL,
    "created" datetime NOT NULL,
    "updated" datetime NOT NULL
);
CREATE TABLE "tag" (
    "id" integer NOT NULL PRIMARY KEY AUTOINCREMENT,
    "name" varchar(30) NOT NULL
);
CREATE TABLE "post_tags" (
    "id" integer NOT NULL PRIMARY KEY AUTOINCREMENT,
    "post_id" integer NOT NULL,
    "tag_id" integer NOT NULL,
    UNIQUE ("post_id", "tag_id")
);
CREATE TABLE "comment" (
    "id" integer NOT NULL PRIMARY KEY AUTOINCREMENT,
    "post_id" integer NOT NULL,
    "content" text NOT NULL,
    "parent_id" integer,
    "created" datetime NOT NULL
);
CREATE TABLE "data" (
    "id" integer NOT NULL PRIMARY KEY AUTOINCREMENT,
    "boolean" bool NOT NULL,
    "char" varchar(50) NOT NULL,
    "text" text NOT NULL,
    "date" date NOT NULL,
    "date_time" datetime NOT NULL,
    "byte" tinyint unsigned NOT NULL,
    "rune" integer NOT NULL,
    "int" integer NOT NULL,
    "int8" tinyint NOT NULL,
    "int16" smallint NOT NULL,
    "int32" integer NOT NULL,
    "int64" bigint NOT NULL,
    "uint" integer unsigned NOT NULL,
    "uint8" tinyint unsigned NOT NULL,
    "uint16" smallint unsigned NOT NULL,
    "uint32" integer unsigned NOT NULL,
    "uint64" bigint unsigned NOT NULL,
    "float32" real NOT NULL,
    "float64" real NOT NULL,
    "decimal" decimal
);
CREATE TABLE "data_null" (
    "id" integer NOT NULL PRIMARY KEY AUTOINCREMENT,
    "boolean" bool,
    "char" varchar(50),
    "text" text,
    "date" date,
    "date_time" datetime,
    "byte" tinyint unsigned,
    "rune" integer,
    "int" integer,
    "int8" tinyint,
    "int16" smallint,
    "int32" integer,
    "int64" bigint,
    "uint" integer unsigned,
    "uint8" tinyint unsigned,
    "uint16" smallint unsigned,
    "uint32" integer unsigned,
    "uint64" bigint unsigned,
    "float32" real,
    "float64" real,
    "decimal" decimal
);
CREATE INDEX "user_141c6eec" ON "user" ("profile_id");
CREATE INDEX "post_fbfc09f1" ON "post" ("user_id");
CREATE INDEX "comment_699ae8ca" ON "comment" ("post_id");
CREATE INDEX "comment_63f17a16" ON "comment" ("parent_id");
`,

	"postgres": `
DROP TABLE IF EXISTS "user_profile";
DROP TABLE IF EXISTS "user";
DROP TABLE IF EXISTS "post";
DROP TABLE IF EXISTS "tag";
DROP TABLE IF EXISTS "post_tags";
DROP TABLE IF EXISTS "comment";
DROP TABLE IF EXISTS "data";
DROP TABLE IF EXISTS "data_null";
CREATE TABLE "user_profile" (
    "id" serial NOT NULL PRIMARY KEY,
    "age" smallint NOT NULL,
    "money" double precision NOT NULL
);
CREATE TABLE "user" (
    "id" serial NOT NULL PRIMARY KEY,
    "user_name" varchar(30) NOT NULL UNIQUE,
    "email" varchar(100) NOT NULL,
    "password" varchar(100) NOT NULL,
    "status" smallint NOT NULL,
    "is_staff" boolean NOT NULL,
    "is_active" boolean NOT NULL,
    "created" date NOT NULL,
    "updated" timestamp with time zone NOT NULL,
    "profile_id" integer
);
CREATE TABLE "post" (
    "id" serial NOT NULL PRIMARY KEY,
    "user_id" integer NOT NULL,
    "title" varchar(60) NOT NULL,
    "content" text NOT NULL,
    "created" timestamp with time zone NOT NULL,
    "updated" timestamp with time zone NOT NULL
);
CREATE TABLE "tag" (
    "id" serial NOT NULL PRIMARY KEY,
    "name" varchar(30) NOT NULL
);
CREATE TABLE "post_tags" (
    "id" serial NOT NULL PRIMARY KEY,
    "post_id" integer NOT NULL,
    "tag_id" integer NOT NULL,
    UNIQUE ("post_id", "tag_id")
);
CREATE TABLE "comment" (
    "id" serial NOT NULL PRIMARY KEY,
    "post_id" integer NOT NULL,
    "content" text NOT NULL,
    "parent_id" integer,
    "created" timestamp with time zone NOT NULL
);
CREATE TABLE "data" (
    "id" serial NOT NULL PRIMARY KEY,
    "boolean" bool NOT NULL,
    "char" varchar(50) NOT NULL,
    "text" text NOT NULL,
    "date" date NOT NULL,
    "date_time" timestamp with time zone NOT NULL,
    "byte" smallint CHECK("byte" >= 0 AND "byte" <= 255) NOT NULL,
    "rune" integer NOT NULL,
    "int" integer NOT NULL,
    "int8" smallint CHECK("int8" >= -127 AND "int8" <= 128) NOT NULL,
    "int16" smallint NOT NULL,
    "int32" integer NOT NULL,
    "int64" bigint NOT NULL,
    "uint" bigint CHECK("uint" >= 0) NOT NULL,
    "uint8" smallint CHECK("uint8" >= 0 AND "uint8" <= 255) NOT NULL,
    "uint16" integer CHECK("uint16" >= 0) NOT NULL,
    "uint32" bigint CHECK("uint32" >= 0) NOT NULL,
    "uint64" bigint CHECK("uint64" >= 0) NOT NULL,
    "float32" double precision NOT NULL,
    "float64" double precision NOT NULL,
    "decimal" numeric(8, 4)
);
CREATE TABLE "data_null" (
    "id" serial NOT NULL PRIMARY KEY,
    "boolean" bool,
    "char" varchar(50),
    "text" text,
    "date" date,
    "date_time" timestamp with time zone,
    "byte" smallint CHECK("byte" >= 0 AND "byte" <= 255),
    "rune" integer,
    "int" integer,
    "int8" smallint CHECK("int8" >= -127 AND "int8" <= 128),
    "int16" smallint,
    "int32" integer,
    "int64" bigint,
    "uint" bigint CHECK("uint" >= 0),
    "uint8" smallint CHECK("uint8" >= 0 AND "uint8" <= 255),
    "uint16" integer CHECK("uint16" >= 0),
    "uint32" bigint CHECK("uint32" >= 0),
    "uint64" bigint CHECK("uint64" >= 0),
    "float32" double precision,
    "float64" double precision,
    "decimal" numeric(8, 4)
);
CREATE INDEX "user_profile_id" ON "user" ("profile_id");
CREATE INDEX "post_user_id" ON "post" ("user_id");
CREATE INDEX "comment_post_id" ON "comment" ("post_id");
CREATE INDEX "comment_parent_id" ON "comment" ("parent_id");
`}

func init() {
	// err := os.Setenv("TZ", "+00:00")
	// fmt.Println(err)

	RegisterModel(new(Data), new(DataNull))
	RegisterModel(new(User))
	RegisterModel(new(Profile))
	RegisterModel(new(Post))
	RegisterModel(new(Tag))
	RegisterModel(new(Comment))

	Debug, _ = StrTo(DBARGS.Debug).Bool()

	if DBARGS.Driver == "" || DBARGS.Source == "" {
		fmt.Println(`need driver and source!

Default DB Drivers.

  driver: url
   mysql: https://github.com/go-sql-driver/mysql
 sqlite3: https://github.com/mattn/go-sqlite3
postgres: https://github.com/lib/pq

eg: mysql
ORM_DRIVER=mysql ORM_SOURCE="root:root@/my_db?charset=utf8" go test github.com/astaxie/beego/orm
`)
		os.Exit(2)
	}

	RegisterDataBase("default", DBARGS.Driver, DBARGS.Source, 20)

	BootStrap()

	dORM = NewOrm()

	queries := strings.Split(initSQLs[DBARGS.Driver], ";")

	for _, query := range queries {
		query = strings.TrimSpace(query)
		if len(query) == 0 {
			continue
		}
		_, err := dORM.Raw(query).Exec()
		if err != nil {
			fmt.Println(err)
			os.Exit(2)
		}
	}
}