Skip to content

Commit 89301a2

Browse files
authored
Merge pull request #172 from 0chain/feat/env-allocid
Use wallet, allocaion from env flags
2 parents a5718ab + cbfe33c commit 89301a2

File tree

2 files changed

+21
-9
lines changed

2 files changed

+21
-9
lines changed

cmd/gateway/zcn/gateway-zcn.go

Lines changed: 16 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -35,13 +35,14 @@ const (
3535
)
3636

3737
var (
38-
configDir string
39-
allocationID string
40-
nonce int64
41-
encrypt bool
42-
compress bool
43-
workDir string
44-
serverConfig serverOptions
38+
configDir string
39+
allocationID string
40+
nonce int64
41+
encrypt bool
42+
compress bool
43+
workDir string
44+
serverConfig serverOptions
45+
walletDetails string
4546
)
4647

4748
var zFlags = []cli.Flag{
@@ -54,12 +55,19 @@ var zFlags = []cli.Flag{
5455
Name: "allocationId",
5556
Usage: "Allocation id of an allocation",
5657
Destination: &allocationID,
58+
EnvVar: "ALLOCATION_ID",
5759
},
5860
cli.Int64Flag{
5961
Name: "nonce",
6062
Usage: "nonce to use in transaction",
6163
Destination: &nonce,
6264
},
65+
cli.StringFlag{
66+
Name: "wallet",
67+
Usage: "wallet details",
68+
Destination: &walletDetails,
69+
EnvVar: "WALLET_DETAILS",
70+
},
6371
}
6472

6573
func init() {
@@ -128,7 +136,7 @@ var (
128136

129137
// NewGatewayLayer initializes 0chain gosdk and return zcnObjects
130138
func (z *ZCN) NewGatewayLayer(creds madmin.Credentials) (minio.ObjectLayer, error) {
131-
err := initializeSDK(configDir, allocationID, nonce)
139+
err := initializeSDK(configDir, allocationID, nonce, walletDetails)
132140
if err != nil {
133141
return nil, err
134142
}

cmd/gateway/zcn/initSDK.go

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@ type serverOptions struct {
2828
MaxConcurrentRequests int `json:"max_concurrent_requests"`
2929
}
3030

31-
func initializeSDK(configDir, allocid string, nonce int64) error {
31+
func initializeSDK(configDir, allocid string, nonce int64, walletDetails string) error {
3232
if configDir == "" {
3333
var err error
3434
configDir, err = getDefaultConfigDir()
@@ -97,6 +97,10 @@ func initializeSDK(configDir, allocid string, nonce int64) error {
9797
zcncore.SetLogLevel(3)
9898
sdk.SetLogLevel(3)
9999

100+
walletInfo := string(walletBytes)
101+
if walletDetails != "" {
102+
walletInfo = walletDetails
103+
}
100104
err = client.InitSDK("{}", cfg.BlockWorker, cfg.ChainID, cfg.SignatureScheme, nonce, false, cfg.MinSubmit, cfg.MinConfirmation, cfg.ConfirmationChainLength, cfg.SharderConsensous)
101105
if err != nil {
102106
return err

0 commit comments

Comments
 (0)