add log functions to all querys

Signed-off-by: Coiling-Dragon <stanislav.viyachev@limechain.tech>
hashgraph · Jul 1, 2023 · 43ec950 · 43ec950
1 parent 6a95c08
commit 43ec950
Show file tree

Hide file tree

Showing 19 changed files with 173 additions and 0 deletions.
diff --git a/account_balance_query.go b/account_balance_query.go
@@ -312,3 +312,12 @@ func (query *AccountBalanceQuery) SetPaymentTransactionID(transactionID Transact
 	query.paymentTransactionIDs._Clear()._Push(transactionID)._SetLocked(true)
 	return query
 }
+
+func (query *AccountBalanceQuery) SetLogLevel(level LogLevel) *AccountBalanceQuery {
+	query.Query.logLevel = &level
+	return query
+}
+
+func (query *AccountBalanceQuery) GetLogLevel() *LogLevel {
+	return query.Query.logLevel
+}
diff --git a/account_info_query.go b/account_info_query.go
@@ -336,3 +336,12 @@ func (query *AccountInfoQuery) SetPaymentTransactionID(transactionID Transaction
 	query.paymentTransactionIDs._Clear()._Push(transactionID)._SetLocked(true)
 	return query
 }
+
+func (query *AccountInfoQuery) SetLogLevel(level LogLevel) *AccountInfoQuery {
+	query.Query.logLevel = &level
+	return query
+}
+
+func (query *AccountInfoQuery) GetLogLevel() *LogLevel {
+	return query.Query.logLevel
+}
diff --git a/account_records_query.go b/account_records_query.go
@@ -338,3 +338,12 @@ func (query *AccountRecordsQuery) SetPaymentTransactionID(transactionID Transact
 	query.paymentTransactionIDs._Clear()._Push(transactionID)._SetLocked(true)
 	return query
 }
+
+func (query *AccountRecordsQuery) SetLogLevel(level LogLevel) *AccountRecordsQuery {
+	query.Query.logLevel = &level
+	return query
+}
+
+func (query *AccountRecordsQuery) GetLogLevel() *LogLevel {
+	return query.Query.logLevel
+}
diff --git a/account_stakers_query.go b/account_stakers_query.go
@@ -345,3 +345,12 @@ func (query *AccountStakersQuery) SetPaymentTransactionID(transactionID Transact
 	query.paymentTransactionIDs._Clear()._Push(transactionID)._SetLocked(true)
 	return query
 }
+
+func (query *AccountStakersQuery) SetLogLevel(level LogLevel) *AccountStakersQuery {
+	query.Query.logLevel = &level
+	return query
+}
+
+func (query *AccountStakersQuery) GetLogLevel() *LogLevel {
+	return query.Query.logLevel
+}
diff --git a/address_book_query.go b/address_book_query.go
@@ -38,6 +38,7 @@ type AddressBookQuery struct {
 	maxAttempts uint64
 	fileID      *FileID
 	limit       int32
+	logLevel	*LogLevel
 }
 
 // Query the mirror node for the address book.
@@ -191,3 +192,12 @@ func (query *AddressBookQuery) Execute(client *Client) (NodeAddressBook, error)
 		NodeAddresses: result,
 	}, subClientError
 }
+
+func (query *AddressBookQuery) SetLogLevel(level LogLevel) *AddressBookQuery {
+	query.logLevel = &level
+	return query
+}
+
+func (query *AddressBookQuery) GetLogLevel() *LogLevel {
+	return query.logLevel
+}
diff --git a/contract_bytecode_query.go b/contract_bytecode_query.go
@@ -327,3 +327,12 @@ func (query *ContractBytecodeQuery) SetPaymentTransactionID(transactionID Transa
 	query.paymentTransactionIDs._Clear()._Push(transactionID)._SetLocked(true)
 	return query
 }
