lbsadmin
/
multitenantStack
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Require an ORM in all models

This commit is contained in:
Lukas Bachschwell 2018-11-07 20:23:46 +01:00
parent 3347161ae9
commit 2adb24d8ce
5 changed files with 20 additions and 40 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
main.go
View File

@ -11,7 +11,7 @@ import (
)
func init() {
orm.RegisterDataBase("default", "postgres", "host=127.0.0.1 port=5435 user=postgres password=postgre dbname=company_template sslmode=disable")
orm.RegisterDataBase("default", "postgres", "host=127.0.0.1 port=5435 user=postgres password=postgre sslmode=disable")
auth.InitJWTService()
orm.DefaultTimeLoc = time.UTC
if beego.BConfig.RunMode == "dev" {

15
models/companyData.go
View File

@ -29,16 +29,14 @@ func init() {
// AddCompanyData insert a new CompanyData into database and returns
// last inserted Id on success.
func AddCompanyData(m *CompanyData) (id int64, err error) {
o := orm.NewOrm()
func AddCompanyData(o orm.Ormer, m *CompanyData) (id int64, err error) {
id, err = o.Insert(m)
return
}
// GetCompanyDataById retrieves CompanyData by Id. Returns error if
// Id doesn't exist
func GetCompanyDataById(id int) (v *CompanyData, err error) {
o := orm.NewOrm()
func GetCompanyDataById(o orm.Ormer, id int) (v *CompanyData, err error) {
v = &CompanyData{Id: id}
if err = o.Read(v); err == nil {
return v, nil
@ -48,9 +46,8 @@ func GetCompanyDataById(id int) (v *CompanyData, err error) {
// GetAllCompanyData retrieves all CompanyData matches certain condition. Returns empty list if
// no records exist
func GetAllCompanyData(query map[string]string, fields []string, sortby []string, order []string,
func GetAllCompanyData(o orm.Ormer, query map[string]string, fields []string, sortby []string, order []string,
offset int64, limit int64) (ml []interface{}, err error) {
o := orm.NewOrm()
qs := o.QueryTable(new(CompanyData))
// query k=v
for k, v := range query {
@ -126,8 +123,7 @@ func GetAllCompanyData(query map[string]string, fields []string, sortby []string
// UpdateCompanyData updates CompanyData by Id and returns error if
// the record to be updated doesn't exist
func UpdateCompanyDataById(m *CompanyData) (err error) {
o := orm.NewOrm()
func UpdateCompanyDataById(o orm.Ormer, m *CompanyData) (err error) {
v := CompanyData{Id: m.Id}
// ascertain id exists in the database
if err = o.Read(&v); err == nil {
@ -141,8 +137,7 @@ func UpdateCompanyDataById(m *CompanyData) (err error) {
// DeleteCompanyData deletes CompanyData by Id and returns error if
// the record to be deleted doesn't exist
func DeleteCompanyData(id int) (err error) {
o := orm.NewOrm()
func DeleteCompanyData(o orm.Ormer, id int) (err error) {
v := CompanyData{Id: id}
// ascertain id exists in the database
if err = o.Read(&v); err == nil {

18
models/companyUser.go
View File

@ -29,16 +29,14 @@ func init() {
// AddCompanyUser insert a new CompanyUser into database and returns
// last inserted Id on success.
func AddCompanyUser(m *CompanyUser) (id int64, err error) {
o := orm.NewOrm()
func AddCompanyUser(o orm.Ormer, m *CompanyUser) (id int64, err error) {
id, err = o.Insert(m)
return
}
// GetCompanyUserById retrieves CompanyUser by Id. Returns error if
// Id doesn't exist
func GetCompanyUserById(id int) (v *CompanyUser, err error) {
o := orm.NewOrm()
func GetCompanyUserById(o orm.Ormer, id int) (v *CompanyUser, err error) {
v = &CompanyUser{Id: id}
if err = o.Read(v); err == nil {
return v, nil
@ -48,9 +46,8 @@ func GetCompanyUserById(id int) (v *CompanyUser, err error) {
// GetAllCompanyUser retrieves all CompanyUser matches certain condition. Returns empty list if
// no records exist
func GetAllCompanyUser(query map[string]string, fields []string, sortby []string, order []string,
func GetAllCompanyUser(o orm.Ormer, query map[string]string, fields []string, sortby []string, order []string,
offset int64, limit int64) (ml []interface{}, err error) {
o := orm.NewOrm()
qs := o.QueryTable(new(CompanyUser))
// query k=v
for k, v := range query {
@ -124,10 +121,8 @@ func GetAllCompanyUser(query map[string]string, fields []string, sortby []string
return nil, err
}
// UpdateCompanyUser updates CompanyUser by Id and returns error if
// the record to be updated doesn't exist
func UpdateCompanyUserById(m *CompanyUser) (err error) {
o := orm.NewOrm()
// UpdateCompanyUserById updates CompanyUser by Id and returns error if the record to be updated doesn't exist
func UpdateCompanyUserById(o orm.Ormer, m *CompanyUser) (err error) {
v := CompanyUser{Id: m.Id}
// ascertain id exists in the database
if err = o.Read(&v); err == nil {
@ -141,8 +136,7 @@ func UpdateCompanyUserById(m *CompanyUser) (err error) {
// DeleteCompanyUser deletes CompanyUser by Id and returns error if
// the record to be deleted doesn't exist
func DeleteCompanyUser(id int) (err error) {
o := orm.NewOrm()
func DeleteCompanyUser(o orm.Ormer, id int) (err error) {
v := CompanyUser{Id: id}
// ascertain id exists in the database
if err = o.Read(&v); err == nil {

10
models/contact.go
View File

@ -32,8 +32,7 @@ func init() {
// AddContact insert a new Contact into database and returns
// last inserted Id on success.
func AddContact(m *Contact) (id int64, err error) {
o := orm.NewOrm()
func AddContact(o orm.Ormer, m *Contact) (id int64, err error) {
id, err = o.Insert(m)
return
}
@ -41,7 +40,6 @@ func AddContact(m *Contact) (id int64, err error) {
// GetContactById retrieves Contact by Id. Returns error if
// Id doesn't exist
func GetContactById(o orm.Ormer, id int) (v *Contact, err error) {
//o := orm.NewOrm()
v = &Contact{Id: id}
if err = o.Read(v); err == nil {
return v, nil
@ -128,8 +126,7 @@ func GetAllContact(o orm.Ormer, query map[string]string, fields []string, sortby
// UpdateContact updates Contact by Id and returns error if
// the record to be updated doesn't exist
func UpdateContactById(m *Contact) (err error) {
o := orm.NewOrm()
func UpdateContactById(o orm.Ormer, m *Contact) (err error) {
v := Contact{Id: m.Id}
// ascertain id exists in the database
if err = o.Read(&v); err == nil {
@ -143,8 +140,7 @@ func UpdateContactById(m *Contact) (err error) {
// DeleteContact deletes Contact by Id and returns error if
// the record to be deleted doesn't exist
func DeleteContact(id int) (err error) {
o := orm.NewOrm()
func DeleteContact(o orm.Ormer, id int) (err error) {
v := Contact{Id: id}
// ascertain id exists in the database
if err = o.Read(&v); err == nil {

15
models/post.go
View File

@ -29,16 +29,14 @@ func init() {
// AddPost insert a new Post into database and returns
// last inserted Id on success.
func AddPost(m *Post) (id int64, err error) {
o := orm.NewOrm()
func AddPost(o orm.Ormer, m *Post) (id int64, err error) {
id, err = o.Insert(m)
return
}
// GetPostById retrieves Post by Id. Returns error if
// Id doesn't exist
func GetPostById(id int) (v *Post, err error) {
o := orm.NewOrm()
func GetPostById(o orm.Ormer, id int) (v *Post, err error) {
v = &Post{Id: id}
if err = o.Read(v); err == nil {
return v, nil
@ -48,9 +46,8 @@ func GetPostById(id int) (v *Post, err error) {
// GetAllPost retrieves all Post matches certain condition. Returns empty list if
// no records exist
func GetAllPost(query map[string]string, fields []string, sortby []string, order []string,
func GetAllPost(o orm.Ormer, query map[string]string, fields []string, sortby []string, order []string,
offset int64, limit int64) (ml []interface{}, err error) {
o := orm.NewOrm()
qs := o.QueryTable(new(Post))
// query k=v
for k, v := range query {
@ -126,8 +123,7 @@ func GetAllPost(query map[string]string, fields []string, sortby []string, order
// UpdatePost updates Post by Id and returns error if
// the record to be updated doesn't exist
func UpdatePostById(m *Post) (err error) {
o := orm.NewOrm()
func UpdatePostById(o orm.Ormer, m *Post) (err error) {
v := Post{Id: m.Id}
// ascertain id exists in the database
if err = o.Read(&v); err == nil {
@ -141,8 +137,7 @@ func UpdatePostById(m *Post) (err error) {
// DeletePost deletes Post by Id and returns error if
// the record to be deleted doesn't exist
func DeletePost(id int) (err error) {
o := orm.NewOrm()
func DeletePost(o orm.Ormer, id int) (err error) {
v := Post{Id: id}
// ascertain id exists in the database
if err = o.Read(&v); err == nil {