HyperNet/Nexus
Archived
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

✨ Nex.Rx, and easier to use alloc mq apis

Browse Source
This commit is contained in:
LittleSheep
2024-10-24 23:52:38 +08:00
parent 23d14e4e46
commit 421834ae5c
6 changed files with 60 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
pkg/nex/cruda/allocator.go
View File

@ -10,9 +10,9 @@ import (
)
func (v *CrudConn) AllocDatabase(name string) (string, error) {
conn := v.Conn.GetNexusGrpcConn()
conn := v.n.GetNexusGrpcConn()
ctx := context.Background()
ctx = metadata.AppendToOutgoingContext(ctx, "client_id", v.Conn.Info.Id)
ctx = metadata.AppendToOutgoingContext(ctx, "client_id", v.n.Info.Id)
out, err := proto.NewDatabaseServiceClient(conn).AllocDatabase(ctx, &proto.AllocDatabaseRequest{
Name: name,
})
@ -24,13 +24,13 @@ func (v *CrudConn) AllocDatabase(name string) (string, error) {
if err != nil {
return "", err
}
v.db = db
v.Db = db
return dsn, nil
}
func MigrateModel[T any](v *CrudConn, model T) error {
if v.db == nil {
if v.Db == nil {
return fmt.Errorf("database has not been allocated")
}
return v.db.AutoMigrate(model)
return v.Db.AutoMigrate(model)
}

16
pkg/nex/cruda/command.go
View File

@ -15,7 +15,7 @@ func AddModel[T any](v *CrudConn, model T, id, prefix string, tags []string) err
funcCmds := []string{".list", "", "", "", ""}
funcMethods := []string{"get", "get", "put", "patch", "delete"}
for idx, fn := range funcList {
if err := v.Conn.AddCommand(prefix+id+funcCmds[idx], funcMethods[idx], tags, fn(v)); err != nil {
if err := v.n.AddCommand(prefix+id+funcCmds[idx], funcMethods[idx], tags, fn(v)); err != nil {
return err
}
}
@ -31,12 +31,12 @@ func cmdList[T any](c *CrudConn) nex.CommandHandler {
var str T
var count int64
if err := c.db.Model(str).Count(&count).Error; err != nil {
if err := c.Db.Model(str).Count(&count).Error; err != nil {
return err
}
var out []T
if err := c.db.Offset(skip).Limit(take).Find(&out).Error; err != nil {
if err := c.Db.Offset(skip).Limit(take).Find(&out).Error; err != nil {
return err
}
@ -55,7 +55,7 @@ func cmdGet[T any](c *CrudConn) nex.CommandHandler {
}
var out T
if err := c.db.First(&out, "id = ?", id).Error; err != nil {
if err := c.Db.First(&out, "id = ?", id).Error; err != nil {
if errors.Is(err, gorm.ErrRecordNotFound) {
return ctx.Write([]byte(err.Error()), "text/plain", http.StatusNotFound)
}
@ -75,7 +75,7 @@ func cmdCreate[T any](c *CrudConn) nex.CommandHandler {
return ctx.Write([]byte(err.Error()), "text/plain+error", http.StatusBadRequest)
}
if err := c.db.Create(&payload).Error; err != nil {
if err := c.Db.Create(&payload).Error; err != nil {
return err
}
@ -98,11 +98,11 @@ func cmdUpdate[T any](c *CrudConn) nex.CommandHandler {
}
var out T
if err := c.db.Model(out).Where("id = ?", id).Updates(&payload).Error; err != nil {
if err := c.Db.Model(out).Where("id = ?", id).Updates(&payload).Error; err != nil {
return err
}
if err := c.db.First(&out, "id = ?", id).Error; err != nil {
if err := c.Db.First(&out, "id = ?", id).Error; err != nil {
return err
}
@ -118,7 +118,7 @@ func cmdDelete[T any](c *CrudConn) nex.CommandHandler {
}
var out T
if err := c.db.Delete(&out, "id = ?", id).Error; err != nil {
if err := c.Db.Delete(&out, "id = ?", id).Error; err != nil {
if errors.Is(err, gorm.ErrRecordNotFound) {
return ctx.Write([]byte(err.Error()), "text/plain", http.StatusNotFound)
}

6
pkg/nex/cruda/conn.go
View File

@ -6,13 +6,13 @@ import (
)
type CrudConn struct {
Conn *nex.Conn
n *nex.Conn
db *gorm.DB
Db *gorm.DB
}
func NewCrudaConn(conn *nex.Conn) *CrudConn {
return &CrudConn{
Conn: conn,
n: conn,
}
}