Base key (#31)

* add QueryTeePodr2Puk

* add QueryTeeWorkerList and QueryTeeWorkerPeerID

* format

core/chain/chain.go

@@ -87,6 +87,12 @@ type Chain interface {
  QuerySpacePricePerGib() (string, error)
  //
  QueryNetSnapShot() (NetSnapShot, error)
+ //
+ QueryTeePodr2Puk() ([]byte, error)
+ //
+ QueryTeeWorker(pubkey []byte) ([]byte, error)
+ //
+ QueryTeeWorkerList() ([]TeeWorkerInfo, error)
 }
 
 type chainClient struct {

core/chain/state.go

@@ -633,6 +633,109 @@ func (c *chainClient) QueryNetSnapShot() (NetSnapShot, error) {
  return data, nil
 }
 
+func (c *chainClient) QueryTeePodr2Puk() ([]byte, error) {
+ defer func() {
+ if err := recover(); err != nil {
+ println(utils.RecoverError(err))
+ }
+ }()
+ var data TeePodr2Pk
+
+ if !c.IsChainClientOk() {
+ c.SetChainState(false)
+ return nil, ERR_RPC_CONNECTION
+ }
+ c.SetChainState(true)
+
+ key, err := types.CreateStorageKey(
+ c.metadata,
+ TEEWORKER,
+ TEEPODR2Pk,
+ )
+ if err != nil {
+ return nil, errors.Wrap(err, "[CreateStorageKey]")
+ }
+
+ ok, err := c.api.RPC.State.GetStorageLatest(key, &data)
+ if err != nil {
+ return nil, errors.Wrap(err, "[GetStorageLatest]")
+ }
+ if !ok {
+ return nil, ERR_RPC_EMPTY_VALUE
+ }
+
+ return []byte(string(data[:])), nil
+}
+
+func (c *chainClient) QueryTeeWorker(pubkey []byte) ([]byte, error) {
+ defer func() {
+ if err := recover(); err != nil {
+ println(utils.RecoverError(err))
+ }
+ }()
+ var data TeeWorkerInfo
+
+ acc, err := types.NewAccountID(pubkey)
+ if err != nil {
+ return nil, errors.Wrap(err, "[NewAccountID]")
+ }
+
+ owner, err := codec.Encode(*acc)
+ if err != nil {
+ return nil, errors.Wrap(err, "[EncodeToBytes]")
+ }
+
+ if !c.IsChainClientOk() {
+ c.SetChainState(false)
+ return nil, ERR_RPC_CONNECTION
+ }
+ c.SetChainState(true)
+
+ key, err := types.CreateStorageKey(
+ c.metadata,
+ TEEWORKER,
+ TEEWORKERMAP,
+ owner,
+ )
+ if err != nil {
+ return nil, errors.Wrap(err, "[CreateStorageKey]")
+ }
+
+ ok, err := c.api.RPC.State.GetStorageLatest(key, &data)
+ if err != nil {
+ return nil, errors.Wrap(err, "[GetStorageLatest]")
+ }
+ if !ok {
+ return nil, ERR_RPC_EMPTY_VALUE
+ }
+
+ return []byte(string(data.Peer_id[:])), nil
+}
+
+func (c *chainClient) QueryTeeWorkerList() ([]TeeWorkerInfo, error) {
+ var list []TeeWorkerInfo
+ key := createPrefixedKey(TEEWORKER, TEEWORKERMAP)
+ keys, err := c.api.RPC.State.GetKeysLatest(key)
+ if err != nil {
+ return list, errors.Wrap(err, "[GetKeysLatest]")
+ }
+ set, err := c.api.RPC.State.QueryStorageAtLatest(keys)
+ if err != nil {
+ return list, errors.Wrap(err, "[QueryStorageAtLatest]")
+ }
+ for _, elem := range set {
+ for _, change := range elem.Changes {
+ var teeWorker TeeWorkerInfo
+ if err := codec.Decode(change.StorageData, &teeWorker); err != nil {
+ log.Println(err)
+ continue
+ }
+ list = append(list, teeWorker)
+ }
+ }
+ return list, nil
+}
+
 // Pallert
 const (
  _FILEBANK = "FileBank"

core/chain/types.go

@@ -47,7 +47,8 @@ const (
  MINERITEMS = "MinerItems"
 
  // TEEWORKER
- SCHEDULERMAP = "SchedulerMap"
+ TEEWORKERMAP = "TeeWorkerMap"
+ TEEPODR2Pk = "TeePodr2Pk"
 
  // FILEBANK
  FILE = "File"
@@ -126,6 +127,8 @@ var (
 
 type FileHash [64]types.U8
 type Random [20]types.U8
+type TeePodr2Pk [294]types.U8
+type PeerID [53]types.U8
 
 type MinerInfo struct {
  PeerId types.U64
@@ -231,3 +234,8 @@ type MinerSnapShot struct {
  Idle_space types.U128
  Service_space types.U128
 }
+
+type TeeWorkerInfo struct {
+ Peer_id PeerID
+ Stash_account types.AccountID
+}

core/client/client.go

@@ -36,6 +36,9 @@ type Client interface {
  QuerySpacePricePerGib() (string, error)
  QueryNetSnapShot() (chain.NetSnapShot, error)
  QueryChallenge(pubkey []byte) (ChallengeInfo, error)
+ QueryTeePodr2Puk() ([]byte, error)
+ QueryTeeWorkerPeerID(pubkey []byte) ([]byte, error)
+ QueryTeeWorkerList() ([]chain.TeeWorkerInfo, error)
  CheckBucketName(bucketname string) bool
  CreateBucket(owner []byte, bucketname string) (string, error)
  ProcessingData(path string) ([]SegmentInfo, string, error)

core/client/idlefile.go

@@ -28,7 +28,7 @@ func (c *Cli) SubmitIdleFile(size uint64, blockNum, blocksize, scansize uint32,
  filehash[i] = types.U8(hash[i])
  }
  var idlefiles = []chain.IdleMetaInfo{
- chain.IdleMetaInfo{
+ {
  Size: types.NewU64(size),
  BlockNum: types.NewU32(blockNum),
  BlockSize: types.NewU32(blocksize),

core/client/query.go

@@ -54,3 +54,15 @@ func (c *Cli) QueryUserSpaceInfo(pubkey []byte) (chain.UserSpaceInfo, error) {
 func (c *Cli) QuerySpacePricePerGib() (string, error) {
  return c.Chain.QuerySpacePricePerGib()
 }
+
+func (c *Cli) QueryTeePodr2Puk() ([]byte, error) {
+ return c.Chain.QueryTeePodr2Puk()
+}
+
+func (c *Cli) QueryTeeWorkerList() ([]chain.TeeWorkerInfo, error) {
+ return c.Chain.QueryTeeWorkerList()
+}
+
+func (c *Cli) QueryTeeWorkerPeerID(pubkey []byte) ([]byte, error) {
+ return c.Chain.QueryTeeWorker(pubkey)
+}