From 89377aeaa2d0502c801c9599c6b87b3679e6c4a1 Mon Sep 17 00:00:00 2001
From: JianBo He <heeejianbo@gmail.com>
Date: Sat, 8 Aug 2020 17:04:58 +0800
Subject: [PATCH] improve(type): refine the message header types

---
 src/emqx_types.erl | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/src/emqx_types.erl b/src/emqx_types.erl
index 739858729..8de122bc6 100644
--- a/src/emqx_types.erl
+++ b/src/emqx_types.erl
@@ -181,9 +181,15 @@
 -type(subscriber() :: {pid(), subid()}).
 -type(payload() :: binary() | iodata()).
 -type(message() :: #message{}).
--type(flag() :: atom()).
+-type(flag() :: sys | dup | retain | atom()).
 -type(flags() :: #{flag() := boolean()}).
--type(headers() :: map()).
+-type(headers() :: #{proto_ver => ver(),
+                     protocol => protocol(),
+                     username => username(),
+                     peerhost => peerhost(),
+                     properties => properties(),
+                     atom() => term()}).
+
 -type(banned() :: #banned{}).
 -type(deliver() :: {deliver, topic(), message()}).
 -type(delivery() :: #delivery{}).