+
+func (query *ContractBytecodeQuery) SetLogLevel(level LogLevel) *ContractBytecodeQuery {
+	query.Query.logLevel = &level
+	return query
+}
+
+func (query *ContractBytecodeQuery) GetLogLevel() *LogLevel {
+	return query.Query.logLevel
+}
diff --git a/contract_call_query.go b/contract_call_query.go
@@ -410,3 +410,12 @@ func (query *ContractCallQuery) SetPaymentTransactionID(transactionID Transactio
 	query.paymentTransactionIDs._Clear()._Push(transactionID)._SetLocked(true)
 	return query
 }
+
+func (query *ContractCallQuery) SetLogLevel(level LogLevel) *ContractCallQuery {
+	query.Query.logLevel = &level
+	return query
+}
+
+func (query *ContractCallQuery) GetLogLevel() *LogLevel {
+	return query.Query.logLevel
+}
diff --git a/contract_info_query.go b/contract_info_query.go
@@ -337,3 +337,12 @@ func (query *ContractInfoQuery) SetPaymentTransactionID(transactionID Transactio
 	query.paymentTransactionIDs._Clear()._Push(transactionID)._SetLocked(true)
 	return query
 }
+
+func (query *ContractInfoQuery) SetLogLevel(level LogLevel) *ContractInfoQuery {
+	query.Query.logLevel = &level
+	return query
+}
+
+func (query *ContractInfoQuery) GetLogLevel() *LogLevel {
+	return query.Query.logLevel
+}
diff --git a/file_contents_query.go b/file_contents_query.go
@@ -325,3 +325,12 @@ func (query *FileContentsQuery) SetPaymentTransactionID(transactionID Transactio
 	query.paymentTransactionIDs._Clear()._Push(transactionID)._SetLocked(true)
 	return query
 }
+
+func (query *FileContentsQuery) SetLogLevel(level LogLevel) *FileContentsQuery {
+	query.Query.logLevel = &level
+	return query
+}
+
+func (query *FileContentsQuery) GetLogLevel() *LogLevel {
+	return query.Query.logLevel
+}
diff --git a/file_info_query.go b/file_info_query.go
@@ -343,3 +343,12 @@ func (query *FileInfoQuery) SetPaymentTransactionID(transactionID TransactionID)
 	query.paymentTransactionIDs._Clear()._Push(transactionID)._SetLocked(true)
 	return query
 }
