added chat messages

ent/weapon_query.go

@@ -422,6 +422,10 @@ func (wq *WeaponQuery) sqlCount(ctx context.Context) (int, error) {
 	if len(wq.modifiers) > 0 {
 		_spec.Modifiers = wq.modifiers
 	}
+	_spec.Node.Columns = wq.fields
+	if len(wq.fields) > 0 {
+		_spec.Unique = wq.unique != nil && *wq.unique
+	}
 	return sqlgraph.CountNodes(ctx, wq.driver, _spec)
 }
 
@@ -493,6 +497,9 @@ func (wq *WeaponQuery) sqlQuery(ctx context.Context) *sql.Selector {
 		selector = wq.sql
 		selector.Select(selector.Columns(columns...)...)
 	}
+	if wq.unique != nil && *wq.unique {
+		selector.Distinct()
+	}
 	for _, m := range wq.modifiers {
 		m(selector)
 	}
@@ -780,9 +787,7 @@ func (wgb *WeaponGroupBy) sqlQuery() *sql.Selector {
 		for _, f := range wgb.fields {
 			columns = append(columns, selector.C(f))
 		}
-		for _, c := range aggregation {
-			columns = append(columns, c)
-		}
+		columns = append(columns, aggregation...)
 		selector.Select(columns...)
 	}
 	return selector.GroupBy(selector.Columns(wgb.fields...)...)