[feat]Basic Infra for node/storage

go.mod

@@ -13,8 +13,6 @@ require (
 
 require (
 	github.com/KyleBanks/depth v1.2.1 // indirect
-	github.com/PuerkitoBio/purell v1.2.1 // indirect
-	github.com/PuerkitoBio/urlesc v0.0.0-20170810143723-de5bf2ad4578 // indirect
 	github.com/armon/go-metrics v0.0.0-20180917152333-f0300d1749da // indirect
 	github.com/chenzhuoyu/iasm v0.9.1 // indirect
 	github.com/davecgh/go-spew v1.1.1 // indirect

internal/community/instance.go

@@ -4,7 +4,6 @@
 	"another_node/conf"
 	"another_node/internal/community/node"
 	"another_node/internal/community/storage"
-	"fmt"
 )
 
 type Community struct {
@@ -19,24 +18,30 @@
 	}
 }
 
+// BindAccount binding a web2 account
 func BindAccount(hashedAccount string, publicKey *string) error {
+	privateKeyValut := "WIP Private Key Vault"
+
 	if publicKey == nil {
 		// TODO: auto dispatch a web3 account
 		publicKey = new(string)
 		*publicKey = "WIP"
+		privateKeyValut = "Auto Dispatched Private Key Vault"
 	}
 
-	rpcAddress := new(string)
-	*rpcAddress = fmt.Sprintf("%s:%d", conf.GetNode().ExternalAddr, conf.GetNode().ExternalPort)
+	rpcAddress := conf.GetNode().ExternalAddr
+	rpcPort := conf.GetNode().ExternalPort
+	version := 0
 
-	if err := storage.UpsertMember(hashedAccount, publicKey, nil, rpcAddress, 0); err != nil {
+	if err := storage.UpsertMember(hashedAccount, *publicKey, privateKeyValut, rpcAddress, rpcPort, &version); err != nil {
 		return err
 	} else {
 		return community.Node.Broadcast(&node.Payload{
 			Account:    hashedAccount,
 			PublicKey:  *publicKey,
-			RpcAddress: *rpcAddress,
-			Version:    0,
+			RpcAddress: rpcAddress,
+			RpcPort:    rpcPort,
+			Version:    *&version,
 		})
 	}
 }

internal/community/node/mapping.go

@@ -12,5 +12,5 @@
 }
 
 func UpcomingHandler(payload *Payload) {
-	_ = storage.UpsertMember(payload.Account, &payload.PublicKey, nil, &payload.RpcAddress, payload.Version)
+	_ = storage.UpsertMember(payload.Account, payload.PublicKey, "", payload.RpcAddress, payload.RpcPort, &payload.Version)
 }

internal/community/node/payload.go

@@ -4,5 +4,6 @@ type Payload struct {
 	Account    string
 	PublicKey  string
 	RpcAddress string
+	RpcPort    int
 	Version    int
 }

internal/community/storage/member.go

@@ -13,6 +13,7 @@
 	Model
 	HashedAccount   string  `gorm:"column:hashed_account; type:varchar(1024); not null; uniqueIndex"`
 	RpcAddress      string  `gorm:"column:rpc_address; type:varchar(128); not null"`
+	RpcPort         int     `gorm:"column:rpc_port; type:int; not null; default:0"`
 	PublicKey       string  `gorm:"column:public_key; type:varchar(1024)"`
 	PrivateKeyVault *string `gorm:"column:private_key_vault; type:varchar(1024); null"`
 }
@@ -22,7 +23,7 @@
 }
 
 // UpsertMember update a member if exists and newer than old by version