+
+func (query *FileInfoQuery) SetLogLevel(level LogLevel) *FileInfoQuery {
+	query.Query.logLevel = &level
+	return query
+}
+
+func (query *FileInfoQuery) GetLogLevel() *LogLevel {
+	return query.Query.logLevel
+}
diff --git a/live_hash_query.go b/live_hash_query.go
@@ -350,3 +350,12 @@ func (query *LiveHashQuery) SetMaxRetry(count int) *LiveHashQuery {
 func (query *LiveHashQuery) GetMaxRetry() int {
 	return query.Query.maxRetry
 }
+
+func (query *LiveHashQuery) SetLogLevel(level LogLevel) *LiveHashQuery {
+	query.Query.logLevel = &level
+	return query
+}
+
+func (query *LiveHashQuery) GetLogLevel() *LogLevel {
+	return query.Query.logLevel
+}
diff --git a/network_version_info_query.go b/network_version_info_query.go
@@ -266,3 +266,12 @@ func (query *NetworkVersionInfoQuery) SetPaymentTransactionID(transactionID Tran
 	query.paymentTransactionIDs._Clear()._Push(transactionID)._SetLocked(true)
 	return query
 }
+
+func (query *NetworkVersionInfoQuery) SetLogLevel(level LogLevel) *NetworkVersionInfoQuery {
+	query.Query.logLevel = &level
+	return query
+}
+
+func (query *NetworkVersionInfoQuery) GetLogLevel() *LogLevel {
+	return query.Query.logLevel
+}
diff --git a/schedule_info_query.go b/schedule_info_query.go
@@ -329,3 +329,12 @@ func (query *ScheduleInfoQuery) SetPaymentTransactionID(transactionID Transactio
 	query.paymentTransactionIDs._Clear()._Push(transactionID)._SetLocked(true)
 	return query
 }
+
+func (query *ScheduleInfoQuery) SetLogLevel(level LogLevel) *ScheduleInfoQuery {
+	query.Query.logLevel = &level
+	return query
+}
+
+func (query *ScheduleInfoQuery) GetLogLevel() *LogLevel {
+	return query.Query.logLevel
+}
diff --git a/token_info_query.go b/token_info_query.go
@@ -326,3 +326,12 @@ func (query *TokenInfoQuery) SetPaymentTransactionID(transactionID TransactionID
 	query.paymentTransactionIDs._Clear()._Push(transactionID)._SetLocked(true)
 	return query
 }
+
+func (query *TokenInfoQuery) SetLogLevel(level LogLevel) *TokenInfoQuery {
+	query.Query.logLevel = &level
+	return query
+}
+
+func (query *TokenInfoQuery) GetLogLevel() *LogLevel {
+	return query.Query.logLevel
+}
diff --git a/token_nft_info_query.go b/token_nft_info_query.go
@@ -389,3 +389,12 @@ func (query *TokenNftInfoQuery) SetPaymentTransactionID(transactionID Transactio
 	query.paymentTransactionIDs._Clear()._Push(transactionID)._SetLocked(true)
 	return query
 }
+
+func (query *TokenNftInfoQuery) SetLogLevel(level LogLevel) *TokenNftInfoQuery {
+	query.Query.logLevel = &level
+	return query
+}
+
+func (query *TokenNftInfoQuery) GetLogLevel() *LogLevel {
+	return query.Query.logLevel
+}
diff --git a/topic_info_query.go b/topic_info_query.go
@@ -326,3 +326,12 @@ func (query *TopicInfoQuery) SetPaymentTransactionID(transactionID TransactionID
 	query.paymentTransactionIDs._Clear()._Push(transactionID)._SetLocked(true)
 	return query
 }
+
+func (query *TopicInfoQuery) SetLogLevel(level LogLevel) *TopicInfoQuery {
+	query.Query.logLevel = &level
+	return query
+}
+
+func (query *TopicInfoQuery) GetLogLevel() *LogLevel {
+	return query.Query.logLevel
+}
diff --git a/topic_message_query.go b/topic_message_query.go
@@ -48,6 +48,7 @@ type TopicMessageQuery struct {
 	startTime         *time.Time
 	endTime           *time.Time
 	limit             uint64
+	logLevel		  *LogLevel
 }
 
 // NewTopicMessageQuery creates TopicMessageQuery which
@@ -304,3 +305,12 @@ func _DefaultRetryHandler(err error) bool {
 		return false
 	}
 }
+
+func (query *TopicMessageQuery) SetLogLevel(level LogLevel) *TopicMessageQuery {
+	query.logLevel = &level
+	return query
+}
+
+func (query *TopicMessageQuery) GetLogLevel() *LogLevel {
+	return query.logLevel
+}
diff --git a/transaction_receipt_query.go b/transaction_receipt_query.go
@@ -370,3 +370,12 @@ func (query *TransactionReceiptQuery) SetPaymentTransactionID(transactionID Tran
 	query.paymentTransactionIDs._Clear()._Push(transactionID)._SetLocked(true)
 	return query
 }
+
+func (query *TransactionReceiptQuery) SetLogLevel(level LogLevel) *TransactionReceiptQuery {
+	query.Query.logLevel = &level
+	return query
+}
+
+func (query *TransactionReceiptQuery) GetLogLevel() *LogLevel {
+	return query.Query.logLevel
+}
diff --git a/transaction_record_query.go b/transaction_record_query.go
@@ -421,3 +421,12 @@ func (query *TransactionRecordQuery) SetPaymentTransactionID(transactionID Trans
 	query.paymentTransactionIDs._Clear()._Push(transactionID)._SetLocked(true)
 	return query
 }
+
+func (query *TransactionRecordQuery) SetLogLevel(level LogLevel) *TransactionRecordQuery {
+	query.Query.logLevel = &level
+	return query
+}
+
+func (query *TransactionRecordQuery) GetLogLevel() *LogLevel {
+	return query.Query.logLevel
+}