chore: Couple of fixes

1. Ensure that query_language is handled identically for policy and policy_compliance resources
2. Add query_language to variable and output in main.tf

examples/resource_lacework_query/main.tf

@@ -16,6 +16,11 @@ variable "query_id" {
   default = "Lql_Terraform_Query"
 }
 
+variable "query_id" {
+  type    = string
+  default = ""
+}
+
 variable "query" {
   type    = string
   default = <<EOT
@@ -40,6 +45,10 @@ output "query_id" {
   value = lacework_query.example.id
 }
 
+output "query_language" {
+  value = lacework_query.example.query_language
+}
+
 output "query" {
   value = lacework_query.example.query
 }

lacework/resource_lacework_policy_compliance.go

@@ -135,7 +135,9 @@ func resourceLaceworkPolicyComplianceCreate(d *schema.ResourceData, meta interfa
 	}
 
 	d.SetId(response.Data.PolicyID)
-	d.Set("query_language", response.Data.QueryLanguage)
+	if response.Data.QueryLanguage != nil {
+		d.Set("query_language", response.Data.QueryLanguage)
+	}
 	d.Set("owner", response.Data.Owner)
 	d.Set("updated_time", response.Data.LastUpdateTime)
 	d.Set("updated_by", response.Data.LastUpdateUser)
@@ -159,7 +161,9 @@ func resourceLaceworkPolicyComplianceRead(d *schema.ResourceData, meta interface
 	d.SetId(response.Data.PolicyID)
 	d.Set("title", response.Data.Title)
 	d.Set("query_id", response.Data.QueryID)
-	d.Set("query_language", response.Data.QueryLanguage)
+	if response.Data.QueryLanguage != nil {
+		d.Set("query_language", response.Data.QueryLanguage)
+	}
 	d.Set("enabled", response.Data.Enabled)
 	d.Set("description", response.Data.Description)
 	d.Set("severity", response.Data.Severity)