-func UpsertMember(hashedAccount string, publicKey, privateKey, rpcAddress *string, version int) error {
+func UpsertMember(hashedAccount, publicKey, privateKey, rpcAddress string, rpcPort int, version *int) error {
 	db := conf.GetDbClient()
 
 	return db.Transaction(func(tx *gorm.DB) error {
@@ -32,43 +33,65 @@
 			return tx.Omit("updated_at", "id").Create(&Member{
 				Model: Model{
 					CreatedAt: time.Now(),
-					Version:   uint(version),
+					Version:   uint(*version),
 				},
-				HashedAccount:   hashedAccount,
-				PublicKey:       *publicKey,
-				PrivateKeyVault: privateKey,
-				RpcAddress:      *rpcAddress,
+				HashedAccount: hashedAccount,
+				PublicKey:     publicKey,
+				PrivateKeyVault: func() *string {
+					if len(privateKey) == 0 {
+						return nil
+					} else {
+						return &privateKey
+					}
+				}(),
+				RpcAddress: rpcAddress,
+				RpcPort:    rpcPort,
 			}).Error
 		} else {
-			if member.Version >= uint(version) {
+			if member.Version >= uint(*version) {
+				*version = int(member.Version)
 				return nil
 			}
 
-			if publicKey == nil {
+			if len(publicKey) == 0 {
 				tx.Omit("public_key")
 			}
-			if privateKey == nil {
+			if len(privateKey) == 0 {
 				tx.Omit("private_key_vault")
 			}
-			if rpcAddress == nil {
+			if len(rpcAddress) == 0 || rpcPort == 0 {
 				tx.Omit("rpc_address")
+				tx.Omit("rpc_port")
 			}
 			err := tx.Where("id=?", member.ID).Updates(Member{
 				PublicKey: func() string {
-					if publicKey != nil {
-						return *publicKey
+					if len(publicKey) > 0 {
+						return publicKey
 					} else {
 						return member.PublicKey
 					}
 				}(),
-				PrivateKeyVault: privateKey,
+				PrivateKeyVault: func() *string {
+					if len(privateKey) > 0 {
+						return &privateKey
+					} else {
+						return member.PrivateKeyVault
+					}
+				}(),
 				RpcAddress: func() string {
-					if rpcAddress != nil {
-						return *rpcAddress
+					if len(rpcAddress) > 0 {
+						return rpcAddress
 					} else {
 						return member.RpcAddress
 					}
 				}(),
+				RpcPort: func() int {
+					if rpcPort > 0 {
+						return rpcPort
+					} else {
+						return member.RpcPort
+					}
+				}(),
 			}).Error
 
 			return err

internal/community/storage/member_test.go

@@ -16,8 +16,10 @@ func TestMember(t *testing.T) {
 	prepare_test()
 
 	ptr := "B"
-	assert.NoError(t, UpsertMember("A", &ptr, &ptr, &ptr, 1))
-	assert.NoError(t, UpsertMember("A", &ptr, &ptr, &ptr, 2))
+	prtN := 1
+	ver := 1
+	assert.NoError(t, UpsertMember("A", ptr, ptr, ptr, prtN, &ver))
+	assert.NoError(t, UpsertMember("A", ptr, ptr, ptr, prtN, &ver))
 }
 
 func TestFindMember(t *testing.T) {
@@ -28,10 +30,13 @@ func TestFindMember(t *testing.T) {
 	}
 	prepare_test()
 
+	ptrN := 1
+	ver := 1
+
 	ptr := "B"
-	assert.NoError(t, UpsertMember("A", &ptr, &ptr, &ptr, 1))
+	assert.NoError(t, UpsertMember("A", ptr, ptr, ptr, ptrN, &ver))
 	ptr = "C"
-	assert.NoError(t, UpsertMember("A", &ptr, &ptr, &ptr, 1))
+	assert.NoError(t, UpsertMember("A", ptr, ptr, ptr, ptrN, &ver))
 
 	member, err := TryFindMember("A")
 	assert.NoError(t, err)

internal/community/storage/type.go

@@ -1,14 +1,11 @@
 package storage
 
 import (
-	"os"
 	"time"
 
 	"gorm.io/gorm"
 )
 
-var rpcAddress string
-
 type Model struct {
 	ID        uint           `gorm:"column:id; primarykey"`
 	CreatedAt time.Time      `gorm:"column:created_at; autoCreateTime"`
@@ -17,7 +14,3 @@ type Model struct {
 	DeletedBy string         `gorm:"column:deleted_by; type: varchar(1024); null"`
 	Version   uint           `gorm:"column:version; default:0;"`
 }
-
-func init() {
-	rpcAddress = os.Getenv("RPC_ADDRESS")
-}